《UNIX网络编程 卷1》 笔记： 错误处理和包裹函数

错误处理是程序中必不可少的一部分。为了处理程序运行中的系统调用错误（比如read系统调用返回出错）和一般性错误（比如程序参数不正确，程序逻辑错误等），我们定义了一些通用的错误处理函数。这些函数的功能包括输出错误消息，输出错误消息并终止进程，将错误消息记录到syslog等等。代码如下：

#include "unp.h"

#include	<syslog.h>

int		daemon_proc; /*后台进程标志，输出消息到syslog*/

static void err_doit(int errnoflag, int level, const char *fmt, va_list ap);

/*与系统调用无关的错误，打印错误消息*/
void err_ret(const char *fmt, ...)
{
	va_list		ap;

	va_start(ap, fmt);
	err_doit(1, LOG_INFO, fmt, ap);
	va_end(ap);
	return;
}

/*与系统调用无关的错误，打印错误消息*/
void err_msg(const char *fmt, ...)
{
	va_list		ap;

	va_start(ap, fmt);
	err_doit(0, LOG_INFO, fmt, ap);
	va_end(ap);
	return;
}

/*与系统调用无关的错误，打印错误消息并终止程序*/
void err_quit(const char *fmt, ...)
{
	va_list		ap;

	va_start(ap, fmt);
	err_doit(0, LOG_ERR, fmt, ap);
	va_end(ap);
	exit(1);
}

/*系统调用出现错误，打印错误消息并终止程序*/
void err_sys(const char *fmt, ...)
{
	va_list		ap;

	va_start(ap, fmt);
	err_doit(1, LOG_ERR, fmt, ap);
	va_end(ap);
	exit(1);
}


/*与系统调用相关的严重错误，打印错误消息，dump core，终止程序运行*/
void err_dump(const char *fmt, ...)
{
	va_list		ap;

	va_start(ap, fmt);
	err_doit(1, LOG_ERR, fmt, ap);
	va_end(ap);
	abort();
	exit(1);
}

static void  err_doit(int errnoflag, int level, const char *fmt, va_list ap)
{
	int errno_save, n;
	char buf[MAXLINE + 1];
	
	errno_save = errno;
	vsnprintf(buf, MAXLINE, fmt, ap);
	n = strlen(buf);
	if (errnoflag)
		snprintf(buf + n, MAXLINE - n, ": %s", strerror(errno_save));
	strcat(buf, "\n");

	if (daemon_proc) {
		syslog(level, buf);
	} else {
		fflush(stdout);
		fputs(buf, stderr);
		fflush(stderr);
	}
	
	return;
}

为了减少主函数中处理错误的代码量，使代码流程看起来更清晰，我们为使用的每一个系统调用都定义了一个包裹函数。比如read系统调用的包裹函数是Read（将系统调用名称首字母大写）：

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);
}

这样在使用时直接写成

Read(fd, ptr, nbytes)；

即可， 而不需要写成如下的代码：

if (read(fd, ptr, nbytes) == -1) {
	printf("read error!\n");
	exit(1);
}