TCP 回射程序（處理子進程的SIGCHLD信號）

原創 juzimeizi 2018-09-03 16:47

客戶端的代碼；

/*************************************************************************
	> File Name: tcpcli.c
	> Author: mhsheng
	> Mail:[email protected] 
	> Created Time: Wed 09 Nov 2016 10:55:40 AM CST
 ************************************************************************/

// c standard head file
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <errno.h>

// system i/o head file
#include <unistd.h>

// socket head file
#include <netinet/in.h>

#define MAXLINE		1024
#define LISTENQ		5
#define SERV_PORT	9066	

void
err_quit(const char *str_err)
{
	fprintf(stderr, "%s\n", str_err);
	exit(-1);
}

void
err_sys(const char *str_err)
{
	fprintf(stderr, "%s\n", str_err);
	exit(-1);
}

int
Socket(int faimly, int type, int protocol)
{
	int sockfd;
	if ( (sockfd = socket(faimly, type, protocol)) < 0)
		err_sys("socket error");

	return sockfd;
}

void
Connect(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen)
{
	if (connect(sockfd, myaddr, addrlen) < 0)
		err_sys("connect error");
}

void 
Bind(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen)
{
	if (bind(sockfd, myaddr, addrlen) < 0)
		err_sys("bind error");
}

void 
Listen(int sockfd, int backlog)
{
	if (listen(sockfd, backlog) < 0)
		err_sys("listen error");
}

int
Accept(int sockfd, struct sockaddr *cliaddr, socklen_t *addrlen)
{
	int connfd;
	
	connfd = accept(sockfd, cliaddr, addrlen);
	if (connfd < 0)
		err_sys("accept error");

	return connfd;
}

void
Close(int sockfd)
{
	if (close(sockfd) < 0)
		err_sys("close error");
}

//////////////////////////////////////////////////////////////////////////
// Writen begin 
//

ssize_t
writen(int fd, const void *vptr, size_t n)
{
	size_t		nleft;
	ssize_t		nwritten;
	const char	*ptr;

	ptr		= vptr;
	nleft	= n;
	while (nleft > 0) {
		if ( (nwritten = write(fd, ptr, nleft)) <= 0) {
			if (nwritten < 0 && errno == EINTR)
				nwritten = 0;		/* and call write() again */
			else
				return (-1);
		}
		
		nleft	-= nwritten;
		ptr		+= nwritten;
	}

	return (n);
}

void
Writen(int fd, void *ptr, size_t nbytes)
{
	if (writen(fd, ptr, nbytes) != nbytes)
		err_sys("writen error");
}

// 
// Writen end 
//////////////////////////////////////////////////////////////////////////

//////////////////////////////////////////////////////////////////////////
// Readline begin
//

static int	read_cnt;
static char	*read_ptr;
static char	read_buf[MAXLINE];

static	ssize_t
my_read(int fd, char *ptr)
{
	if (read_cnt <= 0) {
		for ( ; ; ) {
			if ( (read_cnt = read(fd, read_buf, sizeof(read_buf))) < 0) {
				if (errno == EINTR) 
					continue;
				return -1;
			} else if (read_cnt == 0)
				return 0;

			read_ptr = read_buf;
			break;
		}
	}

	read_cnt--;
	*ptr = *read_ptr++;
	return 1;
}

ssize_t
readline(int fd, void *vptr, size_t maxlen)
{
	ssize_t	n;
	ssize_t	rc;
	char	c;
	char	*ptr;

	ptr = vptr;
	for (n=1; n<maxlen; n++) {
		if ( (rc = my_read(fd, &c)) == 1) {
			*ptr++ = c;
			if (c == '\n')
				break;
		} else if (rc == 0) {
			*ptr = 0;		/* EOF, n - 1 bytes were read */
			return n-1;
		} else {
			return -1;
		}
	}

	*ptr = 0;
	return n;
}

ssize_t
Readline(int fd, void *ptr, size_t maxlen)
{
	ssize_t	n;

	if ( (n = readline(fd, ptr, maxlen)) < 0)
		err_sys("readline error");
	return n;
}

//
// Readline end
//////////////////////////////////////////////////////////////////////////

//////////////////////////////////////////////////////////////////////////
// wrapstdio begin
// 

// Fgets begin
char *
Fgets(char *ptr, int n, FILE *stream)
{
	char *rptr;

	if ( (rptr = fgets(ptr, n, stream)) == NULL && ferror(stream))
		err_sys("fgets error");

	return rptr;
}
// Fgets end

// Fputs begin
void
Fputs(const char *ptr, FILE *stream)
{
	if (fputs(ptr, stream) == EOF)
		err_sys("fputs error");
}
// Fputs end

// 
// wrapstdio end
//////////////////////////////////////////////////////////////////////////


