今天继续探索《Unix网络编程》的框架。并改善上次的程序
在第一次接触中,我将error.h头文件分出,但是发现错误检测和输出大量的存在,因此归入到common.h中。
加入了一些安全包装函数,就是对于原函数的调用并进行错误检查。
用新的函数改善了daytimecpcli,,并学习服务器程序daytimetcpsrv的编写。
目录结构如下:
各个文件的作用为:
common.h:公共头文件,包含一些常量的定义和函数的声明以及常用头文件
error.c:错误输出函数的定义
wrapsock.c:socket API 的安全封装函数的定义
wrapstdio.c:Standard I/O 的安全封装函数的定义
wrapunix.c:Unix 标准API的安全封装函数的定义
error.c见上一次的日记
下面是其它文件的内容:
common.h:
#ifndef __OUR_COMMON_HDR_H#define __OUR_COMMON_HDR_H#include <sys/types.h>#include <sys/socket.h>#include <netinet/in.h>#include <arpa/inet.h>#include <string.h>#include <stdio.h>#include <stdlib.h>#include <unistd.h>#include <errno.h>#define MAXLINE 4096#define SA struct sockaddr#define LISTENQ 1024 /* 2nd argument to listen() *//* socket wrapper functions */int Accept(int, SA *, socklen_t *);void Bind(int, const SA *, socklen_t);void Connect(int, const SA *, socklen_t);void Listen(int, int);int Socket(int, int, int);/* Unix wrapper functions */ssize_t Read(int, void *, size_t);void Write(int, void *, size_t);void Close(int);/* Std I/O wrapper functions */void Fputs(const char *, FILE *);/* Error output functions */void err_quit(const char *fmt, …);void err_ret(const char *fmt, …);void err_sys(const char *fmt, …);void err_dump(const char *fmt, …);void err_msg(const char *fmt, …);#endifwrapsock.c:
#include "common.h"int Accept(int fd, SA *sa, socklen_t *salenptr){int n;again:if ((n = accept(fd, sa, salenptr)) < 0){#ifdef EPROTOif (errno == EPROTO || errno == ECONNABORTED)#elseif (errno == ECONNABORTED)#endifgoto again;elseerr_sys("accept error");}return n;}void Bind(int fd, const SA *sa, socklen_t salen){if (bind(fd, sa, salen) < 0)err_sys("bind error");}void Connect(int fd, const SA *sa, socklen_t salen){if (connect(fd, sa, salen) < 0)err_sys("connect error");}void Listen(int fd, int backlog){char *ptr;if ((ptr = getenv("LISTENQ")) != NULL)backlog = atoi(ptr);if (listen(fd, backlog) < 0)err_sys("listen error");}int Socket(int family, int type, int protocol){int n;if ((n = socket(family, type, protocol)) < 0)err_sys("socket error");return n;}wrapunix.c:
#include "common.h"ssize_t Read(int fd, void *ptr, size_t nbytes){ssize_t n;if ((n = read(fd, ptr, nbytes)) == -1)err_sys("read error");return n;}void Write(int fd, void *ptr, size_t nbytes){if (write(fd, ptr, nbytes) != nbytes)err_sys("write error");}void Close(int fd){if (close(fd) == -1)err_sys("close error");}wrapstdio.c
#include "common.h"void Fputs(const char *ptr, FILE *stream){if (fputs(ptr, stream) == EOF)err_sys("fputs error");}目前为止就是这样了,慢慢用到的函数再回来添加。
用上这些函数后,可以简化程序:
daytimecpcli.c:
#include "common.h"int main(int argc, char **argv){int sockfd, n;char recvline[MAXLINE + 1];struct sockaddr_in servaddr;if (argc != 2)err_quit("usage: ./daytimecpcli.c <IP address>");sockfd = Socket(AF_INET, SOCK_STREAM, 0);bzero(&servaddr, sizeof(servaddr));servaddr.sin_family = AF_INET;servaddr.sin_port = htons(IPPORT_DAYTIME);if (inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <= 0)err_quit("inet_pton error for %s", argv[1]);Connect(sockfd, (SA *) &servaddr, sizeof(servaddr));while ((n = Read(sockfd, recvline, MAXLINE)) > 0){recvline[n] = 0;Fputs(recvline, stdout);}exit(0);}daytimetcpsrv.c:
#include "common.h"#include <time.h>int main(int argc, char **argv){int listenfd, connfd;struct sockaddr_in servaddr;char buff[MAXLINE];time_t ticks;listenfd = Socket(AF_INET, SOCK_STREAM, 0);bzero(&servaddr, sizeof(servaddr));servaddr.sin_family = AF_INET;servaddr.sin_addr.s_addr = htonl(INADDR_ANY);servaddr.sin_port = htons(IPPORT_DAYTIME);Bind(listenfd, (SA *) &servaddr, sizeof(servaddr));Listen(listenfd, LISTENQ);for (;;){connfd = Accept(listenfd, (SA *) NULL, NULL);ticks = time(NULL);snprintf(buff, sizeof(buff), "%s\r\n", ctime(&ticks));Write(connfd, buff, strlen(buff));Close(connfd);}}执行效果为:
年轻是我们唯一拥有权利去编织梦想的时光