Initial import
[samba] / source / libsmb / clidgram.c
1 /* 
2    Unix SMB/CIFS implementation.
3    client dgram calls
4    Copyright (C) Andrew Tridgell 1994-1998
5    Copyright (C) Richard Sharpe 2001
6    Copyright (C) John Terpstra 2001
7
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23 #include "includes.h"
24
25 /*
26  * cli_send_mailslot, send a mailslot for client code ...
27  */
28
29 BOOL cli_send_mailslot(BOOL unique, const char *mailslot,
30                        uint16 priority,
31                        char *buf, int len,
32                        const char *srcname, int src_type, 
33                        const char *dstname, int dest_type,
34                        struct in_addr dest_ip)
35 {
36         struct packet_struct p;
37         struct dgram_packet *dgram = &p.packet.dgram;
38         char *ptr, *p2;
39         char tmp[4];
40         pid_t nmbd_pid;
41
42         if ((nmbd_pid = pidfile_pid("nmbd")) == 0) {
43                 DEBUG(3, ("No nmbd found\n"));
44                 return False;
45         }
46
47         if (!message_init())
48                 return False;
49
50         memset((char *)&p, '\0', sizeof(p));
51
52         /*
53          * Next, build the DGRAM ...
54          */
55
56         /* DIRECT GROUP or UNIQUE datagram. */
57         dgram->header.msg_type = unique ? 0x10 : 0x11; 
58         dgram->header.flags.node_type = M_NODE;
59         dgram->header.flags.first = True;
60         dgram->header.flags.more = False;
61         dgram->header.dgm_id = ((unsigned)time(NULL)%(unsigned)0x7FFF) +
62                 ((unsigned)sys_getpid()%(unsigned)100);
63         /* source ip is filled by nmbd */
64         dgram->header.dgm_length = 0; /* Let build_dgram() handle this. */
65         dgram->header.packet_offset = 0;
66   
67         make_nmb_name(&dgram->source_name,srcname,src_type);
68         make_nmb_name(&dgram->dest_name,dstname,dest_type);
69
70         ptr = &dgram->data[0];
71
72         /* Setup the smb part. */
73         ptr -= 4; /* XXX Ugliness because of handling of tcp SMB length. */
74         memcpy(tmp,ptr,4);
75         set_message(ptr,17,strlen(mailslot) + 1 + len,True);
76         memcpy(ptr,tmp,4);
77
78         SCVAL(ptr,smb_com,SMBtrans);
79         SSVAL(ptr,smb_vwv1,len);
80         SSVAL(ptr,smb_vwv11,len);
81         SSVAL(ptr,smb_vwv12,70 + strlen(mailslot));
82         SSVAL(ptr,smb_vwv13,3);
83         SSVAL(ptr,smb_vwv14,1);
84         SSVAL(ptr,smb_vwv15,priority);
85         SSVAL(ptr,smb_vwv16,2);
86         p2 = smb_buf(ptr);
87         fstrcpy(p2,mailslot);
88         p2 = skip_string(p2,1);
89
90         memcpy(p2,buf,len);
91         p2 += len;
92
93         dgram->datasize = PTR_DIFF(p2,ptr+4); /* +4 for tcp length. */
94
95         p.packet_type = DGRAM_PACKET;
96         p.ip = dest_ip;
97         p.timestamp = time(NULL);
98
99         DEBUG(4,("send_mailslot: Sending to mailslot %s from %s ",
100                  mailslot, nmb_namestr(&dgram->source_name)));
101         DEBUGADD(4,("to %s IP %s\n", nmb_namestr(&dgram->dest_name),
102                     inet_ntoa(dest_ip)));
103
104         return message_send_pid(pid_to_procid(nmbd_pid), MSG_SEND_PACKET, &p, sizeof(p),
105                                 False);
106 }
107
108 /*
109  * cli_get_response: Get a response ...
110  */
111 BOOL cli_get_response(const char *mailslot, char *buf, int bufsiz)
112 {
113         struct packet_struct *p;
114
115         p = receive_unexpected(DGRAM_PACKET, 0, mailslot);
116
117         if (p == NULL)
118                 return False;
119
120         memcpy(buf, &p->packet.dgram.data[92],
121                MIN(bufsiz, p->packet.dgram.datasize-92));
122
123         return True;
124 }
125
126 /*
127  * cli_get_backup_list: Send a get backup list request ...
128  */
129
130 static char cli_backup_list[1024];
131
132 int cli_get_backup_list(const char *myname, const char *send_to_name)
133 {
134         pstring outbuf;
135         char *p;
136         struct in_addr sendto_ip;
137
138         if (!resolve_name(send_to_name, &sendto_ip, 0x1d)) {
139
140                 DEBUG(0, ("Could not resolve name: %s<1D>\n", send_to_name));
141                 return False;
142
143         }
144
145         memset(cli_backup_list, '\0', sizeof(cli_backup_list));
146         memset(outbuf, '\0', sizeof(outbuf));
147
148         p = outbuf;
149
150         SCVAL(p, 0, ANN_GetBackupListReq);
151         p++;
152
153         SCVAL(p, 0, 1); /* Count pointer ... */
154         p++;
155
156         SIVAL(p, 0, 1); /* The sender's token ... */
157         p += 4;
158
159         cli_send_mailslot(True, "\\MAILSLOT\\BROWSE", 1, outbuf, 
160                           PTR_DIFF(p, outbuf), myname, 0, send_to_name, 
161                           0x1d, sendto_ip);
162
163         /* We should check the error and return if we got one */
164
165         /* Now, get the response ... */
166
167         cli_get_response("\\MAILSLOT\\BROWSE",
168                          cli_backup_list, sizeof(cli_backup_list));
169
170         return True;
171
172 }
173
174 /*
175  * cli_get_backup_server: Get the backup list and retrieve a server from it
176  */
177
178 int cli_get_backup_server(char *my_name, char *target, char *servername, int namesize)
179 {
180
181   /* Get the backup list first. We could pull this from the cache later */
182
183   cli_get_backup_list(my_name, target);  /* FIXME: Check the response */
184
185   if (!cli_backup_list[0]) { /* Empty list ... try again */
186
187     cli_get_backup_list(my_name, target);
188
189   }
190
191   strncpy(servername, cli_backup_list, MIN(16, namesize));
192
193   return True;
194
195 }