//////////////////////////////////////////////////////////////////////////
// str_echo begin
//

void
str_echo(int sockfd)
{
	char	buf[MAXLINE];
	ssize_t	n;

	for ( ; ; ) {
		// 從套接字讀取數據
		while ( (n = read(sockfd, buf, MAXLINE)) > 0)
			// 將讀取的數據發送給客戶端
			Writen(sockfd, buf, n);

		if (n < 0 && errno == EINTR)
			continue;
		else if (n < 0)
			err_sys("str_echo: read error");
		else
			break;
	}
}

//
// str_echo end
//////////////////////////////////////////////////////////////////////////

//////////////////////////////////////////////////////////////////////////
// str_cli begin
//

void
str_cli(FILE *fp, int sockfd)
{
	char	sendline[MAXLINE];
	char	recvline[MAXLINE];

	while (Fgets(sendline, MAXLINE, fp) != NULL) {
		Writen(sockfd, sendline, strlen(sendline));

		if (Readline(sockfd, recvline, MAXLINE) == 0)
			err_quit("str_cli: server terminated prematurely");

		Fputs(recvline, stdout);
	}
}

//
// str_cli end
//////////////////////////////////////////////////////////////////////////


int
main(int argc, char **argv)
{
	int					sockfd;
	struct sockaddr_in	servaddr;

	if (argc != 2)
		err_quit("usage: tcpcli <IPaddress>");

	// 申請一個套接字
	sockfd = Socket(AF_INET, SOCK_STREAM, 0);

	// 設置一個服務器套接字
	memset(&servaddr, 0, sizeof(servaddr));
	servaddr.sin_family	= AF_INET;
	servaddr.sin_port	= htons(SERV_PORT);
	inet_pton(AF_INET, argv[1], &servaddr.sin_addr);

	Connect(sockfd, (struct sockaddr *) &servaddr, sizeof(servaddr));

	str_cli(stdin, sockfd);

	exit(0);
}



服務器端的代碼：

/*************************************************************************
	> File Name: tcpserv.c
	> Author: mhsheng
	> Mail:[email protected] 
	> Created Time: Mon 14 Nov 2016 10:35:23 AM CST
 ************************************************************************/

// c standard head file
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <errno.h>
#include <signal.h>

// system i/o head file
#include <unistd.h>

// socket head file
#include <netinet/in.h>

#define MAXLINE		1024
#define LISTENQ		5
#define SERV_PORT	9066	

void
err_quit(const char *str_err)
{
	fprintf(stderr, "%s\n", str_err);
	exit(-1);
}

void
err_sys(const char *str_err)
{
	fprintf(stderr, "%s\n", str_err);
	exit(-1);
}

int
Socket(int faimly, int type, int protocol)
{
	int sockfd;
	if ( (sockfd = socket(faimly, type, protocol)) < 0)
		err_sys("socket error");

	return sockfd;
}

void
Connect(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen)
{
	if (connect(sockfd, myaddr, addrlen) < 0)
		err_sys("connect error");
}

void 
Bind(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen)
{
	if (bind(sockfd, myaddr, addrlen) < 0)
		err_sys("bind error");
}

void 
Listen(int sockfd, int backlog)
{
	if (listen(sockfd, backlog) < 0)
		err_sys("listen error");
}

int
Accept(int sockfd, struct sockaddr *cliaddr, socklen_t *addrlen)
{
	int connfd;
	
	for ( ; ; ) {
		if ( (connfd = accept(sockfd, cliaddr, addrlen)) < 0) {
			if (errno == EINTR)			// 信號中斷引起的錯誤
				continue;
			else
				err_sys("accept error");
		}
		break;
	}

	return connfd;
}

void
Close(int sockfd)
{
	if (close(sockfd) < 0)
		err_sys("close error");
}

//////////////////////////////////////////////////////////////////////////
// Writen begin 
//

ssize_t
writen(int fd, const void *vptr, size_t n)
{
	size_t		nleft;
	ssize_t		nwritten;
	const char	*ptr;

	ptr		= vptr;
	nleft	= n;
	while (nleft > 0) {
		if ( (nwritten = write(fd, ptr, nleft)) <= 0) {
			if (nwritten < 0 && errno == EINTR)
				nwritten = 0;		/* and call write() again */
			else
				return (-1);
		}
		
		nleft	-= nwritten;
		ptr		+= nwritten;
	}

	return (n);
}

void
Writen(int fd, void *ptr, size_t nbytes)
{
	if (writen(fd, ptr, nbytes) != nbytes)
		err_sys("writen error");
}

// 
// Writen end 
//////////////////////////////////////////////////////////////////////////

//////////////////////////////////////////////////////////////////////////
// Readline begin
//

static int	read_cnt;
static char	*read_ptr;
static char	read_buf[MAXLINE];

