main.c

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <errno.h>
#include <netinet/in.h>
#include <string.h>
#include <stdlib.h>
#include <signal.h>

#include "threadpool.h"

struct threadpool *gThreadPool;

/* 
 * ===  FUNCTION  ======================================================================
 *         Name:  get_listen_socket
 *  Description:  创建监听socket
 * =====================================================================================
 */
int get_listen_socket()
{
	int sock,res;
	int reuse = 1;
	struct sockaddr_in addr;
	sock = socket(AF_INET,SOCK_STREAM,0);

	if(0 >= sock)
	{
		perror("socket error!");
		exit(0);
	}

	bzero(&addr, sizeof(addr));
	addr.sin_addr.s_addr = inet_addr("0.0.0.0");
	addr.sin_port = htons(51005);
	addr.sin_family = AF_INET;

	res = setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse));
	if(res)
	{
		perror("setsockopt error!");
		exit(0);
	}

	res = bind(sock,(struct sockaddr *)&addr,sizeof(struct sockaddr));
	if(res)
	{
		perror("bind error!");
		exit(0);
	}

	res = listen(sock,10);
	if(res)
	{
		perror("listen error!");
		exit(0);
	}

	return sock;
}

/* 
 * ===  FUNCTION  ======================================================================
 *         Name:  do_job
 *  Description:  业务处理函数，client：客户端socket。
 * =====================================================================================
 */
void do_job(client)
{
	char buf[1024];
	int len;
	while(1)
	{
		len = read(client,buf,sizeof(buf));
		if(len <= 0)
		{
			perror("read error!");
			break;
		}
		if(!memcmp("exit",buf,len))
		{
			send(client,"GoodBye",7,0);
			break;
		}
		len = send(client,buf,len,0);
		if(len <= 0)
		{
			perror("send error!");
			break;
		}
	}
	close(client);
}

/* 
 * ===  FUNCTION  ======================================================================
 *         Name:  work
 *  Description:  领导者/追随者模式实例函数，arg:监听socket。
 * =====================================================================================
 */
void *work(void *arg)
{
	int *listensock = (int *)arg;
	int client;
	struct sockaddr_in remote;
	socklen_t len;
	
	client = accept(*listensock,(struct sockaddr *)&remote,&len);
	if(client < 0)
	{
		perror("accept error!");
		return;
	}

	threadpool_add_job( gThreadPool, work, (void *)listensock );

	do_job(client);
}

int main(int argc, char *argv[])
{
	int *listensock;

	signal(SIGPIPE, SIG_IGN);

	listensock = (int *)malloc(sizeof(int));
	*listensock = get_listen_socket();
	gThreadPool = threadpool_init(6,20);
	
	//work(listensock);
	threadpool_add_job( gThreadPool, work, (void *)listensock );

	while(1)
	{
		sleep(10);
	}

	return 0;
}