source/utils/nbio.c

   1 #define NBDEBUG 0
   2
   3 /*
   4    Unix SMB/Netbios implementation.
   5    Version 1.9.
   6    SMB torture tester
   7    Copyright (C) Andrew Tridgell 1997-1998
   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 #define MAX_FILES 1000
  29
  30 static char buf[70000];
  31 extern int line_count;
  32
  33 static struct {
  34         int fd;
  35         int handle;
  36 } ftable[MAX_FILES];
  37
  38 static struct cli_state *c;
  39
  40 static void sigsegv(int sig)
  41 {
  42         char line[200];
  43         printf("segv at line %d\n", line_count);
  44         slprintf(line, sizeof(line), "/usr/X11R6/bin/xterm -e gdb /proc/%d/exe %d",
  45                 (int)getpid(), (int)getpid());
  46         system(line);
  47         exit(1);
  48 }
  49
  50 void nb_setup(struct cli_state *cli)
  51 {
  52         signal(SIGSEGV, sigsegv);
  53         c = cli;
  54 }
  55
  56
  57 void nb_unlink(char *fname)
  58 {
  59         strupper(fname);
  60
  61         if (!cli_unlink(c, fname)) {
  62 #if NBDEBUG
  63                 printf("(%d) unlink %s failed (%s)\n",
  64                        line_count, fname, cli_errstr(c));
  65 #endif
  66         }
  67 }
  68
  69 void nb_open(char *fname, int handle, int size)
  70 {
  71         int fd, i;
  72         int flags = O_RDWR|O_CREAT;
  73         size_t st_size;
  74         static int count;
  75
  76         strupper(fname);
  77
  78         if (size == 0) flags |= O_TRUNC;
  79
  80         fd = cli_open(c, fname, flags, DENY_NONE);
  81         if (fd == -1) {
  82 #if NBDEBUG
  83                 printf("(%d) open %s failed for handle %d (%s)\n",
  84                        line_count, fname, handle, cli_errstr(c));
  85 #endif
  86                 return;
  87         }
  88         cli_getattrE(c, fd, NULL, &st_size, NULL, NULL, NULL);
  89         if (size > st_size) {
  90 #if NBDEBUG
  91                 printf("(%d) needs expanding %s to %d from %d\n",
  92                        line_count, fname, size, (int)st_size);
  93 #endif
  94         } else if (size < st_size) {
  95 #if NBDEBUG
  96                 printf("(%d) needs truncating %s to %d from %d\n",
  97                        line_count, fname, size, (int)st_size);
  98 #endif
  99         }
 100         for (i=0;i<MAX_FILES;i++) {
 101                 if (ftable[i].handle == 0) break;
 102         }
 103         if (i == MAX_FILES) {
 104                 printf("file table full for %s\n", fname);
 105                 exit(1);
 106         }
 107         ftable[i].handle = handle;
 108         ftable[i].fd = fd;
 109         if (count++ % 100 == 0) {
 110                 printf(".");
 111         }
 112 }
 113
 114 void nb_write(int handle, int size, int offset)
 115 {
 116         int i;
 117
 118         if (buf[0] == 0) memset(buf, 1, sizeof(buf));
 119
 120         for (i=0;i<MAX_FILES;i++) {
 121                 if (ftable[i].handle == handle) break;
 122         }
 123         if (i == MAX_FILES) {
 124 #if NBDEBUG
 125                 printf("(%d) nb_write: handle %d was not open size=%d ofs=%d\n",
 126                        line_count, handle, size, offset);
 127 #endif
 128                 return;
 129         }
 130         if (cli_smbwrite(c, ftable[i].fd, buf, offset, size) != size) {
 131                 printf("(%d) write failed on handle %d, fd %d \
 132 errno %d (%s)\n", line_count, handle, ftable[i].fd, errno, strerror(errno));
 133                 if (errno == ENOSPC) {
 134                         printf("Halting.\n");
 135                         fflush(stdout);
 136                         fflush(stderr);
 137                         exit(3);
 138                 }
 139         }
 140 }
 141
 142 void nb_read(int handle, int size, int offset)
 143 {
 144         int i, ret;
 145
 146         for (i=0;i<MAX_FILES;i++) {
 147                 if (ftable[i].handle == handle) break;
 148         }
 149         if (i == MAX_FILES) {
 150                 printf("(%d) nb_read: handle %d was not open size=%d ofs=%d\n",
 151                        line_count, handle, size, offset);
 152                 return;
 153         }
 154         if ((ret=cli_read(c, ftable[i].fd, buf, offset, size)) != size) {
 155 #if NBDEBUG
 156                 printf("(%d) read failed on handle %d ofs=%d size=%d res=%d fd %d errno %d (%s)\n",
 157                         line_count, handle, offset, size, ret, ftable[i].fd, errno, strerror(errno));
 158 #endif
 159         }
 160 }
 161
 162 void nb_close(int handle)
 163 {
 164         int i;
 165         for (i=0;i<MAX_FILES;i++) {
 166                 if (ftable[i].handle == handle) break;
 167         }
 168         if (i == MAX_FILES) {
 169                 printf("(%d) nb_close: handle %d was not open\n",
 170                        line_count, handle);
 171                 return;
 172         }
 173         cli_close(c, ftable[i].fd);
 174         ftable[i].handle = 0;
 175 }
 176
 177 void nb_mkdir(char *fname)
 178 {
 179         strupper(fname);
 180
 181         if (!cli_mkdir(c, fname)) {
 182 #if NBDEBUG
 183                 printf("mkdir %s failed (%s)\n",
 184                        fname, cli_errstr(c));
 185 #endif
 186         }
 187 }
 188
 189 void nb_rmdir(char *fname)
 190 {
 191         strupper(fname);
 192
 193         if (!cli_rmdir(c, fname)) {
 194 #if NBDEBUG
 195                 printf("rmdir %s failed (%s)\n",
 196                        fname, cli_errstr(c));
 197 #endif
 198         }
 199 }
 200
 201 void nb_rename(char *old, char *new)
 202 {
 203         strupper(old);
 204         strupper(new);
 205
 206         if (!cli_rename(c, old, new)) {
 207 #if NBDEBUG
 208                 printf("rename %s %s failed (%s)\n",
 209                        old, new, cli_errstr(c));
 210 #endif
 211         }
 212 }
 213
 214
 215 void nb_stat(char *fname, int size)
 216 {
 217         size_t st_size;
 218
 219         strupper(fname);
 220
 221         if (!cli_getatr(c, fname, NULL, &st_size, NULL)) {
 222 #if NBDEBUG
 223                 printf("(%d) nb_stat: %s size=%d %s\n",
 224                        line_count, fname, size, cli_errstr(c));
 225 #endif
 226                 return;
 227         }
 228         if (st_size != size) {
 229 #if NBDEBUG
 230                 printf("(%d) nb_stat: %s wrong size %d %d\n",
 231                        line_count, fname, (int)st_size, size);
 232 #endif
 233         }
 234 }
 235
 236 void nb_create(char *fname, int size)
 237 {
 238         nb_open(fname, 5000, size);
 239         nb_close(5000);
 240 }