static	ssize_t
my_read(int fd, char *ptr)
{
	if (read_cnt <= 0) {
		for ( ; ; ) {
			if ( (read_cnt = read(fd, read_buf, sizeof(read_buf))) < 0) {
				if (errno == EINTR) 
					continue;
				return -1;
			} else if (read_cnt == 0)
				return 0;

			read_ptr = read_buf;
			break;
		}
	}

	read_cnt--;
	*ptr = *read_ptr++;
	return 1;
}

ssize_t
readline(int fd, void *vptr, size_t maxlen)
{
	ssize_t	n;
	ssize_t	rc;
	char	c;
	char	*ptr;

	ptr = vptr;
	for (n=1; n<maxlen; n++) {
		if ( (rc = my_read(fd, &c)) == 1) {
			*ptr++ = c;
			if (c == '\n')
				break;
		} else if (rc == 0) {
			*ptr = 0;		/* EOF, n - 1 bytes were read */
			return n-1;
		} else {
			return -1;
		}
	}

	*ptr = 0;
	return n;
}

ssize_t
Readline(int fd, void *ptr, size_t maxlen)
{
	ssize_t	n;

	if ( (n = readline(fd, ptr, maxlen)) < 0)
		err_sys("readline error");
	return n;
}

//
// Readline end
//////////////////////////////////////////////////////////////////////////

//////////////////////////////////////////////////////////////////////////
// wrapstdio begin
// 

// Fgets begin
char *
Fgets(char *ptr, int n, FILE *stream)
{
	char *rptr;

	if ( (rptr = fgets(ptr, n, stream)) == NULL && ferror(stream))
		err_sys("fgets error");

	return rptr;
}
// Fgets end

// Fputs begin
void
Fputs(const char *ptr, FILE *stream)
{
	if (fputs(ptr, stream) == EOF)
		err_sys("fputs error");
}
// Fputs end

// 
// wrapstdio end
//////////////////////////////////////////////////////////////////////////


//////////////////////////////////////////////////////////////////////////
// str_echo begin
//

void
str_echo(int sockfd)
{
	char	buf[MAXLINE];
	ssize_t	n;

	for ( ; ; ) {
		// 從套接字讀取數據
		while ( (n = read(sockfd, buf, MAXLINE)) > 0)
			// 將讀取的數據發送給客戶端
			Writen(sockfd, buf, n);

		if (n < 0 && errno == EINTR)
			continue;
		else if (n < 0)
			err_sys("str_echo: read error");
		else
			break;
	}
}

//
// str_echo end
//////////////////////////////////////////////////////////////////////////

//////////////////////////////////////////////////////////////////////////
// str_cli begin
//

void
str_cli(FILE *fp, int sockfd)
{
	char	sendline[MAXLINE];
	char	recvline[MAXLINE];

	while (Fgets(sendline, MAXLINE, fp) != NULL) {
		Writen(sockfd, sendline, strlen(sendline));

		if (Readline(sockfd, recvline, MAXLINE) == 0)
			err_quit("str_cli: server terminated prematurely");

		Fputs(recvline, stdout);
	}
}

//
// str_cli end
//////////////////////////////////////////////////////////////////////////


//////////////////////////////////////////////////////////////////////////
// signal begin
//
typedef void Sigfunc(int); 

Sigfunc *
signal(int signo, Sigfunc *func)
{
	struct sigaction	act;
	struct sigaction	oact;

	act.sa_handler = func;
	sigemptyset(&act.sa_mask);
	act.sa_flags = 0;
	
	if (signo == SIGALRM) {
#ifdef SA_INTERRUPT
		act.sa_flags |= SA_INTERRUPT;
#endif
	} else {
#ifdef SA_RESTART
		act.sa_flags |= SA_RESTART;
#endif
	}


	if (sigaction(signo, &act, &oact) < 0)
		return SIG_ERR;

	return oact.sa_handler;
}

Sigfunc *
Signal(int signo, Sigfunc *func)
{
	Sigfunc *sigfunc;

	if ( (sigfunc = signal(signo, func)) == SIG_ERR) 
		err_sys("signal error");

	return sigfunc;
}

//
// signal end
//////////////////////////////////////////////////////////////////////////


//////////////////////////////////////////////////////////////////////////
// sig_chld begin
//

void
sig_chld(int signo)
{	
	pid_t	pid;
	int		stat;

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

	return ;
}

//
// sig_chld end
//////////////////////////////////////////////////////////////////////////


