source/libsmb/clidgram.c

   1 /*
   2    Unix SMB/Netbios implementation.
   3    Version 3.0
   4    client dgram calls
   5    Copyright (C) Andrew Tridgell 1994-1998
   6    Copyright (C) Richard Sharpe 2001
   7    Copyright (C) John Terpstra 2001
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 2 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program; if not, write to the Free Software
  21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  22 */
  23
  24 #define NO_SYSLOG
  25
  26 #include "includes.h"
  27
  28 /*
  29  * cli_send_mailslot, send a mailslot for client code ...
  30  */
  31
  32 int cli_send_mailslot(int dgram_sock, BOOL unique, char *mailslot,
  33                       char *buf, int len,
  34                       const char *srcname, int src_type,
  35                       const char *dstname, int dest_type,
  36                       struct in_addr dest_ip, struct in_addr src_ip,
  37                       int dest_port, int src_port)
  38 {
  39   struct packet_struct p;
  40   struct dgram_packet *dgram = &p.packet.dgram;
  41   char *ptr, *p2;
  42   char tmp[4];
  43
  44   memset((char *)&p, '\0', sizeof(p));
  45
  46   /*
  47    * Next, build the DGRAM ...
  48    */
  49
  50   /* DIRECT GROUP or UNIQUE datagram. */
  51   dgram->header.msg_type = unique ? 0x10 : 0x11;
  52   dgram->header.flags.node_type = M_NODE;
  53   dgram->header.flags.first = True;
  54   dgram->header.flags.more = False;
  55   dgram->header.dgm_id = ((unsigned)time(NULL)%(unsigned)0x7FFF) + ((unsigned)sys_getpid()%(unsigned)100);
  56   dgram->header.source_ip.s_addr = src_ip.s_addr;
  57   /*fprintf(stderr, "Source IP = %0X\n", dgram->header.source_ip); */
  58   dgram->header.source_port = ntohs(src_port);
  59   fprintf(stderr, "Source Port = %0X\n", dgram->header.source_port);
  60   dgram->header.dgm_length = 0; /* Let build_dgram() handle this. */
  61   dgram->header.packet_offset = 0;
  62
  63   make_nmb_name(&dgram->source_name,srcname,src_type);
  64   make_nmb_name(&dgram->dest_name,dstname,dest_type);
  65
  66   ptr = &dgram->data[0];
  67
  68   /* Setup the smb part. */
  69   ptr -= 4; /* XXX Ugliness because of handling of tcp SMB length. */
  70   memcpy(tmp,ptr,4);
  71   set_message(ptr,17,17 + len,True);
  72   memcpy(ptr,tmp,4);
  73
  74   SCVAL(ptr,smb_com,SMBtrans);
  75   SSVAL(ptr,smb_vwv1,len);
  76   SSVAL(ptr,smb_vwv11,len);
  77   SSVAL(ptr,smb_vwv12,70 + strlen(mailslot));
  78   SSVAL(ptr,smb_vwv13,3);
  79   SSVAL(ptr,smb_vwv14,1);
  80   SSVAL(ptr,smb_vwv15,1);
  81   SSVAL(ptr,smb_vwv16,2);
  82   p2 = smb_buf(ptr);
  83   pstrcpy(p2,mailslot);
  84   p2 = skip_string(p2,1);
  85
  86   memcpy(p2,buf,len);
  87   p2 += len;
  88
  89   dgram->datasize = PTR_DIFF(p2,ptr+4); /* +4 for tcp length. */
  90
  91   p.ip = dest_ip;
  92   p.port = dest_port;
  93   p.fd = dgram_sock;
  94   p.timestamp = time(NULL);
  95   p.packet_type = DGRAM_PACKET;
  96
  97   DEBUG(4,("send_mailslot: Sending to mailslot %s from %s IP %s ", mailslot,
  98                     nmb_namestr(&dgram->source_name), inet_ntoa(src_ip)));
  99   DEBUG(4,("to %s IP %s\n", nmb_namestr(&dgram->dest_name), inet_ntoa(dest_ip)));
 100
 101   return send_packet(&p);
 102
 103 }
 104
 105 /*
 106  * cli_get_response: Get a response ...
 107  */
 108 int cli_get_response(int dgram_sock, BOOL unique, char *mailslot, char *buf, int bufsiz)
 109 {
 110   struct packet_struct *packet;
 111
 112   packet = receive_dgram_packet(dgram_sock, 5, mailslot);
 113
 114   if (packet) { /* We got one, pull what we want out of the SMB data ... */
 115
 116     struct dgram_packet *dgram = &packet->packet.dgram;
 117
 118     /*
 119      * We should probably parse the SMB, but for now, we will pull what
 120      * from fixed, known locations ...
 121      */
 122
 123     /* Copy the data to buffer, respecting sizes ... */
 124
 125     memcpy(buf, &dgram->data[92], MIN(bufsiz, (dgram->datasize - 92)));
 126
 127   }
 128   else
 129     return -1;
 130
 131   return 0;
 132
 133 }
 134
 135 /*
 136  * cli_get_backup_list: Send a get backup list request ...
 137  */
 138
 139 static char cli_backup_list[1024];
 140
 141 int cli_get_backup_list(const char *myname, const char *send_to_name)
 142 {
 143   char outbuf[15];
 144   char *p;
 145   struct in_addr sendto_ip, my_ip;
 146   int dgram_sock;
 147   struct sockaddr_in sock_out;
 148   socklen_t name_size;
 149
 150   if (!resolve_name(send_to_name, &sendto_ip, 0x1d)) {
 151
 152     DEBUG(0, ("Could not resolve name: %s<1D>\n", send_to_name));
 153     return False;
 154
 155   }
 156
 157   my_ip.s_addr = inet_addr("0.0.0.0");
 158
 159   if (!resolve_name(myname, &my_ip, 0x00)) { /* FIXME: Call others here */
 160
 161     DEBUG(0, ("Could not resolve name: %s<00>\n", myname));
 162
 163   }
 164
 165   if ((dgram_sock = open_socket_out(SOCK_DGRAM, &sendto_ip, 138, LONG_CONNECT_TIMEOUT)) < 0) {
 166
 167     DEBUG(4, ("open_sock_out failed ..."));
 168     return False;
 169
 170   }
 171
 172   /* Make it a broadcast socket ... */
 173
 174   set_socket_options(dgram_sock, "SO_BROADCAST");
 175
 176   /* Make it non-blocking??? */
 177
 178   if (fcntl(dgram_sock, F_SETFL, O_NONBLOCK) < 0) {
 179
 180     DEBUG(0, ("Unable to set non blocking on dgram sock\n"));
 181
 182   }
 183
 184   /* Now, bind a local addr to it ... Try port 138 first ... */
 185
 186   memset((char *)&sock_out, '\0', sizeof(sock_out));
 187   sock_out.sin_addr.s_addr = INADDR_ANY;
 188   sock_out.sin_port = htons(138);
 189   sock_out.sin_family = AF_INET;
 190
 191   if (bind(dgram_sock, (struct sockaddr *)&sock_out, sizeof(sock_out)) < 0) {
 192
 193     /* Try again on any port ... */
 194
 195     sock_out.sin_port = INADDR_ANY;
 196
 197     if (bind(dgram_sock, (struct sockaddr *)&sock_out, sizeof(sock_out)) < 0) {
 198
 199       DEBUG(4, ("failed to bind socket to address ...\n"));
 200       return False;
 201
 202     }
 203
 204   }
 205
 206   /* Now, figure out what socket name we were bound to. We want the port */
 207
 208   name_size = sizeof(sock_out);
 209
 210   getsockname(dgram_sock, (struct sockaddr *)&sock_out, &name_size);
 211
 212   DEBUG(5, ("Socket bound to IP:%s, port: %d\n", inet_ntoa(sock_out.sin_addr), ntohs(sock_out.sin_port)));
 213
 214   /* Now, build the request */
 215
 216   memset(cli_backup_list, '\0', sizeof(cli_backup_list));
 217   memset(outbuf, '\0', sizeof(outbuf));
 218
 219   p = outbuf;
 220
 221   SCVAL(p, 0, ANN_GetBackupListReq);
 222   p++;
 223
 224   SCVAL(p, 0, 1); /* Count pointer ... */
 225   p++;
 226
 227   SIVAL(p, 0, 1); /* The sender's token ... */
 228   p += 4;
 229
 230   cli_send_mailslot(dgram_sock, True, "\\MAILSLOT\\BROWSE", outbuf,
 231                     PTR_DIFF(p, outbuf), myname, 0, send_to_name,
 232                     0x1d, sendto_ip, my_ip, 138, sock_out.sin_port);
 233
 234   /* We should check the error and return if we got one */
 235
 236   /* Now, get the response ... */
 237
 238   cli_get_response(dgram_sock, True, "\\MAILSLOT\\BROWSE", cli_backup_list, sizeof(cli_backup_list));
 239
 240   /* Should check the response here ... FIXME */
 241
 242   close(dgram_sock);
 243
 244   return True;
 245
 246 }
 247
 248 /*
 249  * cli_get_backup_server: Get the backup list and retrieve a server from it
 250  */
 251
 252 int cli_get_backup_server(char *my_name, char *target, char *servername, int namesize)
 253 {
 254
 255   /* Get the backup list first. We could pull this from the cache later */
 256
 257   cli_get_backup_list(my_name, target);  /* FIXME: Check the response */
 258
 259   if (!cli_backup_list[0]) { /* Empty list ... try again */
 260
 261     cli_get_backup_list(my_name, target);
 262
 263   }
 264
 265   strncpy(servername, cli_backup_list, MIN(16, namesize));
 266
 267   return True;
 268
 269 }
 270
 271
 272