source3/smbwrapper/smbw_dir.c

   1 /*
   2    Unix SMB/Netbios implementation.
   3    Version 2.0
   4    SMB wrapper directory functions
   5    Copyright (C) Andrew Tridgell 1998
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 2 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program; if not, write to the Free Software
  19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include "includes.h"
  23 #include "wrapper.h"
  24
  25 extern pstring smb_cwd;
  26
  27 static struct smbw_dir *smbw_dirs;
  28
  29 extern struct bitmap *smbw_file_bmap;
  30 extern int DEBUGLEVEL;
  31
  32 extern int smbw_busy;
  33
  34 /*****************************************************
  35 map a fd to a smbw_dir structure
  36 *******************************************************/
  37 struct smbw_dir *smbw_dir(int fd)
  38 {
  39         struct smbw_dir *dir;
  40
  41         for (dir=smbw_dirs;dir;dir=dir->next) {
  42                 if (dir->fd == fd) return dir;
  43         }
  44         return NULL;
  45 }
  46
  47 /*****************************************************
  48 check if a DIR* is one of ours
  49 *******************************************************/
  50 int smbw_dirp(DIR *dirp)
  51 {
  52         struct smbw_dir *d = (struct smbw_dir *)dirp;
  53         struct smbw_dir *dir;
  54
  55         for (dir=smbw_dirs;dir;dir=dir->next) {
  56                 if (dir == d) return 1;
  57         }
  58         return 0;
  59 }
  60
  61 /*****************************************************
  62 free a smbw_dir structure and all entries
  63 *******************************************************/
  64 static void free_dir(struct smbw_dir *dir)
  65 {
  66         if (dir->list) {
  67                 free(dir->list);
  68         }
  69         if (dir->path) free(dir->path);
  70         ZERO_STRUCTP(dir);
  71         free(dir);
  72 }
  73
  74
  75 static struct smbw_dir *cur_dir;
  76
  77 /*****************************************************
  78 add a entry to a directory listing
  79 *******************************************************/
  80 static void smbw_dir_add(struct file_info *finfo)
  81 {
  82         DEBUG(5,("%s\n", finfo->name));
  83
  84         if (cur_dir->malloced == cur_dir->count) {
  85                 cur_dir->list = (struct file_info *)Realloc(cur_dir->list,
  86                                                             sizeof(cur_dir->list[0])*
  87                                                             (cur_dir->count+100));
  88                 if (!cur_dir->list) {
  89                         /* oops */
  90                         return;
  91                 }
  92                 cur_dir->malloced += 100;
  93         }
  94
  95         cur_dir->list[cur_dir->count] = *finfo;
  96         cur_dir->count++;
  97 }
  98
  99 /*****************************************************
 100 add a entry to a directory listing
 101 *******************************************************/
 102 static void smbw_share_add(const char *share, uint32 type, const char *comment)
 103 {
 104         struct file_info finfo;
 105
 106         ZERO_STRUCT(finfo);
 107
 108         pstrcpy(finfo.name, share);
 109         finfo.mode = aRONLY | aDIR;
 110
 111         smbw_dir_add(&finfo);
 112 }
 113
 114
 115 /*****************************************************
 116 open a directory on the server
 117 *******************************************************/
 118 int smbw_dir_open(const char *fname)
 119 {
 120         fstring server, share;
 121         pstring path;
 122         struct smbw_server *srv=NULL;
 123         struct smbw_dir *dir=NULL;
 124         pstring mask;
 125         int fd;
 126         char *s;
 127
 128         DEBUG(4,("%s\n", __FUNCTION__));
 129
 130         if (!fname) {
 131                 errno = EINVAL;
 132                 return -1;
 133         }
 134
 135         smbw_init();
 136
 137         /* work out what server they are after */
 138         s = smbw_parse_path(fname, server, share, path);
 139
 140         DEBUG(4,("dir_open share=%s\n", share));
 141
 142         /* get a connection to the server */
 143         srv = smbw_server(server, share);
 144         if (!srv) {
 145                 /* smbw_server sets errno */
 146                 goto failed;
 147         }
 148
 149         dir = (struct smbw_dir *)malloc(sizeof(*dir));
 150         if (!dir) {
 151                 errno = ENOMEM;
 152                 goto failed;
 153         }
 154
 155         ZERO_STRUCTP(dir);
 156
 157         cur_dir = dir;
 158
 159         slprintf(mask, sizeof(mask)-1, "%s\\*", path);
 160         string_sub(mask,"\\\\","\\");
 161
 162         if (strcmp(share,"IPC$") == 0) {
 163                 DEBUG(4,("doing NetShareEnum\n"));
 164                 if (cli_RNetShareEnum(&srv->cli, smbw_share_add) <= 0) {
 165                         errno = smbw_errno(&srv->cli);
 166                         goto failed;
 167                 }
 168         } else {
 169                 if (cli_list(&srv->cli, mask, aHIDDEN|aSYSTEM|aDIR,
 170                              smbw_dir_add) <= 0) {
 171                         errno = smbw_errno(&srv->cli);
 172                         goto failed;
 173                 }
 174         }
 175
 176         cur_dir = NULL;
 177
 178         fd = bitmap_find(smbw_file_bmap, 0);
 179         if (fd == -1) {
 180                 errno = EMFILE;
 181                 goto failed;
 182         }
 183
 184         DLIST_ADD(smbw_dirs, dir);
 185
 186         bitmap_set(smbw_file_bmap, fd);
 187
 188         dir->fd = fd + SMBW_FD_OFFSET;
 189         dir->srv = srv;
 190         dir->path = strdup(s);
 191
 192         DEBUG(4,("  -> %d\n", dir->count));
 193
 194         return dir->fd;
 195
 196  failed:
 197         if (dir) {
 198                 free_dir(dir);
 199         }
 200
 201         return -1;
 202 }
 203
 204 /*****************************************************
 205 a wrapper for fstat() on a directory
 206 *******************************************************/
 207 int smbw_dir_fstat(int fd, struct stat *st)
 208 {
 209         struct smbw_dir *dir;
 210
 211         DEBUG(4,("%s\n", __FUNCTION__));
 212
 213         dir = smbw_dir(fd);
 214         if (!dir) {
 215                 errno = EBADF;
 216                 return -1;
 217         }
 218
 219         ZERO_STRUCTP(st);
 220
 221         smbw_setup_stat(st, "", dir->count*sizeof(struct dirent), aDIR);
 222
 223         st->st_dev = dir->srv->dev;
 224
 225         return 0;
 226 }
 227
 228 /*****************************************************
 229 close a directory handle
 230 *******************************************************/
 231 int smbw_dir_close(int fd)
 232 {
 233         struct smbw_dir *dir;
 234
 235         DEBUG(4,("%s\n", __FUNCTION__));
 236
 237         dir = smbw_dir(fd);
 238         if (!dir) {
 239                 DEBUG(4,("%s(%d)\n", __FUNCTION__, __LINE__));
 240                 errno = EBADF;
 241                 return -1;
 242         }
 243
 244         bitmap_clear(smbw_file_bmap, dir->fd - SMBW_FD_OFFSET);
 245
 246         DLIST_REMOVE(smbw_dirs, dir);
 247
 248         free_dir(dir);
 249
 250         return 0;
 251 }
 252
 253
 254 /*****************************************************
 255 a wrapper for getdents()
 256 *******************************************************/
 257 int smbw_getdents(unsigned int fd, struct dirent *dirp, int count)
 258 {
 259         struct smbw_dir *dir;
 260         int n=0;
 261
 262         DEBUG(4,("%s\n", __FUNCTION__));
 263
 264         smbw_busy++;
 265
 266         dir = smbw_dir(fd);
 267         if (!dir) {
 268                 errno = EBADF;
 269                 smbw_busy--;
 270                 return -1;
 271         }
 272
 273         while (count>=sizeof(*dirp) && (dir->offset < dir->count)) {
 274                 dirp->d_off = (dir->offset+1)*sizeof(*dirp);
 275                 dirp->d_reclen = sizeof(*dirp);
 276                 safe_strcpy(&dirp->d_name[0], dir->list[dir->offset].name,
 277                             sizeof(dirp->d_name)-1);
 278                 dirp->d_ino = smbw_inode(dir->list[dir->offset].name);
 279                 dir->offset++;
 280                 count -= dirp->d_reclen;
 281                 if (dir->offset == dir->count) {
 282                         dirp->d_off = -1;
 283                 }
 284                 dirp++;
 285                 n++;
 286         }
 287
 288         smbw_busy--;
 289         return n*sizeof(*dirp);
 290 }
 291
 292
 293 /*****************************************************
 294 a wrapper for chdir()
 295 *******************************************************/
 296 int smbw_chdir(const char *name)
 297 {
 298         struct smbw_server *srv;
 299         fstring server, share;
 300         pstring path;
 301         uint32 mode = aDIR;
 302         char *cwd;
 303
 304         smbw_init();
 305
 306         if (smbw_busy) return real_chdir(cwd);
 307
 308         smbw_busy++;
 309
 310         if (!name) {
 311                 errno = EINVAL;
 312                 goto failed;
 313         }
 314
 315         DEBUG(4,("%s (%s)\n", __FUNCTION__, name));
 316
 317         /* work out what server they are after */
 318         cwd = smbw_parse_path(name, server, share, path);
 319
 320         if (strncmp(cwd,SMBW_PREFIX,strlen(SMBW_PREFIX))) {
 321                 if (real_chdir(cwd) == 0) {
 322                         DEBUG(4,("set SMBW_CWD to %s\n", cwd));
 323                         pstrcpy(smb_cwd, cwd);
 324                         if (setenv(SMBW_PWD_ENV, smb_cwd, 1)) {
 325                                 DEBUG(4,("setenv failed\n"));
 326                         }
 327                         goto success;
 328                 }
 329                 errno = ENOENT;
 330                 goto failed;
 331         }
 332
 333         /* get a connection to the server */
 334         srv = smbw_server(server, share);
 335         if (!srv) {
 336                 /* smbw_server sets errno */
 337                 goto failed;
 338         }
 339
 340         if (strcmp(share,"IPC$") &&
 341             !smbw_getatr(srv, path,
 342                          &mode, NULL, NULL, NULL, NULL)) {
 343                 errno = smbw_errno(&srv->cli);
 344                 goto failed;
 345         }
 346
 347         if (!(mode & aDIR)) {
 348                 errno = ENOTDIR;
 349                 goto failed;
 350         }
 351
 352         DEBUG(4,("set SMBW_CWD2 to %s\n", cwd));
 353         pstrcpy(smb_cwd, cwd);
 354         if (setenv(SMBW_PWD_ENV, smb_cwd, 1)) {
 355                 DEBUG(4,("setenv failed\n"));
 356         }
 357
 358         /* we don't want the old directory to be busy */
 359         real_chdir("/");
 360
 361  success:
 362         smbw_busy--;
 363         return 0;
 364
 365  failed:
 366         smbw_busy--;
 367         return -1;
 368 }
 369
 370
 371 /*****************************************************
 372 a wrapper for lseek() on directories
 373 *******************************************************/
 374 off_t smbw_dir_lseek(int fd, off_t offset, int whence)
 375 {
 376         struct smbw_dir *dir;
 377         off_t ret;
 378
 379         DEBUG(4,("%s offset=%d whence=%d\n", __FUNCTION__,
 380                  (int)offset, whence));
 381
 382         dir = smbw_dir(fd);
 383         if (!dir) {
 384                 errno = EBADF;
 385                 return -1;
 386         }
 387
 388         switch (whence) {
 389         case SEEK_SET:
 390                 dir->offset = offset/sizeof(struct dirent);
 391                 break;
 392         case SEEK_CUR:
 393                 dir->offset += offset/sizeof(struct dirent);
 394                 break;
 395         case SEEK_END:
 396                 dir->offset = (dir->count * sizeof(struct dirent)) + offset;
 397                 dir->offset /= sizeof(struct dirent);
 398                 break;
 399         }
 400
 401         ret = dir->offset * sizeof(struct dirent);
 402
 403         DEBUG(4,("   -> %d\n", (int)ret));
 404
 405         return ret;
 406 }
 407
 408
 409 /*****************************************************
 410 a wrapper for mkdir()
 411 *******************************************************/
 412 int smbw_mkdir(const char *fname, mode_t mode)
 413 {
 414         struct smbw_server *srv;
 415         fstring server, share;
 416         pstring path;
 417
 418         DEBUG(4,("%s (%s)\n", __FUNCTION__, fname));
 419
 420         if (!fname) {
 421                 errno = EINVAL;
 422                 return -1;
 423         }
 424
 425         smbw_init();
 426
 427         smbw_busy++;
 428
 429         /* work out what server they are after */
 430         smbw_parse_path(fname, server, share, path);
 431
 432         /* get a connection to the server */
 433         srv = smbw_server(server, share);
 434         if (!srv) {
 435                 /* smbw_server sets errno */
 436                 goto failed;
 437         }
 438
 439         if (!cli_mkdir(&srv->cli, path)) {
 440                 errno = smbw_errno(&srv->cli);
 441                 goto failed;
 442         }
 443
 444         smbw_busy--;
 445         return 0;
 446
 447  failed:
 448         smbw_busy--;
 449         return -1;
 450 }
 451
 452 /*****************************************************
 453 a wrapper for rmdir()
 454 *******************************************************/
 455 int smbw_rmdir(const char *fname)
 456 {
 457         struct smbw_server *srv;
 458         fstring server, share;
 459         pstring path;
 460
 461         DEBUG(4,("%s (%s)\n", __FUNCTION__, fname));
 462
 463         if (!fname) {
 464                 errno = EINVAL;
 465                 return -1;
 466         }
 467
 468         smbw_init();
 469
 470         smbw_busy++;
 471
 472         /* work out what server they are after */
 473         smbw_parse_path(fname, server, share, path);
 474
 475         /* get a connection to the server */
 476         srv = smbw_server(server, share);
 477         if (!srv) {
 478                 /* smbw_server sets errno */
 479                 goto failed;
 480         }
 481
 482         if (!cli_rmdir(&srv->cli, path)) {
 483                 errno = smbw_errno(&srv->cli);
 484                 goto failed;
 485         }
 486
 487         smbw_busy--;
 488         return 0;
 489
 490  failed:
 491         smbw_busy--;
 492         return -1;
 493 }
 494
 495
 496 /*****************************************************
 497 a wrapper for getcwd()
 498 *******************************************************/
 499 char *smbw_getcwd(char *buf, size_t size)
 500 {
 501         smbw_init();
 502
 503         if (smbw_busy) {
 504                 return __getcwd(buf, size);
 505         }
 506
 507         smbw_busy++;
 508
 509         if (!buf) {
 510                 if (size <= 0) size = strlen(smb_cwd)+1;
 511                 buf = (char *)malloc(size);
 512                 if (!buf) {
 513                         errno = ENOMEM;
 514                         smbw_busy--;
 515                         return NULL;
 516                 }
 517         }
 518
 519         if (strlen(smb_cwd) > size-1) {
 520                 errno = ERANGE;
 521                 smbw_busy--;
 522                 return NULL;
 523         }
 524
 525         safe_strcpy(buf, smb_cwd, size);
 526
 527         smbw_busy--;
 528         return buf;
 529 }
 530
 531 /*****************************************************
 532 a wrapper for fchdir()
 533 *******************************************************/
 534 int smbw_fchdir(unsigned int fd)
 535 {
 536         struct smbw_dir *dir;
 537
 538         DEBUG(4,("%s\n", __FUNCTION__));
 539
 540         smbw_busy++;
 541
 542         dir = smbw_dir(fd);
 543         if (!dir) {
 544                 errno = EBADF;
 545                 smbw_busy--;
 546                 return -1;
 547         }
 548
 549         smbw_busy--;
 550
 551         return chdir(dir->path);
 552 }
 553
 554 /*****************************************************
 555 open a directory on the server
 556 *******************************************************/
 557 DIR *smbw_opendir(const char *fname)
 558 {
 559         int fd;
 560
 561         smbw_busy++;
 562
 563         fd = smbw_dir_open(fname);
 564
 565         if (fd == -1) {
 566                 smbw_busy--;
 567                 return NULL;
 568         }
 569
 570         smbw_busy--;
 571
 572         return (DIR *)smbw_dir(fd);
 573 }
 574
 575 /*****************************************************
 576 read one entry from a directory
 577 *******************************************************/
 578 struct dirent *smbw_readdir(DIR *dirp)
 579 {
 580         struct smbw_dir *d = (struct smbw_dir *)dirp;
 581         static struct dirent de;
 582
 583         if (smbw_getdents(d->fd, &de, sizeof(struct dirent)) > 0)
 584                 return &de;
 585
 586         return NULL;
 587 }
 588
 589 /*****************************************************
 590 close a DIR*
 591 *******************************************************/
 592 int smbw_closedir(DIR *dirp)
 593 {
 594         struct smbw_dir *d = (struct smbw_dir *)dirp;
 595         return smbw_close(d->fd);
 596 }
 597
 598 /*****************************************************
 599 seek in a directory
 600 *******************************************************/
 601 void smbw_seekdir(DIR *dirp, off_t offset)
 602 {
 603         struct smbw_dir *d = (struct smbw_dir *)dirp;
 604         smbw_dir_lseek(d->fd,offset, SEEK_SET);
 605 }
 606
 607 /*****************************************************
 608 current loc in a directory
 609 *******************************************************/
 610 off_t smbw_telldir(DIR *dirp)
 611 {
 612         struct smbw_dir *d = (struct smbw_dir *)dirp;
 613         return smbw_dir_lseek(d->fd,0,SEEK_CUR);
 614 }