source3/libsmb/climessage.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    client message handling routines
   4    Copyright (C) Andrew Tridgell 1994-1998
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include "includes.h"
  21
  22 /****************************************************************************
  23  Start a message sequence.
  24 ****************************************************************************/
  25
  26 int cli_message_start_build(struct cli_state *cli, const char *host, const char *username)
  27 {
  28         char *p;
  29
  30         /* construct a SMBsendstrt command */
  31         memset(cli->outbuf,'\0',smb_size);
  32         cli_set_message(cli->outbuf,0,0,True);
  33         SCVAL(cli->outbuf,smb_com,SMBsendstrt);
  34         SSVAL(cli->outbuf,smb_tid,cli->cnum);
  35         cli_setup_packet(cli);
  36
  37         p = smb_buf(cli->outbuf);
  38         *p++ = 4;
  39         p += clistr_push(cli, p, username,
  40                         cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_ASCII|STR_TERMINATE);
  41         *p++ = 4;
  42         p += clistr_push(cli, p, host,
  43                         cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_ASCII|STR_TERMINATE);
  44
  45         cli_setup_bcc(cli, p);
  46
  47         return(PTR_DIFF(p, cli->outbuf));
  48 }
  49
  50 bool cli_message_start(struct cli_state *cli, const char *host, const char *username,
  51                               int *grp)
  52 {
  53         cli_message_start_build(cli, host, username);
  54         cli_send_smb(cli);
  55
  56         if (!cli_receive_smb(cli)) {
  57                 return False;
  58         }
  59
  60         if (cli_is_error(cli)) return False;
  61
  62         *grp = SVAL(cli->inbuf,smb_vwv0);
  63
  64         return True;
  65 }
  66
  67 /****************************************************************************
  68  Send a message
  69 ****************************************************************************/
  70
  71 int cli_message_text_build(struct cli_state *cli, const char *msg, int len, int grp)
  72 {
  73         char *msgdos;
  74         int lendos;
  75         char *p;
  76
  77         memset(cli->outbuf,'\0',smb_size);
  78         cli_set_message(cli->outbuf,1,0,True);
  79         SCVAL(cli->outbuf,smb_com,SMBsendtxt);
  80         SSVAL(cli->outbuf,smb_tid,cli->cnum);
  81         cli_setup_packet(cli);
  82
  83         SSVAL(cli->outbuf,smb_vwv0,grp);
  84
  85         p = smb_buf(cli->outbuf);
  86         *p++ = 1;
  87
  88         if ((lendos = (int)convert_string_allocate(NULL,CH_UNIX, CH_DOS, msg,len, (void **)(void *)&msgdos, True)) < 0 || !msgdos) {
  89                 DEBUG(3,("Conversion failed, sending message in UNIX charset\n"));
  90                 SSVAL(p, 0, len); p += 2;
  91                 if (len > cli->bufsize - PTR_DIFF(p,cli->outbuf)) {
  92                         return -1;
  93                 }
  94                 memcpy(p, msg, len);
  95                 p += len;
  96         } else {
  97                 SSVAL(p, 0, lendos); p += 2;
  98                 if (lendos > cli->bufsize - PTR_DIFF(p,cli->outbuf)) {
  99                         return -1;
 100                 }
 101                 memcpy(p, msgdos, lendos);
 102                 p += lendos;
 103                 SAFE_FREE(msgdos);
 104         }
 105
 106         cli_setup_bcc(cli, p);
 107
 108         return(PTR_DIFF(p, cli->outbuf));
 109 }
 110
 111 bool cli_message_text(struct cli_state *cli, const char *msg, int len, int grp)
 112 {
 113         cli_message_text_build(cli, msg, len, grp);
 114
 115         cli_send_smb(cli);
 116
 117         if (!cli_receive_smb(cli)) {
 118                 return False;
 119         }
 120
 121         if (cli_is_error(cli)) return False;
 122
 123         return True;
 124 }
 125
 126 /****************************************************************************
 127  End a message.
 128 ****************************************************************************/
 129
 130 int cli_message_end_build(struct cli_state *cli, int grp)
 131 {
 132         char *p;
 133
 134         memset(cli->outbuf,'\0',smb_size);
 135         cli_set_message(cli->outbuf,1,0,True);
 136         SCVAL(cli->outbuf,smb_com,SMBsendend);
 137         SSVAL(cli->outbuf,smb_tid,cli->cnum);
 138
 139         SSVAL(cli->outbuf,smb_vwv0,grp);
 140
 141         cli_setup_packet(cli);
 142
 143         p = smb_buf(cli->outbuf);
 144
 145         return(PTR_DIFF(p, cli->outbuf));
 146 }
 147
 148 bool cli_message_end(struct cli_state *cli, int grp)
 149 {
 150         cli_message_end_build(cli, grp);
 151
 152         cli_send_smb(cli);
 153
 154         if (!cli_receive_smb(cli)) {
 155                 return False;
 156         }
 157
 158         if (cli_is_error(cli)) return False;
 159
 160         return True;
 161 }