1/*
2   Unix SMB/CIFS implementation.
3   Samba utility functions
4   Copyright (C) Andrew Tridgell 1992-1998
5   Copyright (C) Tim Potter      2000-2001
6
7   This program is free software; you can redistribute it and/or modify
8   it under the terms of the GNU General Public License as published by
9   the Free Software Foundation; either version 3 of the License, or
10   (at your option) any later version.
11
12   This program is distributed in the hope that it will be useful,
13   but WITHOUT ANY WARRANTY; without even the implied warranty of
14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15   GNU General Public License for more details.
16
17   You should have received a copy of the GNU General Public License
18   along with this program.  If not, see <http://www.gnu.org/licenses/>.
19*/
20
21#include "includes.h"
22
23/*******************************************************************
24this is like socketpair but uses tcp. It is used by the Samba
25regression test code
26The function guarantees that nobody else can attach to the socket,
27or if they do that this function fails and the socket gets closed
28returns 0 on success, -1 on failure
29the resulting file descriptors are symmetrical
30 ******************************************************************/
31static int socketpair_tcp(int fd[2])
32{
33	int listener;
34	struct sockaddr_in sock;
35	struct sockaddr_in sock2;
36	socklen_t socklen = sizeof(sock);
37	int connect_done = 0;
38
39	fd[0] = fd[1] = listener = -1;
40
41	memset(&sock, 0, sizeof(sock));
42
43	if ((listener = socket(PF_INET, SOCK_STREAM, 0)) == -1) goto failed;
44
45        memset(&sock2, 0, sizeof(sock2));
46#ifdef HAVE_SOCK_SIN_LEN
47        sock2.sin_len = sizeof(sock2);
48#endif
49        sock2.sin_family = PF_INET;
50
51        if (bind(listener, (struct sockaddr *)&sock2, sizeof(sock2)) != 0) goto failed;
52
53	if (listen(listener, 1) != 0) goto failed;
54
55	if (getsockname(listener, (struct sockaddr *)&sock, &socklen) != 0) goto failed;
56
57	if ((fd[1] = socket(PF_INET, SOCK_STREAM, 0)) == -1) goto failed;
58
59	set_blocking(fd[1], 0);
60
61	sock.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
62
63	if (sys_connect(fd[1], (struct sockaddr *)&sock) == -1) {
64		if (errno != EINPROGRESS) goto failed;
65	} else {
66		connect_done = 1;
67	}
68
69	if ((fd[0] = accept(listener, (struct sockaddr *)&sock, &socklen)) == -1) goto failed;
70
71	close(listener);
72	if (connect_done == 0) {
73		if (sys_connect(fd[1], (struct sockaddr *)&sock) != 0
74		    && errno != EISCONN) goto failed;
75	}
76
77	set_blocking(fd[1], 1);
78
79	/* all OK! */
80	return 0;
81
82 failed:
83	if (fd[0] != -1) close(fd[0]);
84	if (fd[1] != -1) close(fd[1]);
85	if (listener != -1) close(listener);
86	return -1;
87}
88
89
90/*******************************************************************
91run a program on a local tcp socket, this is used to launch smbd
92when regression testing
93the return value is a socket which is attached to a subprocess
94running "prog". stdin and stdout are attached. stderr is left
95attached to the original stderr
96 ******************************************************************/
97int sock_exec(const char *prog)
98{
99	int fd[2];
100	if (socketpair_tcp(fd) != 0) {
101		DEBUG(0,("socketpair_tcp failed (%s)\n", strerror(errno)));
102		return -1;
103	}
104	if (fork() == 0) {
105		close(fd[0]);
106		close(0);
107		close(1);
108		if (dup(fd[1]) == -1) {
109			exit(1);
110		}
111		if (dup(fd[1]) == -1) {
112			exit(1);
113		}
114		exit(system(prog));
115	}
116	close(fd[1]);
117	return fd[0];
118}
119