simple web server code

不飘不浮

this is a simple web server code wrote by me.. it may has some incorrect.

hope boss can tell me how to improve.

char in array or pointer problem confused me..

had tested in os x 10.5.2
:wink:

:wink: :wink:

1. 
   
2. #ifndef _SERVER_C_
   
3. #define _SERVER_C_
   
4. 
   
5. #include <stdio.h>
   
6. #include <string.h>
   
7. #include <unistd.h>
   
8. #include <fcntl.h>
   
9. #include <sys/types.h>
   
10. #include <sys/stat.h>
    
11. #include <sys/socket.h>
    
12. #include <arpa/inet.h>
    
13. 
    
14. int port = 8080;
    
15. int backlog = 10;
    
16. 
    
17. struct request
    
18. {
    
19.         char *fullpath;
    
20.         char *last_modified;
    
21.         char *content_type;
    
22. };
    
23. 
    
24. int main( void )
    
25. {
    
26.         int sockfd;
    
27.         if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
    
28.         {
    
29.                 printf("Error create main socket!\n");
    
30.                 return -1;
    
31.         }
    
32.        
    
33.         struct sockaddr_in addr;
    
34.         memset(&addr, 0, sizeof(struct sockaddr_in));
    
35.         addr.sin_family = AF_INET;
    
36.         addr.sin_port = htons(port);
    
37.         addr.sin_addr.s_addr = INADDR_ANY;
    
38.         if (bind(sockfd, (struct sockaddr *)&addr, (socklen_t)sizeof(addr)) == -1)
    
39.         {
    
40.                 printf("Error to bind address!\n");
    
41.                 return -1;
    
42.         }
    
43.        
    
44.         if (listen(sockfd, backlog) == -1)
    
45.         {
    
46.                 printf("Error to listen on port %d\n", port);
    
47.                 return -1;
    
48.         }
    
49.        
    
50.        
    
51.         int newfd;
    
52.         struct sockaddr_in remote_addr;
    
53.         socklen_t remote_sinsize;
    
54.         while (1)
    
55.         {
    
56.                 if ((newfd = accept(sockfd, (struct sockaddr *)&remote_addr, &remote_sinsize)) == -1)
    
57.                 {
    
58.                         printf("Error to accept new connection.\n");
    
59.                         continue;
    
60.                 }
    
61.                
    
62.                 int rv;
    
63.                 char buf[2048];
    
64.                 char *request = NULL;
    
65.                 if ((rv = recv(newfd, buf, sizeof(buf), 0)) < 0)
    
66.                 {
    
67.                         printf("recv error!\n");
    
68.                 }
    
69.                
    
70.                 int c;
    
71.                 for (c=0; c < rv; c++)
    
72.                 {
    
73.                         if (buf[c] == '\n')
    
74.                         {
    
75.                                 buf[c] = '\0';
    
76.                         }
    
77.                        
    
78.                         if (c > 1 && buf[c-1] == '\r')
    
79.                         {
    
80.                                 buf[c-1] = '\0';
    
81.                                 request = buf;
    
82.                                 break;
    
83.                         }
    
84.                 }
    
85.                
    
86.                 //printf("%s\n", request);
    
87.                
    
88.                 char uri[1024];
    
89.                 char fullpath[2048] = "/Users/jimmyfan/Sites";
    
90.                
    
91.                 sscanf(buf, "%*s %s %*s", uri);
    
92.                 for (c=0; c < sizeof(uri); c++)
    
93.                 {
    
94.                         if (uri[c] == '/' && uri[c+1] == '\0')
    
95.                         {
    
96.                                 strncat(uri, "index.html", sizeof(uri));
    
97.                         }
    
98.                 }
    
99.                 //printf("%s\n", uri);
    
100.                 strncat(fullpath, uri, sizeof(fullpath));
     
101.                 //printf("%s\n", fullpath);
     
102.                
     
103.                 char *out;
     
104.                 struct stat filebuf;
     
105.                
     
106.                 if (lstat(fullpath, &filebuf) < 0)
     
107.                 {
     
108.                         printf("lstat error\n");
     
109.                         out = "HTTP/1.0 404 File Not Found\r\n";
     
110.                         send(newfd, out, strlen(out), 0);
     
111.                         close(newfd);
     
112.                         continue;
     
113.                 }
     
114.                 else if (!S_ISREG(filebuf.st_mode))
     
115.                 {
     
116.                         out = "HTTP/1.0 404 File Not Found\r\n";
     
117.                         send(newfd, out, strlen(out), 0);
     
118.                         close(newfd);
     
119.                         continue;
     
120.                 }
     
121.                
     
122.                 out = "HTTP/1.0 200 OK\r\n";
     
123.                 send(newfd, out, strlen(out), 0);
     
124.                 out = "Server: J/1.23\r\n";
     
125.                 send(newfd, out, strlen(out), 0);
     
126.                 printf("%s\n", fullpath);
     
127.                 for (c=strlen(fullpath); c > 0; c--)
     
128.                 {
     
129.                         if (fullpath[c] == ".")
     
130.                         {
     
131.                                 fullpath++;
     
132.                                 if (strcasecmp(fullpath, "html") == 0)
     
133.                                 {
     
134.                                         out = "Content-Type: text/html\r\n\r\n";
     
135.                                         break;
     
136.                                 }
     
137.                                 if (strcasecmp(fullpath, "jpg") == 0)
     
138.                                 {
     
139.                                         out = "Content-Type: image/jpeg\r\n\r\n";
     
140.                                         break;
     
141.                                 }
     
142.                         }
     
143.                 }
     
144.                 printf("%s\n", out);
     
145.                 send(newfd, out, strlen(out), 0);
     
146.                
     
147.                 char outb[2048];
     
148.                 FILE *fp;
     
149.                 fp = open(fullpath, O_RDONLY);
     
150.                
     
151.                 while ((rv = read(fp, outb, sizeof(outb))) > 0)
     
152.                 {
     
153.                         send(newfd, outb, rv, 0);
     
154.                 }
     
155.                 close(fp);
     
156.                
     
157.                 close(newfd);
     
158.         }
     
159. 
     
160.         return 0;
     
161. }
     
162. 
     
163. #endif /* _SERVER_C_ */
     

复制代码

不飘不浮

up up~~~~~

xinglp

too simple

不飘不浮

i hv no idea about when i should use array for a string or use pointer.. someone can tell me?