int
main(int argc, char **argv)
{
	int					listenfd;
	int					connfd;
	pid_t				pid;
	struct sockaddr_in	servaddr;
	
	void sig_chld(int);

	// 創建一個監聽套接字描述符
	listenfd = Socket(AF_INET, SOCK_STREAM, 0);

	// 設置服務器的監聽套接字
	memset(&servaddr, 0, sizeof(servaddr));
	servaddr.sin_family			= AF_INET;
	servaddr.sin_port			= htons(SERV_PORT);
	servaddr.sin_addr.s_addr	= htonl(INADDR_ANY);
	
	// 綁定套接字到套接字描述符
	Bind(listenfd, (struct sockaddr *) & servaddr, sizeof(servaddr));

	// 啓動監聽套接字
	Listen(listenfd, LISTENQ);	

	// 處理子進程的信號 SIGCHLD 
	Signal(SIGCHLD, sig_chld);

	for ( ; ; ) {
		connfd = Accept(listenfd, (struct sockaddr *)NULL, NULL);

		if ( (pid = fork()) == 0) {
			Close(listenfd);

			str_echo(connfd);

			exit(0);
		}

		Close(connfd);
	}

	exit(0);
}




Makefile文件：

PROGS	= tcpserv \
		  tcpcli

OBJS	= tcpserv.o \
		  tcpcli.o

All:	${PROGS}

tcpserv:	tcpserv.o
	cc -o tcpserv tcpserv.o

tcpcli:		tcpcli.o
	cc -o tcpcli tcpcli.o

clean:
	rm $(PROGS) $(OBJS)


發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章

netstat 基本用法

常用的命令形式： netstat -anlpt 轉自：https://linux.cn/article-2434-1.html Netstat 簡介 Netstat 是一款命令行工具，可用於列出系統上所有的網絡套接字連接情況，包括 tc

juzimeizi 2020-07-02 12:32:09

Socket編程實踐(13) --UNIX域協議

UNIX域協議    UNIX域套接字與TCP相比, 在同一臺主機上, UNIX域套接字更有效率, 幾乎是TCP的兩倍(由於UNIX域套接字不需要經過網絡協議棧,不需要打包/拆包,計算校驗和,維護序號和應答等,只是將應用層數據從一個進程

acp小鸡炖蘑菇 2020-06-30 22:45:45

Socket編程實踐(5) --TCP粘包問題與解決

TCP粘包問題 由於TCP協議是基於字節流且無邊界的傳輸協議, 因此很有可能產生粘包問題, 問題描述如下    對於Host A 發送的M1與M2兩個各10K的數據塊, Host B 接收數據的方式不確定, 有以下方式接收:

acp小鸡炖蘑菇 2020-06-30 22:45:45

Linux 下網絡相關命令

1. ifconfig 查看或臨時配置當先生效的 IP 地址和子網掩碼 2. 禁用和啓用網卡 禁用網卡 ifdown 網卡設備名 啓用網卡 ifup 網卡設備名 3. 查看網絡狀態 netstat 選項 -t 列出 TCP

王慧勇a 2020-06-28 20:52:05

VMware 虛擬機下網絡參數配置

1. 設置 IP 地址 使用 setup 工具或修改網絡配置文件。 網卡信息文件 /etc/sysconfig/network-scripts/ifcfg-eth0 主機名文件 /etc/sysconfig/network D

王慧勇a 2020-06-28 20:52:04

《網絡編程》線程

線程基本函數         當一個程序被啓動時，只有一個主線程，若要實現對其他線程的基本操作，首先必須創建新的線程，新的線程創建可以使用 pthread_create 函數實現，該函數的 API 定義如下： /* 函數功能：創建新的

逆风微积分 2020-06-26 06:25:41

《網絡編程》原始套接字 ---ping程序實現

概述         基於字節流套接字（SOCK_STREAM）和數據報套接字（SOCK_DGRAM）不可以訪問傳輸層協議，只是對應用層的報文進行操作，傳輸層的數據報格式都是由系統提供的協議棧實現，用戶只需要填充相應的應用層報文，由系統完

逆风微积分 2020-06-26 06:25:30

select 模式 非阻塞

    在windows 下設置select 模式的非阻塞。     具體實現如下       客戶端：（window）  創建一個socket ，非阻塞， 調用connect ， connect 立即返回。 然後用select 檢查，

aobai219 2020-06-16 13:11:10

非阻塞socket 的連接

非阻塞socket 的連接 2010-02-21 14:17 方案1：使用while和usleep的timeout int connect_socket_timeout(int sockfd,char *dest_host, int

aobai219 2020-06-16 13:11:10

UNIX網絡編程 函數歸類

我是小x 2020-04-14 07:54:22

Conncet 問題

aobai219 2020-02-24 11:04:01

常見問題

juzimeizi 2020-02-24 05:55:14

爲什麼TIME_WAIT的時間是2MSL

juzimeizi 2020-02-24 05:55:14

Centos7搭建簡單的FTP服務器

juzimeizi 2020-02-24 05:55:14

UNIX 回射程序

juzimeizi 2020-02-24 05:55:14