解决socket在execl之后被自动继承的问题

本文同步自：https://zohead.com/archives/close-socket-after-execl/

Linux下的进程在创建子进程（fork）后，子进程会自动继承父进程的文件描述符，这种机制在没有 execl 运行时一般都没有问题，而且也比较方便在子进程中直接使用父进程中打开的文件描述符，但如果子进程通过 execl 类函数运行新的程序时，这些继承的文件描述符却没有被关闭，导致 execl 之后的程序中也有父进程中文件描述符的拷贝，有些情况就会出现错误。

由于 socket 也在这些文件描述符范围内，本文简单说下解决 socket 被自动继承的几种方法。

首先是测试程序，头文件就没列出来了。

# ... 包含的头文件... #

int main(int argc, char ** argv)
{
	int ret = 0, sock = 0;
	struct sockaddr_in in_svr = {0};
	pid_t p_ret = 0;

	signal(SIGCHLD, SIG_IGN);

	sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
	if (sock < 0) {
		perror("create socket failed");
		return 1;
	}

	in_svr.sin_family = AF_INET;
	in_svr.sin_addr.s_addr = htonl(INADDR_ANY);
	in_svr.sin_port = htons(5555);

	ret = bind(sock, (struct sockaddr *)&in_svr, sizeof(in_svr));
	if (ret != 0) {
		perror("bind failed");
		close(sock);
		return 2;
	}

	p_ret = fork();

	if (p_ret < 0) {
		perror("fork failed");
		close(sock);
		return 3;
	} else if (p_ret == 0) {
		execl("/bin/sleep", "sleep", "10", NULL);
		exit(0);
	}

	sleep(10);
	close(sock);

	return 0;
}

父进程和子进程分别 sleep 10秒钟，方便外部查看打开的文件描述符，编译运行程序，使用 lsof 命令就可以看到子进程在 execl 之后的 sleep 进程中也打开了父进程的 socket 描述符（UDP连接），这不是我们想要的效果滴：

[root@localhost ~]# lsof -c sleep
    COMMAND     PID   USER   FD     TYPE  DEVICE SIZE/OFF   NODE      NAME
    sleep       5750  root   cwd    DIR   8,1    4096       8145      /root
    sleep       5750  root   rtd    DIR   8,1    4096       2         /
    sleep       5750  root   txt    REG   8,1    27880      456104    /bin/sleep
    sleep       5750  root   mem    REG   8,1    155696     301801    /lib64/ld-2.12.so
    sleep       5750  root   mem    REG   8,1    1912928    301802    /lib64/libc-2.12.so
    sleep       5750  root   mem    REG   8,1    99158752   391655    /usr/lib/locale/locale-archive
    sleep       5750  root   0u     CHR   136,1  0t0        4         /dev/pts/1
    sleep       5750  root   1u     CHR   136,1  0t0        4         /dev/pts/1
    sleep       5750  root   2u     CHR   136,1  0t0        4         /dev/pts/1
    sleep       5750  root   3u     IPv4  15022  0t0        UDP       *:5555

解决办法有如下几种：

比较死板的方法，在子进程 fork 之后关闭所有的 socket 描述符，在 execl 之前加上：
```
int fd = 0, fdtablesize = 0;

for (fd = 0, fdtablesize = getdtablesize(); fd &lt; fdtablesize; fd++) {
  // close all socket handle inherited from parent process
  if (isfdtype(fd, S_IFSOCK)) {
      close(fd);
  }
}
```
这种处理比较彻底，即使子进程没有 execl 也会关闭继承的 socket 句柄，如果要关闭从父进程继承的所有句柄，去掉 isfdtype 判断即可。
使用 fcntl 为父进程 socket 设置 O_CLOEXEC 标志，表示此 socket 在 execl 之前自动关闭（注意并非在 fork 之后），在 socket 创建完成之后加上：
```
ret = fcntl(sock, F_GETFD);
ret = fcntl(sock, F_SETFD, ret | FD_CLOEXEC);
```
Linux 2.6.27 之后创建 socket 时开始支持 SOCK_CLOEXEC 参数，与上面的效果相似，只要改创建 socket 的地方：
```
sock = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, IPPROTO_UDP);
```
另外 Linux 2.6.27 之后创建 socket 另外支持 SOCK_NONBLOCK 参数，如果设置此参数，socket 自动创建为非阻塞式。

以上三种办法均在 RHEL 6.1 64位系统上编译测试通过，有任何问题欢迎交流，玩的开心 ^_^

发表评论 取消回复

发表评论取消回复