internet_srv.c

linux系统下的socket源码

/*
 * server.c
 *  Created on: 2008-10-18
 *      Author: xunzhao
 * internet domain 通信，服务器端
*/

#include <stdio.h> 
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>
#include <errno.h>
#include <signal.h>
#include <sys/wait.h>
#include <netdb.h>
#include <stdlib.h>
#include <ctype.h>
#include <arpa/inet.h>
#include <time.h>
#include "usock.h"

#define port 5120  //服务器监听端口

//捕获子进程退出信号，在退出时给出提示信息
void sig_handler(int signo){
	pid_t pid;
	int stat;

	pid = waitpid(-1, &stat, WNOHANG);
	while(pid > 0){
		printf("Child process terminated (PID: %ld) \n",(long)getpid());
		pid = waitpid(-1, &stat, WNOHANG);
	}

	return ;
}


int main(int argc, char *argv[])
{
	int listen_fd,com_fd,ret,on,i;
	socklen_t len;
	pid_t pid;
	time_t timep;
	struct tm *echo_time;	
	char username[30];
	static char snd_buf[1024];
	static char recv_buf[1024];

	//struct sockaddr_in clt_addr;
	//struct sockaddr_in srv_addr;
	struct usock clt_addr;  //usock是我自定义的一个结构体，把原来的sockaddr_in结构体作为usock的一个字段
	struct usock srv_addr;  //并增加一个用户名字段,
							//我的初衷是实现用户第一次连接服务器时就自动发送用户名
	time(&timep);
	echo_time=localtime(&timep);  //获取当地时间

	//参数验证
	if(argc != 1){
		printf("Usage: %s \n", argv[0]);
		return 1;
	}

	//设置信号处理函数，也可以使用sigaction函数
	if(signal(SIGCHLD, sig_handler) < 0 ){
		perror("Cann't set the signal \n");
		return 1;
	}
	
	//创建套接字用于服务器的监听
	listen_fd = socket(AF_INET,SOCK_STREAM,0);
	if(listen_fd < 0){
		perror("Cann't creat listening socket!\n");
		return 1;
	}

	on = 1;//实现地址/端口重用，避免出现Address already in use的错误
	ret = setsockopt(listen_fd,SOL_SOCKET,SO_REUSEADDR, &on, sizeof(on) );

	//设置服务器地址参数
	memset(&srv_addr, 0, sizeof(srv_addr));
	srv_addr.sockaddr.sin_family = AF_INET;
	srv_addr.sockaddr.sin_addr.s_addr = htonl(INADDR_ANY);//填入本机IP地址
	srv_addr.sockaddr.sin_port = htons(port);//

	//绑定套接字与服务器地址信息
	//ret = bind(listen_fd,(struct sockaddr*)&srv_addr, sizeof(srv_addr));
	ret=bind(listen_fd,(struct sockaddr*)&srv_addr.sockaddr,sizeof(srv_addr));
	if(ret == -1 ){
		perror("Cann't bind server socket ! \n");
		close(listen_fd);
		return 1;
	}
	//监听指定端口，连接5个客户端
	ret = listen (listen_fd,5);
	if (ret == -1){
		perror("Cann't listen the client connect request! \n");
		close(listen_fd);
		return 1;
	} else 
		puts("服务器启动成功，等待用户登录！\n");

	//对每个连接来的客户端创建一个子进程，单独与其进行通信
	//首先调用read函数读取客户端发送过来的信息，
	//将其转换成大写后发送回客户端
	//当输入"@"时，程序退出
	while(1){
	len = sizeof(clt_addr.sockaddr);
	//printf("This is server ,length of clt_addr is %d \n",len);
	com_fd = accept(listen_fd,(struct sockaddr*)&clt_addr.sockaddr, &len);
	if(com_fd > 0){
		//printf("This is server,com_fd is %d \n",com_fd);
		recv(com_fd,username,30,0);//这是我目前采用的获取用户名的方式，客户端连接成功后主动send当前用户名到服务端
									//我自定义的usock没能起到预期的作用
		}
	if(com_fd < 0){
		if(errno == EINTR){
			//printf("EINTR!\n");
			continue;
		}else{
			perror("Cann't accept client connect request !\n");
			close(listen_fd);
			return 1;
		}
	}
	
	time(&timep);
	echo_time=localtime(&timep);  //获取当地时间
	printf("用户%s(%s:%d)于%d:%d:%d 登录\n",username,inet_ntoa(clt_addr.sockaddr.sin_addr),ntohs(clt_addr.sockaddr.sin_port),echo_time->tm_hour,echo_time->tm_min,echo_time->tm_sec);
	
	pid = fork();//创建子进程，我想在此获得子进程的socket描述符并保存成一个列表，以实现客户端之间的互相通
	//printf("after fork,the pid is %d",pid);
	if (pid < 0 ){
		perror("Cannot creat the child process \n");
		close(listen_fd);
		return 1;
	}else if ( pid == 0 ){
		//清空缓存并从客户端读取数据
		memset(recv_buf,0,1024);
		while((len = recv(com_fd,recv_buf,1024,0)) > 0 ) {
			recv_buf[len] = '\0';
			//printf("Message from com_fd: %d\n",com_fd);//这里我的初衷是输入出子进程的socket描述符，但实际上一直输出"4"
			time(&timep);
			echo_time=localtime(&timep);  //获取当地时间
			printf("%s (%d:%d:%d) 说:\n%s",username,echo_time->tm_hour,echo_time->tm_min,echo_time->tm_sec,recv_buf);
			if(recv_buf[0] == '@' ){
				printf("用户%s退出!\n",username);
				break;
			}
			//数据自动处理，进行大小写转换后发送给客户端，也可以设置为服务器端的用户输入，实现聊天
			for (i=0; i<len; i++)
				snd_buf[i] = toupper(recv_buf[i]);
				snd_buf[len] = '\0';
			//发送数据
			send(com_fd,snd_buf,len,0); 
		/*write(STDOUT_FILENO,"Server:",8);
		//从键盘读取用户输入的数据，并写入发送缓存snd_buf
		memset(snd_buf,0,1024);
		len=read(STDIN_FILENO,snd_buf,1024);
		snd_buf[len] = '\0';
		send(com_fd,snd_buf,len,0);
		//这段代码可以实现服务端从键盘读取数据，并发送给客户端，这样就可以自由聊天了
		*/
		}
	close(com_fd);
	return 0;
	}else
		close(com_fd);
	}

return 0 ;
}