linux 多线程 代码,Linux 多线程服务器源代码详解

对http://tuoxie174.blog.51cto.com/1446064/413189 的源代码分析

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

using namespace std;

#define PORT        8848  //Port

#define BACKLOG     5     // Max Link num

#define MAXDATASIZE 1000  // Max data buf

#define DEBUG 1

//1. Func Decalare

void process_cli(int connectfd, sockaddr_in client);

int  sendobj(int connectfd, char* serverfilepath);

int  isDIR(char* path);

int  fileordirExist(char* fpath);

char* gettextname(char*);

int   writehead(FILE* cfp, char* extname);

void* start_routing(void* arg);

void  msg404(int connectfd);

//2. Structure Decalare

struct ARG

{

int         connfd;

sockaddr_in client;

};

int main()

{

//cout << "Hello world!" << endl;

int listenfd, connectfd;

//! 1. thread id

pthread_t thread;

//! 2. pass this var to the thread

ARG*      arg;

struct sockaddr_in server;

struct sockaddr_in client;

int    sin_size;

//! 3. create tcp socket

#ifdef DEBUG

printf("socket...\n");

#endif

if((listenfd=socket(AF_INET,SOCK_STREAM,0))==-1)

{

perror("creating socket failed!\n");

exit(1);

}

//! 4. set the socket -- close, timeout

int opt = SO_REUSEADDR;

setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

bzero(&server, sizeof(server));

server.sin_family = AF_INET;

server.sin_port   = htons(PORT);

server.sin_addr.s_addr = htonl(INADDR_ANY);

printf("bind...\n");

//! 5. bind a ip and port to the socket

if(bind(listenfd, (struct sockaddr*)&server, sizeof(struct sockaddr)))

{

perror("bind error!");

exit(1);

}

printf("listen...\n");

//! 6. listen the socket

if(listen(listenfd,BACKLOG)==-1)

{

perror("listen error!\n");

exit(1);

}

// a datastructure for programmer to use struct sockaddr

sin_size = sizeof(struct sockaddr_in);

while(1)

{

// begin to process the client's connect

//! 7. accept using main thread

printf("accepting...\n");

if((connectfd=accept(listenfd, (struct sockaddr*)&client, (socklen_t*)&sin_size))==-1)

{

printf("accept error!\n");

}

//connectfd is a socket

arg = new ARG;

arg->connfd = connectfd;

memcpy((void*)&arg->client, &client, sizeof(client));

#ifdef DEBUG

printf("thread creating...\n");

#endif

//! 8. invoke start_routing to handle this thread. -- start_routing is a callback func

if(pthread_create(&thread, NULL, start_routing, (void*)arg))

{

perror("pthread_create error!\n");

exit(1);

}

}

close(listenfd);

return 0;

}

//! 1.1 handle the request

void process_cli(int connectfd, sockaddr_in client)

{

int  num;

char requestline[MAXDATASIZE];

char filepath[MAXDATASIZE];

char cmd[MAXDATASIZE];

char extname[MAXDATASIZE];

int c;

FILE *fp;

//! Because everything is file in UNIX

//! 1.1.1 open the client socket

fp = fdopen(connectfd, "r");

#ifdef DEBUG

printf("the host is: %s \n", inet_ntoa(client.sin_addr));

#endif

fgets(requestline, MAXDATASIZE, fp);

#ifdef DEBUG

printf("The request is: %s \n", requestline);

#endif

strcpy(filepath, "./");

sscanf(requestline, "%s%s\n", cmd, filepath+2);

//! use func gettextname

strcpy(extname, gettextname(filepath));

#ifdef DEBUG

printf("cmd: %s\n filepath: %s\n extname: %s\n", cmd, filepath, extname);

printf("string comparing\n:::::::start:::::::\n");

#endif

if(strcmp(cmd, "GET")==0)

{

#ifdef DEBUG

printf("cmd(%s)==GET\n", cmd);

#endif

if(fileordirExist(filepath))

{

if(isDIR(filepath))

{

//! is a DIR

#ifdef DEBUG

printf("%s is a DIR \n", filepath);

#endif

if(fileordirExist(strcat(filepath, "index.html")))

{

//! send

sendobj(connectfd, "index.html");

}

else

{

msg404(connectfd);

}

}

else

{

//! is a file

#ifdef DEBUG

printf("%s is a file \n", filepath);

#endif

//! send

sendobj(connectfd, filepath);

}

}

else

{

#ifdef DEBUG

printf("404\n");

#endif

msg404(connectfd);

}

}

else

{

#ifdef DEBUG

printf("cmd(%s)!=GET\n", cmd);

#endif

}

#ifdef DEBUG

printf("::::::::::::::::::::end::::::::::::::::::\n");

close(connectfd);

#endif

}

