#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <pthread.h>
#define TRUE 1
#define BUF_SIZE 1024
#define SMALL_BUF 100
typedef struct sockaddr SA;
void* request_handler(void* arg);
void send_data(FILE* fp, char* ct, char* file_name);
char* content_type(char* file);
void send_error(FILE* fp);
void error_handling(char* message);
int main(int argc, char* argv[])
{
int serv_sock, clnt_sock;
struct sockaddr_in serv_adr, clnt_adr;
int clnt_adr_sz;
char buf[BUF_SIZE] = { 0 };
pthread_t tid;
if (argc != 2) {
printf("Usage: %s <port>\n", argv[0]);
exit(1);
}
serv_sock = socket(PF_INET, SOCK_STREAM, 0);
//为serv_sock套接字文件描述符设置SO_REUSEADDR可选项
int option = TRUE;
int optlen = sizeof(option);
setsockopt(serv_sock, SOL_SOCKET, SO_REUSEADDR, (void*)& option, optlen);
memset(&serv_adr, 0, sizeof(serv_adr));
serv_adr.sin_family = AF_INET;
serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
serv_adr.sin_port = htons(atoi(argv[1]));
if (bind(serv_sock, (SA*)& serv_adr, sizeof(serv_adr)) == -1)
error_handling("bind() error!");
if (listen(serv_sock, 20) == -1)
error_handling("listen() error!");
while (1)
{
clnt_adr_sz = sizeof(clnt_adr);
clnt_sock = accept(serv_sock, (SA*)& clnt_adr, &clnt_adr_sz);
if (clnt_sock == -1) {
error_handling("accept() error!");
continue;
}
printf("Connection Request: %s:%d\n", inet_ntoa(clnt_adr.sin_addr),
ntohs(clnt_adr.sin_port));
pthread_create(&tid, NULL, request_handler, &clnt_sock); //创建线程
pthread_detach(tid); //线程分离
}
close(serv_sock);
return 0;
}
void* request_handler(void* arg)
{
int clnt_sock = *(int*)arg;
char req_line[SMALL_BUF] = { 0 };
FILE* clnt_read, * clnt_write;
char method[10] = { 0 };
char ct[15] = { 0 };
char file_name[30] = { 0 };
clnt_read = fdopen(clnt_sock, "r");
clnt_write = fdopen(dup(clnt_sock), "w");
fgets(req_line, SMALL_BUF, clnt_read);
if (strstr(req_line, "HTTP/") == NULL) //查看是否为HTTP提出的请求
{
send_error(clnt_write);
fclose(clnt_read);
fclose(clnt_write);
return NULL;
}
strcpy(method, strtok(req_line, " /")); //获取请求行中的方法
strcpy(file_name, strtok(NULL, " /")); //获取请求文件名
//printf("file_name: %s\n", file_name);
strcpy(ct, content_type(file_name)); //获取Content-type
if (strcmp(method, "GET") != 0) //查看是否为GET方式请求
{
send_error(clnt_write);
fclose(clnt_read);
fclose(clnt_write);
return NULL;
}
fclose(clnt_read);
send_data(clnt_write, ct, file_name); //响应处理
return NULL;
}
//发送响应消息给客户端
void send_data(FILE* fp, char* ct, char* file_name)
{
char protocol[] = "HTTP/1.1 200 OK\r\n";
char server[] = "Server:Linux Web Server \r\n";
char cnt_len[] = "Content-length:2048\r\n";
char cnt_type[SMALL_BUF] = { 0 };
char buf[BUF_SIZE] = { 0 };
FILE* send_file;
sprintf(cnt_type, "Content-type:%s\r\n\r\n", ct);
send_file = fopen(file_name, "r");
if (send_file == NULL)
{
printf("open file [%s] error!\n", file_name);
send_error(fp);
return;
}
//传输HTTP的状态行+消息头信息
fputs(protocol, fp);
fputs(server, fp);
fputs(cnt_len, fp);
fputs(cnt_type, fp);
//传输HTTP的消息体信息,即HTML文件内容
while (fgets(buf, BUF_SIZE, send_file) != NULL)
{
fputs(buf, fp);
fflush(fp);
}
fflush(fp);
fclose(fp);
}
//区分Content-type
char* content_type(char* file)
{
char extension[SMALL_BUF] = { 0 }; //存放文件扩展名
char file_name[SMALL_BUF] = { 0 }; //存放文件名
strcpy(file_name, file);
strtok(file_name, ".");
strcpy(extension, strtok(NULL, "."));
if (!strcmp(extension, "html") || !strcmp(extension, "htm"))
return "text/html";
else
return "text/plain";
}
void send_error(FILE * fp)
{
char protocol[] = "HTTP/1.1 400 Bad Request\r\n";
char server[] = "Server:Linux Web Server \r\n";
char cnt_len[] = "Content-length:2048\r\n";
char cnt_type[] = "Content-type:text/html\r\n\r\n";
char content[] = "<html>\
<head><title>NETWORK</title></head>\
<body><font size=+5><br>\
发生错误!请检查请求文件名和请求方式!\
</font></body>\
</html>";
//传输HTTP的状态行+消息头信息
fputs(protocol, fp);
fputs(server, fp);
fputs(cnt_len, fp);
fputs(cnt_type, fp);
//传输HTTP的消息体信息,即HTML文件内容
fputs(content, fp);
fflush(fp);
fclose(fp);
}
void error_handling(char* message)
{
fputs(message, stderr);
fputc('\n', stderr);
exit(1);
}