/* A simple server in the internet domain using TCP The port number is passed as an argument */ #include #include #include #include #include #include #include static char* text[]={ "HTTP/1.0 404 Not Found\n\n", "HTTP/1.0 200 OK\nContent-Type: text/html\n\nHell, World!", "HTTP/1.0 200 OK\nContent-Type: text/css\n\nbody {background-color: #ff0000}" }; void error(const char *msg) { perror(msg); exit(1); } int main(int argc, char *argv[]) { int sockfd, newsockfd, portno, page; socklen_t clilen; char buffer[256]; char *token; struct sockaddr_in serv_addr, cli_addr; int n; if (argc < 2) { fprintf(stderr,"ERROR, no port provided\n"); exit(1); } sockfd = socket(AF_INET, SOCK_STREAM, 0); if (sockfd < 0) error("ERROR opening socket"); bzero((char *) &serv_addr, sizeof(serv_addr)); portno = atoi(argv[1]); serv_addr.sin_family = AF_INET; serv_addr.sin_addr.s_addr = INADDR_ANY; serv_addr.sin_port = htons(portno); if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) error("ERROR on binding"); while (1) { listen(sockfd,5); clilen = sizeof(cli_addr); newsockfd = accept(sockfd, (struct sockaddr *) &cli_addr, &clilen); if (newsockfd < 0) error("ERROR on accept"); bzero(buffer,256); page=0; n = read(newsockfd,buffer,255); if (n>0) { buffer[n]=0; token=strtok(buffer," "); printf("%s\n",token); if (!strcmp(token,"GET")) { token=strtok(NULL," "); printf(">%s<\n",token); if (!strcmp(token,"/")) { page=1; } else if (!strcmp(token,"/index.html")) { page=1; } else if (!strcmp(token,"/style.css")) { page=2; } } } printf("%d\n",page); while (n==255) { n = read(newsockfd,buffer,255); }; if (n < 0) error("ERROR reading from socket"); n = write(newsockfd,text[page],strlen(text[page])); if (n < 0) error("ERROR writing to socket"); close(newsockfd); }; close(sockfd); return 0; }