UNIX网络变成-套接字联网之TCP客户/服务器程序学习

最近开始学习UNIX网络编程，参考书是大师W.Richard Stevens的著作《UNIX网络编程-卷1：套接字联网API》。首先，阅读本书必须有一定的UNIX或者LINUX下C编程的基础知识。

首先，看一个linux下的C/S程序，来了解以下UNIX环境下套接字编程到底是怎么一回事。

/===================================================服务器程序=============================================================/

/* File Name: server.c */

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <errno.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <netinet/in.h>

#include <signal.h>

/*********************************************宏定义*******************************************************/

#define DEFAULT_PORT 8000

#define MAXLINE 4096

/*********************************************函数声明*******************************************************/

void sig_child(int signo);

/*********************************************函数定义*******************************************************/

int main(int argc, char** argv)

{

int socket_fd, connect_fd;

struct sockaddr_in servaddr;

char buff[4096];

int n;

//初始化Socket

if( (socket_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1 ){

printf("create socket error: %s(errno: %d)\n",strerror(errno),errno);

exit(0);

}

//初始化

memset(&servaddr, 0, sizeof(servaddr));

servaddr.sin_family = AF_INET;

servaddr.sin_addr.s_addr = htonl(INADDR_ANY);//IP地址设置成INADDR_ANY,让系统自动获取本机的IP地址。

servaddr.sin_port = htons(DEFAULT_PORT);//设置的端口为DEFAULT_PORT

//将本地地址绑定到所创建的套接字上

if( bind(socket_fd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1){

printf("bind socket error: %s(errno: %d)\n",strerror(errno),errno);

exit(0);

}

//开始监听是否有客户端连接,并立即处理为客户端建立的子进程的状态

signal(SIGCHLD, sig_child);// 防止子进程变为僵尸进程

if( listen(socket_fd, 10) == -1){

printf("listen socket error: %s(errno: %d)\n",strerror(errno),errno);

exit(0);

}

printf("======waiting for client's request======\n");

while(1){

//阻塞直到有客户端连接，不然多浪费CPU资源。

if((connect_fd = accept(socket_fd, (struct sockaddr*)NULL, NULL)) < 0){

if(errno == EINTR)

continue; /*back ro for() */

else

printf("accept socket error: %s(errno: %d)",strerror(errno),errno);

continue;

}

//接受客户端传过来的数据

n = recv(connect_fd, buff, MAXLINE, 0);

//向客户端发送回应数据

if(!fork()){ /* 子进程 */

if(send(connect_fd, "Hello,you are connected!\n", 26,0) == -1)

perror("send error");

close(connect_fd);

exit(0);

}

buff
= '\0';

printf("recv msg from client: %s\n", buff);

close(connect_fd);

}

close(socket_fd);

}

//信号处理函数，用于处理服务器为客户端建立的子进程的状态信息

void sig_child(int signo)

{

pid_t pid;

int stat;

//pid = wait(&stat);

// WNOHANG,如果此时还有未结束的子进程，选择不阻塞

while ((pid = waitpid(-1,&stat,WNOHANG)) > 0)

printf("child %d terminated\n",pid);

return;

}

/****************************************************************************以下是客户端程序*******************************************************************************/

/* File Name: client.c */

#include<stdio.h>

#include<stdlib.h>

#include<string.h>

#include<errno.h>

#include<sys/types.h>

#include<sys/socket.h>

#include<netinet/in.h>

/**************************************************函数定义*****************************************/

#define MAXLINE 4096

#define max(a,b) (((a) > (b))? (a):(b))

#define min(a,b) (((a) < (b))? (a):(b))

/**************************************************函数声明*****************************************/

void str_cli(FILE *fp, int sockfd);

/**************************************************函数定义*****************************************/

int main(int argc, char** argv)

{

int sockfd, n,rec_len;

char recvline[4096], sendline[4096];

char buf[MAXLINE];

struct sockaddr_in servaddr;

// 提示，如果没有执行参数，则报出使用方法

if( argc != 2){

printf("usage: ./client <ipaddress>\n");

exit(0);

}

// 建立SOCKET套接字

if( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0){

printf("create socket error: %s(errno: %d)\n", strerror(errno),errno);

exit(0);

}

//

memset(&servaddr, 0, sizeof(servaddr));

servaddr.sin_family = AF_INET;

servaddr.sin_port = htons(8000);

if( inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <= 0){

printf("inet_pton error for %s\n",argv[1]);

exit(0);

}

if( connect(sockfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0){

printf("connect error: %s(errno: %d)\n",strerror(errno),errno);

exit(0);

}

//printf("send msg to server: \n");

//fgets(sendline, 4096, stdin);

//if( send(sockfd, sendline, strlen(sendline), 0) < 0)

//{

// printf("send msg error: %s(errno: %d)\n", strerror(errno), errno);

//exit(0);

//}

//if((rec_len = recv(sockfd, buf, MAXLINE,0)) == -1) {

// perror("recv error");

// exit(1);

//}

//buf[rec_len] = '\0';

//printf("Received : %s ",buf);

//close(sockfd);

str_cli(stdin,sockfd);

exit(0);

}

//

void str_cli(FILE *fp, int sockfd)

{

int maxfdp1,stdineof;

fd_set rset;

//char senline[MAXLINE], recvline[MAXLINE];

char buf[MAXLINE];

int n;

stdineof = 0;

FD_ZERO(&rset);

while(1){

if(stdineof == 0)

FD_SET(fileno(fp),&rset);

//FD_SET(fileno(fp), &rset); // 设置文本描述符读集

FD_SET(sockfd, &rset);

maxfdp1 = max(fileno(fp),sockfd) + 1;

select(maxfdp1, &rset, NULL, NULL,NULL);

// 套接字是可读的

if (FD_ISSET(sockfd, &rset)){/* socket is readble */

if ((n = read(sockfd, buf,MAXLINE)) == 0){

if(stdineof == 1)

return;

else{

printf("str_cli: server terminated prematurely");

exit(1);

}

}

//fputs(recvline,stdout);

write(fileno(stdout),buf,n);

}

// 输入是可读的

if (FD_ISSET(fileno(fp), &rset)){

if((n = read(fileno(fp),buf,MAXLINE)) == 0){

stdineof = 1;

shutdown(sockfd,SHUT_WR);//发送FIN

FD_CLR(fileno(fp),&rset);

continue;

}

write(sockfd,buf,n);

}

}

}