summaryrefslogtreecommitdiff
path: root/source3/utils
diff options
context:
space:
mode:
authorLuke Leighton <lkcl@samba.org>1999-12-03 23:36:53 +0000
committerLuke Leighton <lkcl@samba.org>1999-12-03 23:36:53 +0000
commit0ca1f87930a57dfd2510b98443423d8a67cfa70b (patch)
tree3017e6ef3204c58314f8dae4a9042d963a5979fd /source3/utils
parentc2a3b11b0738f57accafe19c84b682f761fa9631 (diff)
downloadsamba-0ca1f87930a57dfd2510b98443423d8a67cfa70b.tar.gz
samba-0ca1f87930a57dfd2510b98443423d8a67cfa70b.tar.bz2
samba-0ca1f87930a57dfd2510b98443423d8a67cfa70b.zip
argh! smb-agent redirection client reusage is a nightmare!
moved smb-agent over to a single-process model instead of fork() in order to reuse client connections. except, of course, you can't do a select() on the same socket connections! argh! (This used to be commit e9e5a34de8e8f9a69e817aceb8c16284334d4642)
Diffstat (limited to 'source3/utils')
-rw-r--r--source3/utils/smb-agent.c368
1 files changed, 225 insertions, 143 deletions
diff --git a/source3/utils/smb-agent.c b/source3/utils/smb-agent.c
index bcc5eebd75..a52d696615 100644
--- a/source3/utils/smb-agent.c
+++ b/source3/utils/smb-agent.c
@@ -33,74 +33,139 @@
#define CLI_CAPABILITY_MASK CAP_UNICODE
#define CLI_CAPABILITY_SET 0
-static char *netbiosname;
static char packet[BUFFER_SIZE];
extern int DEBUGLEVEL;
-static void agent_reply(char *buf)
-{
- int msg_type = CVAL(buf,0);
- int type = CVAL(buf,smb_com);
- unsigned x;
-
- if (msg_type) return;
- switch (type) {
+struct sock_redir
+{
+ int c;
+ struct cli_state *s;
- case SMBnegprot:
- /* force the security bits */
- x = CVAL(buf, smb_vwv1);
- x = (x | SECURITY_SET) & ~SECURITY_MASK;
- SCVAL(buf, smb_vwv1, x);
+};
- /* force the capabilities */
- x = IVAL(buf,smb_vwv9+1);
- x = (x | CAPABILITY_SET) & ~CAPABILITY_MASK;
- SIVAL(buf, smb_vwv9+1, x);
- break;
+static uint32 num_socks = 0;
+static struct sock_redir **socks = NULL;
+/****************************************************************************
+terminate sockent connection
+****************************************************************************/
+static void sock_redir_free(struct sock_redir *sock)
+{
+ close(sock->c);
+ sock->c = -1;
+ if (sock->s != NULL)
+ {
+ cli_net_use_del(sock->s->desthost, &sock->s->usr,
+ False, NULL);
+ sock->s = NULL;
}
+ free(sock);
+}
+
+/****************************************************************************
+free a sockent array
+****************************************************************************/
+static void free_sock_array(uint32 num_entries, struct sock_redir **entries)
+{
+ void(*fn)(void*) = (void(*)(void*))&sock_redir_free;
+ free_void_array(num_entries, (void**)entries, *fn);
}
-static void agent_request(char *buf)
+/****************************************************************************
+add a sockent state to the array
+****************************************************************************/
+static struct sock_redir* add_sock_to_array(uint32 *len,
+ struct sock_redir ***array,
+ struct sock_redir *sock)
{
- int msg_type = CVAL(buf,0);
- int type = CVAL(buf,smb_com);
- pstring name1,name2;
- unsigned x;
-
- if (msg_type) {
- /* it's a netbios special */
- switch (msg_type) {
- case 0x81:
- /* session request */
- name_extract(buf,4,name1);
- name_extract(buf,4 + name_len(buf + 4),name2);
- DEBUG(0,("sesion_request: %s -> %s\n",
- name1, name2));
- if (netbiosname) {
- /* replace the destination netbios name */
- name_mangle(netbiosname, buf+4, 0x20);
- }
+ int i;
+ for (i = 0; i < num_socks; i++)
+ {
+ if (socks[i] == NULL)
+ {
+ socks[i] = sock;
+ return sock;
}
- return;
}
- /* it's an ordinary SMB request */
- switch (type) {
- case SMBsesssetupX:
- /* force the client capabilities */
- x = IVAL(buf,smb_vwv11);
- x = (x | CLI_CAPABILITY_SET) & ~CLI_CAPABILITY_MASK;
- SIVAL(buf, smb_vwv11, x);
- break;
+ return (struct sock_redir*)add_item_to_array(len,
+ (void***)array, (void*)sock);
+
+}
+
+/****************************************************************************
+initiate sockent array
+****************************************************************************/
+void init_sock_redir(void)
+{
+ socks = NULL;
+ num_socks = 0;
+}
+
+/****************************************************************************
+terminate sockent array
+****************************************************************************/
+void free_sock_redir(void)
+{
+ free_sock_array(num_socks, socks);
+ init_sock_redir();
+}
+
+/****************************************************************************
+create a new sockent state from user credentials
+****************************************************************************/
+static struct sock_redir *sock_redir_get(int fd, struct cli_state *cli)
+{
+ struct sock_redir *sock = (struct sock_redir*)malloc(sizeof(*sock));
+
+ if (sock == NULL)
+ {
+ return NULL;
}
+ ZERO_STRUCTP(sock);
+
+ sock->c = fd;
+ sock->s = cli;
+
+ return sock;
+}
+
+/****************************************************************************
+init sock state
+****************************************************************************/
+static void sock_add(int fd, struct cli_state *cli)
+{
+ struct sock_redir *sock;
+ sock = sock_redir_get(fd, cli);
+ if (sock != NULL)
+ {
+ add_sock_to_array(&num_socks, &socks, sock);
+ }
}
-#define AGENT_CMD_CON 0
-#define AGENT_CMD_CON_REUSE 1
+/****************************************************************************
+delete a sockent state
+****************************************************************************/
+static BOOL sock_del(int fd)
+{
+ int i;
+
+ for (i = 0; i < num_socks; i++)
+ {
+ if (socks[i] == NULL) continue;
+ if (socks[i]->c == fd)
+ {
+ sock_redir_free(socks[i]);
+ socks[i] = NULL;
+ return True;
+ }
+ }
+
+ return False;
+}
static struct cli_state *init_client_connection(int c)
{
@@ -142,7 +207,7 @@ static struct cli_state *init_client_connection(int c)
if (rl < 0)
{
DEBUG(0,("Unable to read from connection\n"));
- exit(1);
+ return NULL;
}
#ifdef DEBUG_PASSWORD
@@ -162,10 +227,10 @@ static struct cli_state *init_client_connection(int c)
if (PTR_DIFF(p, buf) < rl)
{
- memcpy(ntpw, p, 16);
- p += 16;
memcpy(lmpw, p, 16);
p += 16;
+ memcpy(ntpw, p, 16);
+ p += 16;
pwd_set_lm_nt_16(&usr.pwd, lmpw, ntpw);
}
else
@@ -177,7 +242,7 @@ static struct cli_state *init_client_connection(int c)
{
DEBUG(0,("Buffer size %d %d!\n",
PTR_DIFF(p, buf), rl));
- exit(1);
+ return NULL;
}
switch (command)
@@ -221,88 +286,60 @@ static struct cli_state *init_client_connection(int c)
return NULL;
}
-static void agent_child(int c)
+void process_cli_sock(struct sock_redir **sock)
{
- struct cli_state *s = NULL;
-
- DEBUG(10,("agent_child: %d\n", c));
-
- while (c != -1)
+ struct cli_state *s = (*sock)->s;
+ if (s == NULL)
{
- fd_set fds;
- int num;
- int maxfd = 0;
-
- FD_ZERO(&fds);
- if (s != NULL)
- {
- FD_SET(s->fd, &fds);
- maxfd = MAX(s->fd, maxfd);
- }
-
- if (c != -1)
+ s = init_client_connection((*sock)->c);
+ if (s == NULL)
{
- FD_SET(c, &fds);
- maxfd = MAX(c, maxfd);
+ sock_redir_free(*sock);
+ *sock = NULL;
+ return ;
}
-
- num = sys_select(maxfd+1,&fds,NULL, NULL);
- if (num <= 0) continue;
-
- if (c != -1 && FD_ISSET(c, &fds))
+ (*sock)->s = s;
+ }
+ else
+ {
+ if (!receive_smb((*sock)->c, packet, 0))
{
- if (s == NULL)
- {
- s = init_client_connection(c);
- if (s == NULL)
- {
- exit(1);
- }
- }
- else
- {
- if (!receive_smb(c, packet, 0))
- {
- DEBUG(0,("client closed connection\n"));
- exit(0);
- }
- /* ignore keep-alives */
- if (CVAL(packet, 0) != 0x85)
- {
- if (!send_smb(s->fd, packet))
- {
- DEBUG(0,("server is dead\n"));
- exit(1);
- }
- }
- }
+ DEBUG(0,("client closed connection\n"));
+ sock_redir_free(*sock);
+ *sock = NULL;
+ return;
}
- if (s != NULL && FD_ISSET(s->fd, &fds))
+ /* ignore keep-alives */
+ if (CVAL(packet, 0) != 0x85)
{
- if (!receive_smb(s->fd, packet, 0))
+ if (!send_smb(s->fd, packet))
{
- DEBUG(0,("server closed connection\n"));
- exit(0);
- }
-#if 0
- agent_reply(packet);
-#endif
- if (!send_smb(c, packet))
- {
- DEBUG(0,("client is dead\n"));
- cli_shutdown(s);
- free(s);
- exit(1);
+ DEBUG(0,("server is dead\n"));
+ sock_redir_free(*sock);
+ *sock = NULL;
+ return;
}
}
}
- DEBUG(0,("Connection closed\n"));
- if (s != NULL)
+}
+
+void process_srv_sock(struct sock_redir **sock)
+{
+ struct cli_state *s = (*sock)->s;
+ if (!receive_smb(s->fd, packet, 0))
{
- cli_shutdown(s);
- free(s);
+ DEBUG(0,("server closed connection\n"));
+ sock_redir_free(*sock);
+ (*sock) = NULL;
+ return;
}
- exit(0);
+ if (!send_smb((*sock)->c, packet))
+ {
+ DEBUG(0,("client is dead\n"));
+ sock_redir_free(*sock);
+ (*sock) = NULL;
+ return;
+ }
}
@@ -311,13 +348,22 @@ static void start_agent(void)
int s, c;
struct sockaddr_un sa;
fstring path;
- slprintf(path, sizeof(path)-1, "/tmp/smb-agent/smb.%d", getuid());
+ fstring dir;
CatchChild();
- /* start listening on unix socket */
- mkdir("/tmp/smb-agent", 777);
+ slprintf(dir, sizeof(dir)-1, "/tmp/.smb.%d", getuid());
+ mkdir(dir, S_IRUSR|S_IWUSR|S_IXUSR);
+
+ slprintf(path, sizeof(path)-1, "%s/agent", dir);
+ if (chmod(dir, S_IRUSR|S_IWUSR|S_IXUSR) < 0)
+ {
+ fprintf(stderr, "chmod on %s failed\n", sa.sun_path);
+ exit(1);
+ }
+
+ /* start listening on unix socket */
s = socket(AF_UNIX, SOCK_STREAM, 0);
if (s < 0)
@@ -338,14 +384,6 @@ static void start_agent(void)
exit(1);
}
- if (chmod(path, S_IRUSR|S_IWUSR|S_ISVTX) < 0)
- {
- fprintf(stderr, "chmod on %s failed\n", sa.sun_path);
- close(s);
- remove(path);
- exit(1);
- }
-
if (s == -1)
{
DEBUG(0,("bind failed\n"));
@@ -361,29 +399,73 @@ static void start_agent(void)
while (1)
{
+ int i;
fd_set fds;
int num;
struct sockaddr_un addr;
int in_addrlen = sizeof(addr);
+ int maxfd = s;
FD_ZERO(&fds);
FD_SET(s, &fds);
- num = sys_select(s+1,&fds,NULL, NULL);
- if (num > 0)
+ for (i = 0; i < num_socks; i++)
{
- c = accept(s, (struct sockaddr*)&addr, &in_addrlen);
- if (c != -1) {
- if (fork() == 0)
+ if (socks[i] != NULL)
+ {
+ int fd = socks[i]->c;
+ FD_SET(fd, &fds);
+ maxfd = MAX(maxfd, fd);
+
+ if (socks[i]->s != NULL)
{
- close(s);
- agent_child(c);
- exit(0);
- } else {
- close(c);
+ fd = socks[i]->s->fd;
+ FD_SET(fd, &fds);
+ maxfd = MAX(fd, maxfd);
}
}
}
+
+ dbgflush();
+ num = sys_select(maxfd+1,&fds,NULL, NULL);
+
+ if (num <= 0)
+ {
+ continue;
+ }
+
+ if (FD_ISSET(s, &fds))
+ {
+ c = accept(s, (struct sockaddr*)&addr, &in_addrlen);
+ if (c != -1)
+ {
+ sock_add(c, NULL);
+ }
+ }
+
+ for (i = 0; i < num_socks; i++)
+ {
+ if (socks[i] == NULL)
+ {
+ continue;
+ }
+ if (FD_ISSET(socks[i]->c, &fds))
+ {
+ process_cli_sock(&socks[i]);
+ }
+ if (socks[i] == NULL)
+ {
+ continue;
+ }
+ if (socks[i]->s == NULL)
+ {
+ continue;
+ }
+ if (FD_ISSET(socks[i]->s->fd, &fds))
+ {
+ process_srv_sock(&socks[i]);
+ }
+ }
}
}