//! 1.2

void msg404(int connectfd)

{

char* msg;

msg = "HTTP/1.0 404 Not Found Content-Type: text/plain 404 not found by Manio\n";

send(connectfd, msg, strlen(msg), 0);

}

//! 1.3 is the filepath a file or direatory

int fileordirExist(char* fpath)

{

struct stat filestat;

return ( stat(fpath, &filestat)!=-1);

}

//! 1.4 is the filepath a directory

int isDIR(char* fpath)

{

#ifdef DEBUG

printf("IN IsDir\ns");

#endif

struct stat filestat;

return ( stat(fpath, &filestat)!=-1 && S_ISDIR(filestat.st_mode));

}

//! 1.5 send the data of the file which the client want

int sendobj(int connectfd, char* serverfilepath)

{

//printf("server: %s\n", serverfilepath)

FILE* sfp=NULL, *cfp=NULL;

char c;

sfp = fopen(serverfilepath, "r");

cfp = fdopen(connectfd, "w");

if(sfp == NULL)

printf("can't open the file!... \n");

if(cfp == NULL)

printf("can't open the link file...!\n");

writehead(cfp, gettextname(serverfilepath));

while( (c=getc(sfp)) !=EOF){

putc(c, cfp);

//putchar(c);

}

fflush(cfp);

return 0;

}

//! 1.6 write the packet header to the client

int writehead(FILE* cfp, char* extname)

{

#ifdef DEBUG

printf("INWRITEHEAD:::::extname is %s::::::\n", extname);

#endif

char* content = "text/plain";

if(strcmp(extname, "html")==0 || strcmp(extname, "htm")==0)

content = "text/html";

else if(strcmp(extname,"css")==0)

content = "text/css";

else if(strcmp(extname,"gif")==0)

content = "p_w_picpath/gif";

else if(strcmp(extname,"jpeg")==0||strcmp(extname,"jpg")==0)

content = "p_w_picpath/jpeg";

else if(strcmp(extname, "png")==0)

content = "p_w_picpath/png";

#ifdef DEBUG

printf("HTTP/1.1 200 OK\n");

printf("Content-Type: %s \n", content);

#endif

fprintf(cfp, "HTTP/1.1 200 OK ");

fprintf(cfp, "Content-Type: %s \n", content);

return 0;

}

//! 1.7 get the extent name of the file

char* gettextname(char* filepath)

{

char* p;

if((p=strrchr(filepath,'.'))!=NULL)

return p+1;

return NULL;

}

//! 1.8 invoked by pthread_create

void* start_routing(void* arg)

{

ARG* info;

info = (ARG*) arg;

// handle client's requirement

process_cli(info->connfd, info->client);

delete arg;

pthread_exit(NULL);

}

//1.  只要 g++ -g -o execname filename -lpthread

//2. 然后在浏览器输入： 127.0.0.1：8848

//3. 与程序同目录下 有 index.html (注意扩展名， 程序中为判断)

问题： 中文会出现乱码， 查看源代码， 发现

的很大一部分没有了， 只留下后面的， 于是乎编码没有设置。 (英文网页时，前半部分也没了)。 继续解决。