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 2 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, write to the Free Software
19   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20*/
21
22#include "includes.h"
23
24/*******************************************************************
25this is like socketpair but uses tcp. It is used by the Samba
26regression test code
27The function guarantees that nobody else can attach to the socket,
28or if they do that this function fails and the socket gets closed
29returns 0 on success, -1 on failure
30the resulting file descriptors are symmetrical
31 ******************************************************************/
32static int socketpair_tcp(int fd[2])
33{
34	int listener;
35	struct sockaddr_in sock;
36	struct sockaddr_in sock2;
37	socklen_t socklen = sizeof(sock);
38	int connect_done = 0;
39
40	fd[0] = fd[1] = listener = -1;
41
42	memset(&sock, 0, sizeof(sock));
43
44	if ((listener = socket(PF_INET, SOCK_STREAM, 0)) == -1) goto failed;
45
46        memset(&sock2, 0, sizeof(sock2));
47#ifdef HAVE_SOCK_SIN_LEN
48        sock2.sin_len = sizeof(sock2);
49#endif
50        sock2.sin_family = PF_INET;
51
52        bind(listener, (struct sockaddr *)&sock2, sizeof(sock2));
53
54	if (listen(listener, 1) != 0) goto failed;
55
56	if (getsockname(listener, (struct sockaddr *)&sock, &socklen) != 0) goto failed;
57
58	if ((fd[1] = socket(PF_INET, SOCK_STREAM, 0)) == -1) goto failed;
59
60	set_blocking(fd[1], 0);
61
62	sock.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
63
64	if (connect(fd[1],(struct sockaddr *)&sock,sizeof(sock)) == -1) {
65		if (errno != EINPROGRESS) goto failed;
66	} else {
67		connect_done = 1;
68	}
69
70	if ((fd[0] = accept(listener, (struct sockaddr *)&sock, &socklen)) == -1) goto failed;
71
72	close(listener);
73	if (connect_done == 0) {
74		if (connect(fd[1],(struct sockaddr *)&sock,sizeof(sock)) != 0
75		    && errno != EISCONN) goto failed;
76	}
77
78	set_blocking(fd[1], 1);
79
80	/* all OK! */
81	return 0;
82
83 failed:
84	if (fd[0] != -1) close(fd[0]);
85	if (fd[1] != -1) close(fd[1]);
86	if (listener != -1) close(listener);
87	return -1;
88}
89
90
91/*******************************************************************
92run a program on a local tcp socket, this is used to launch smbd
93when regression testing
94the return value is a socket which is attached to a subprocess
95running "prog". stdin and stdout are attached. stderr is left
96attached to the original stderr
97 ******************************************************************/
98int sock_exec(const char *prog)
99{
100	int fd[2];
101	if (socketpair_tcp(fd) != 0) {
102		DEBUG(0,("socketpair_tcp failed (%s)\n", strerror(errno)));
103		return -1;
104	}
105	if (fork() == 0) {
106		close(fd[0]);
107		close(0);
108		close(1);
109		dup(fd[1]);
110		dup(fd[1]);
111		exit(system(prog));
112	}
113	close(fd[1]);
114	return fd[0];
115}
116