#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <pthread.h>
#define BUF_SIZE 1024
#define SMALL_BUF 100
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;//create sev and clnt sock;
struct sockaddr_in serv_adr,clnt_adr;
int clnt_adr_size;
char buf[BUF_SIZE];
pthread_t t_id;//create thread id
if(argc!=2)
{
printf("Usage :%s <port>\n",argv[0]);
exit(1);
}
serv_sock = socket(PF_INET,SOCK_STREAM,0);//create
memset(&serv_adr,0,sizeof(serv_adr));//init the srv addr
serv_adr.sin_family = AF_INET;//协议地址族,v4
serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);//字节序转换
serv_adr.sin_port = htons(atoi(argv[1]));//端口指定
if(bind(serv_sock,(struct sockaddr*)&serv_adr,sizeof(serv_adr))==-1)
error_handling("bind() error!");//地址&&端口绑定
if(listen(serv_sock,20)==-1)
error_handling("listen() error!");//监听请求
//监听队列大小为20,这时候相当于创建了服务器套接字(充当守门员)
while(1)
{
clnt_adr_size = sizeof(clnt_adr);
clnt_sock = accept(serv_sock,(struct sockaddr*)&clnt_adr,&clnt_adr_size);
//创建请求处理套接字
printf("Connection Request :%s:%d\n",
inet_ntoa(clnt_adr.sin_addr),ntohs(clnt_adr.sin_port));
//创建子线程
pthread_create(&t_id,NULL,request_handler,&clnt_sock);
//线程id为t_id,NULL表示创建默认属性,传递线程的thread_main函数,最后一个
//参数是线程main函数的参数
pthread_detach(t_id);//销毁子线程
}
close(serv_sock);//关闭套接字
return 0;
}
//请求套接字的处理
void* request_handler(void* arg)
{
int clnt_sock = *((int *)arg);
char read_line[SMALL_BUF];
FILE* clnt_read;
FILE* clnt_write;
char method[10];
char ct[15];
char file_name[30];
clnt_read = fdopen(clnt_sock,"r");//将文件描述符转换成文件指针
clnt_write = fdopen(dup(clnt_sock),"w");
//为了实现半关闭状态,这里复制套接字,实现较好的IO分流
fgets(read_line,SMALL_BUF,clnt_read);
//读取可读套接字缓冲区中的数据
if(strstr(read_line,"HTTP/")==NULL)
{
send_error(clnt_write);
fclose(clnt_write);
fclose(clnt_read);
//return;
//exit(-1);
}
strcpy(method,strtok(read_line," /"));
strcpy(file_name,strtok(NULL," /"));
strcpy(ct,content_type(file_name));
if(strcmp(method,"GET")!=0)
{
send_error(clnt_write);
fclose(clnt_read);
fclose(clnt_write);
//return;
//exit(-1);
}
fclose(clnt_read);
send_data(clnt_write,ct,file_name);
//return ;
}
void send_data(FILE* fp,char* ct,char *file_name)
{
char protocol[] = "HTTP/1.0 200 ok\r\n";
char server[] = "Server:Linux web Server\r\n";
char clnt_len[] = "Content-length:2048";
char clnt_type[SMALL_BUF];
char buf[BUF_SIZE];
FILE* send_file;
sprintf(clnt_type,"Content type :%s\r\n\r\n",ct);
send_file = fopen(file_name,"r");
if(send_file==NULL)
{
send_error(fp);
return ;
}
//传输头信息
fputs(protocol,fp);
fputs(server,fp);
fputs(clnt_len,fp);
fputs(clnt_type,fp);
//传输请求数据
while(fgets(buf,BUF_SIZE,send_file)!=NULL)
{
fputs(buf,fp);
fflush(fp);
}
fflush(fp);
fclose(fp);
}
char* content_type(char* file)
{
char extension[SMALL_BUF];
char file_name[SMALL_BUF];
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.0 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></title>NETWORK</head>"
"<body><font size=+5><br>发生错误!查看请求文件名和请求方式!"
"</font></body></html>";
fputs(protocol,fp);
fputs(server,fp);
fputs(cnt_len,fp);
fputs(cnt_type,fp);
fflush(fp);
}
void error_handling(char *message)
{
fputs(message,stderr);
fputc('\n',stderr);
exit(1);
}
基于Linux的多线程Web Server
最新推荐文章于 2022-02-16 21:38:30 发布