source/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 "realcalls.h"
  24
  25 extern pstring smbw_cwd;
  26 extern fstring smbw_prefix;
  27
  28 static struct smbw_dir *smbw_dirs;
  29
  30 extern struct bitmap *smbw_file_bmap;
  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         SAFE_FREE(dir->list);
  67         SAFE_FREE(dir->path);
  68         ZERO_STRUCTP(dir);
  69         SAFE_FREE(dir);
  70 }
  71
  72
  73 static struct smbw_dir *cur_dir;
  74
  75 /*****************************************************
  76 add a entry to a directory listing
  77 *******************************************************/
  78 static void smbw_dir_add(struct file_info *finfo, const char *mask, void *state)
  79 {
  80         DEBUG(5,("%s\n", finfo->name));
  81
  82         if (cur_dir->malloced == cur_dir->count) {
  83                 cur_dir->list = (struct file_info *)Realloc(cur_dir->list,
  84                                                             sizeof(cur_dir->list[0])*
  85                                                             (cur_dir->count+100));
  86                 if (!cur_dir->list) {
  87                         /* oops */
  88                         return;
  89                 }
  90                 cur_dir->malloced += 100;
  91         }
  92
  93         cur_dir->list[cur_dir->count] = *finfo;
  94         cur_dir->count++;
  95 }
  96
  97 /*****************************************************
  98 add a entry to a directory listing
  99 *******************************************************/
 100 static void smbw_share_add(const char *share, uint32 type,
 101                            const char *comment, void *state)
 102 {
 103         struct file_info finfo;
 104
 105         if (strcmp(share,"IPC$") == 0) return;
 106
 107         ZERO_STRUCT(finfo);
 108
 109         pstrcpy(finfo.name, share);
 110         finfo.mode = aRONLY | aDIR;
 111
 112         smbw_dir_add(&finfo, NULL, NULL);
 113 }
 114
 115
 116 /*****************************************************
 117 add a server to a directory listing
 118 *******************************************************/
 119 static void smbw_server_add(const char *name, uint32 type,
 120                             const char *comment, void *state)
 121 {
 122         struct file_info finfo;
 123
 124         ZERO_STRUCT(finfo);
 125
 126         pstrcpy(finfo.name, name);
 127         finfo.mode = aRONLY | aDIR;
 128
 129         smbw_dir_add(&finfo, NULL, NULL);
 130 }
 131
 132
 133 /*****************************************************
 134 add a entry to a directory listing
 135 *******************************************************/
 136 static void smbw_printjob_add(struct print_job_info *job)
 137 {
 138         struct file_info finfo;
 139
 140         ZERO_STRUCT(finfo);
 141
 142         pstrcpy(finfo.name, job->name);
 143         finfo.mode = aRONLY | aDIR;
 144         finfo.mtime = job->t;
 145         finfo.atime = job->t;
 146         finfo.ctime = job->t;
 147         finfo.uid = nametouid(job->user);
 148         finfo.mode = aRONLY;
 149         finfo.size = job->size;
 150
 151         smbw_dir_add(&finfo, NULL, NULL);
 152 }
 153
 154
 155 /*****************************************************
 156 open a directory on the server
 157 *******************************************************/
 158 int smbw_dir_open(const char *fname)
 159 {
 160         fstring server, share;
 161         pstring path;
 162         struct smbw_server *srv=NULL;
 163         struct smbw_dir *dir=NULL;
 164         pstring mask;
 165         int fd;
 166         char *s, *p;
 167
 168         if (!fname) {
 169                 errno = EINVAL;
 170                 return -1;
 171         }
 172
 173         smbw_init();
 174
 175         /* work out what server they are after */
 176         s = smbw_parse_path(fname, server, share, path);
 177
 178         DEBUG(4,("dir_open share=%s\n", share));
 179
 180         /* get a connection to the server */
 181         srv = smbw_server(server, share);
 182         if (!srv) {
 183                 /* smbw_server sets errno */
 184                 goto failed;
 185         }
 186
 187         dir = (struct smbw_dir *)malloc(sizeof(*dir));
 188         if (!dir) {
 189                 errno = ENOMEM;
 190                 goto failed;
 191         }
 192
 193         ZERO_STRUCTP(dir);
 194
 195         cur_dir = dir;
 196
 197         slprintf(mask, sizeof(mask)-1, "%s\\*", path);
 198         all_string_sub(mask,"\\\\","\\",0);
 199
 200         if ((p=strstr(srv->server_name,"#01"))) {
 201                 *p = 0;
 202                 smbw_server_add(".",0,"",NULL);
 203                 smbw_server_add("..",0,"",NULL);
 204                 cli_NetServerEnum(&srv->cli, srv->server_name, SV_TYPE_DOMAIN_ENUM,
 205                                   smbw_server_add,NULL);
 206                 *p = '#';
 207         } else if ((p=strstr(srv->server_name,"#1D"))) {
 208                 DEBUG(4,("doing NetServerEnum\n"));
 209                 *p = 0;
 210                 smbw_server_add(".",0,"",NULL);
 211                 smbw_server_add("..",0,"",NULL);
 212                 cli_NetServerEnum(&srv->cli, srv->server_name, SV_TYPE_ALL,
 213                                   smbw_server_add,NULL);
 214                 *p = '#';
 215         } else if (strcmp(srv->cli.dev,"IPC") == 0) {
 216                 DEBUG(4,("doing NetShareEnum\n"));
 217                 smbw_share_add(".",0,"",NULL);
 218                 smbw_share_add("..",0,"",NULL);
 219                 if (cli_RNetShareEnum(&srv->cli, smbw_share_add,NULL) < 0) {
 220                         errno = smbw_errno(&srv->cli);
 221                         goto failed;
 222                 }
 223         } else if (strncmp(srv->cli.dev,"LPT",3) == 0) {
 224                 smbw_share_add(".",0,"",NULL);
 225                 smbw_share_add("..",0,"",NULL);
 226                 if (cli_print_queue(&srv->cli, smbw_printjob_add) < 0) {
 227                         errno = smbw_errno(&srv->cli);
 228                         goto failed;
 229                 }
 230         } else {
 231 #if 0
 232                 if (strcmp(path,"\\") == 0) {
 233                         smbw_share_add(".",0,"",NULL);
 234                         smbw_share_add("..",0,"",NULL);
 235                 }
 236 #endif
 237                 if (cli_list(&srv->cli, mask, aHIDDEN|aSYSTEM|aDIR, smbw_dir_add,NULL) < 0) {
 238                         errno = smbw_errno(&srv->cli);
 239                         goto failed;
 240                 }
 241         }
 242
 243         cur_dir = NULL;
 244
 245         fd = open(SMBW_DUMMY, O_WRONLY);
 246         if (fd == -1) {
 247                 errno = EMFILE;
 248                 goto failed;
 249         }
 250
 251         if (bitmap_query(smbw_file_bmap, fd)) {
 252                 DEBUG(0,("ERROR: fd used in smbw_dir_open\n"));
 253                 errno = EIO;
 254                 goto failed;
 255         }
 256
 257         DLIST_ADD(smbw_dirs, dir);
 258
 259         bitmap_set(smbw_file_bmap, fd);
 260
 261         dir->fd = fd;
 262         dir->srv = srv;
 263         dir->path = strdup(s);
 264
 265         DEBUG(4,("  -> %d\n", dir->count));
 266
 267         return dir->fd;
 268
 269  failed:
 270         if (dir) {
 271                 free_dir(dir);
 272         }
 273
 274         return -1;
 275 }
 276
 277 /*****************************************************
 278 a wrapper for fstat() on a directory
 279 *******************************************************/
 280 int smbw_dir_fstat(int fd, struct stat *st)
 281 {
 282         struct smbw_dir *dir;
 283
 284         dir = smbw_dir(fd);
 285         if (!dir) {
 286                 errno = EBADF;
 287                 return -1;
 288         }
 289
 290         ZERO_STRUCTP(st);
 291
 292         smbw_setup_stat(st, "", dir->count*DIRP_SIZE, aDIR);
 293
 294         st->st_dev = dir->srv->dev;
 295
 296         return 0;
 297 }
 298
 299 /*****************************************************
 300 close a directory handle
 301 *******************************************************/
 302 int smbw_dir_close(int fd)
 303 {
 304         struct smbw_dir *dir;
 305
 306         dir = smbw_dir(fd);
 307         if (!dir) {
 308                 errno = EBADF;
 309                 return -1;
 310         }
 311
 312         bitmap_clear(smbw_file_bmap, dir->fd);
 313         close(dir->fd);
 314
 315         DLIST_REMOVE(smbw_dirs, dir);
 316
 317         free_dir(dir);
 318
 319         return 0;
 320 }
 321
 322 /*****************************************************
 323 a wrapper for getdents()
 324 *******************************************************/
 325 int smbw_getdents(unsigned int fd, struct dirent *dirp, int count)
 326 {
 327         struct smbw_dir *dir;
 328         int n=0;
 329
 330         smbw_busy++;
 331
 332         dir = smbw_dir(fd);
 333         if (!dir) {
 334                 errno = EBADF;
 335                 smbw_busy--;
 336                 return -1;
 337         }
 338
 339         while (count>=DIRP_SIZE && (dir->offset < dir->count)) {
 340 #if HAVE_DIRENT_D_OFF
 341                 dirp->d_off = (dir->offset+1)*DIRP_SIZE;
 342 #endif
 343                 dirp->d_reclen = DIRP_SIZE;
 344                 fstrcpy(&dirp->d_name[0], dir->list[dir->offset].name);
 345                 dirp->d_ino = smbw_inode(dir->list[dir->offset].name);
 346                 dir->offset++;
 347                 count -= dirp->d_reclen;
 348 #if HAVE_DIRENT_D_OFF
 349                 if (dir->offset == dir->count) {
 350                         dirp->d_off = -1;
 351                 }
 352 #endif
 353                 dirp = (struct dirent *)(((char *)dirp) + DIRP_SIZE);
 354                 n++;
 355         }
 356
 357         smbw_busy--;
 358         return n*DIRP_SIZE;
 359 }
 360
 361
 362 /*****************************************************
 363 a wrapper for chdir()
 364 *******************************************************/
 365 int smbw_chdir(const char *name)
 366 {
 367         struct smbw_server *srv;
 368         fstring server, share;
 369         pstring path;
 370         uint16 mode = aDIR;
 371         char *cwd;
 372         int len;
 373
 374         smbw_init();
 375
 376         len = strlen(smbw_prefix);
 377
 378         if (smbw_busy) return real_chdir(name);
 379
 380         smbw_busy++;
 381
 382         if (!name) {
 383                 errno = EINVAL;
 384                 goto failed;
 385         }
 386
 387         DEBUG(4,("smbw_chdir(%s)\n", name));
 388
 389         /* work out what server they are after */
 390         cwd = smbw_parse_path(name, server, share, path);
 391
 392         /* a special case - accept cd to /smb */
 393         if (strncmp(cwd, smbw_prefix, len-1) == 0 &&
 394             cwd[len-1] == 0) {
 395                 goto success1;
 396         }
 397
 398         if (strncmp(cwd,smbw_prefix,strlen(smbw_prefix))) {
 399                 if (real_chdir(cwd) == 0) {
 400                         goto success2;
 401                 }
 402                 goto failed;
 403         }
 404
 405         /* get a connection to the server */
 406         srv = smbw_server(server, share);
 407         if (!srv) {
 408                 /* smbw_server sets errno */
 409                 goto failed;
 410         }
 411
 412         if (strncmp(srv->cli.dev,"IPC",3) &&
 413             strncmp(srv->cli.dev,"LPT",3) &&
 414             !smbw_getatr(srv, path,
 415                          &mode, NULL, NULL, NULL, NULL, NULL)) {
 416                 errno = smbw_errno(&srv->cli);
 417                 goto failed;
 418         }
 419
 420         if (!(mode & aDIR)) {
 421                 errno = ENOTDIR;
 422                 goto failed;
 423         }
 424
 425  success1:
 426         /* we don't want the old directory to be busy */
 427         real_chdir("/");
 428
 429  success2:
 430
 431         DEBUG(4,("set SMBW_CWD to %s\n", cwd));
 432
 433         pstrcpy(smbw_cwd, cwd);
 434
 435         smbw_busy--;
 436         return 0;
 437
 438  failed:
 439         smbw_busy--;
 440         return -1;
 441 }
 442
 443
 444 /*****************************************************
 445 a wrapper for lseek() on directories
 446 *******************************************************/
 447 off_t smbw_dir_lseek(int fd, off_t offset, int whence)
 448 {
 449         struct smbw_dir *dir;
 450         off_t ret;
 451
 452         dir = smbw_dir(fd);
 453         if (!dir) {
 454                 errno = EBADF;
 455                 return -1;
 456         }
 457
 458         switch (whence) {
 459         case SEEK_SET:
 460                 dir->offset = offset/DIRP_SIZE;
 461                 break;
 462         case SEEK_CUR:
 463                 dir->offset += offset/DIRP_SIZE;
 464                 break;
 465         case SEEK_END:
 466                 dir->offset = (dir->count * DIRP_SIZE) + offset;
 467                 dir->offset /= DIRP_SIZE;
 468                 break;
 469         }
 470
 471         ret = dir->offset * DIRP_SIZE;
 472
 473         DEBUG(4,("   -> %d\n", (int)ret));
 474
 475         return ret;
 476 }
 477
 478
 479 /*****************************************************
 480 a wrapper for mkdir()
 481 *******************************************************/
 482 int smbw_mkdir(const char *fname, mode_t mode)
 483 {
 484         struct smbw_server *srv;
 485         fstring server, share;
 486         pstring path;
 487
 488         if (!fname) {
 489                 errno = EINVAL;
 490                 return -1;
 491         }
 492
 493         smbw_init();
 494
 495         smbw_busy++;
 496
 497         /* work out what server they are after */
 498         smbw_parse_path(fname, server, share, path);
 499
 500         /* get a connection to the server */
 501         srv = smbw_server(server, share);
 502         if (!srv) {
 503                 /* smbw_server sets errno */
 504                 goto failed;
 505         }
 506
 507         if (!cli_mkdir(&srv->cli, path)) {
 508                 errno = smbw_errno(&srv->cli);
 509                 goto failed;
 510         }
 511
 512         smbw_busy--;
 513         return 0;
 514
 515  failed:
 516         smbw_busy--;
 517         return -1;
 518 }
 519
 520 /*****************************************************
 521 a wrapper for rmdir()
 522 *******************************************************/
 523 int smbw_rmdir(const char *fname)
 524 {
 525         struct smbw_server *srv;
 526         fstring server, share;
 527         pstring path;
 528
 529         if (!fname) {
 530                 errno = EINVAL;
 531                 return -1;
 532         }
 533
 534         smbw_init();
 535
 536         smbw_busy++;
 537
 538         /* work out what server they are after */
 539         smbw_parse_path(fname, server, share, path);
 540
 541         /* get a connection to the server */
 542         srv = smbw_server(server, share);
 543         if (!srv) {
 544                 /* smbw_server sets errno */
 545                 goto failed;
 546         }
 547
 548         if (!cli_rmdir(&srv->cli, path)) {
 549                 errno = smbw_errno(&srv->cli);
 550                 goto failed;
 551         }
 552
 553         smbw_busy--;
 554         return 0;
 555
 556  failed:
 557         smbw_busy--;
 558         return -1;
 559 }
 560
 561
 562 /*****************************************************
 563 a wrapper for getcwd()
 564 *******************************************************/
 565 char *smbw_getcwd(char *buf, size_t size)
 566 {
 567         smbw_init();
 568
 569         if (smbw_busy) {
 570                 return (char *)real_getcwd(buf, size);
 571         }
 572
 573         smbw_busy++;
 574
 575         if (!buf) {
 576                 if (size <= 0) size = strlen(smbw_cwd)+1;
 577                 buf = (char *)malloc(size);
 578                 if (!buf) {
 579                         errno = ENOMEM;
 580                         smbw_busy--;
 581                         return NULL;
 582                 }
 583         }
 584
 585         if (strlen(smbw_cwd) > size-1) {
 586                 errno = ERANGE;
 587                 smbw_busy--;
 588                 return NULL;
 589         }
 590
 591         safe_strcpy(buf, smbw_cwd, size);
 592
 593         smbw_busy--;
 594         return buf;
 595 }
 596
 597 /*****************************************************
 598 a wrapper for fchdir()
 599 *******************************************************/
 600 int smbw_fchdir(unsigned int fd)
 601 {
 602         struct smbw_dir *dir;
 603         int ret;
 604
 605         smbw_busy++;
 606
 607         dir = smbw_dir(fd);
 608         if (dir) {
 609                 smbw_busy--;
 610                 return chdir(dir->path);
 611         }
 612
 613         ret = real_fchdir(fd);
 614         if (ret == 0) {
 615                 sys_getwd(smbw_cwd);
 616         }
 617
 618         smbw_busy--;
 619         return ret;
 620 }
 621
 622 /*****************************************************
 623 open a directory on the server
 624 *******************************************************/
 625 DIR *smbw_opendir(const char *fname)
 626 {
 627         int fd;
 628
 629         smbw_busy++;
 630
 631         fd = smbw_dir_open(fname);
 632
 633         if (fd == -1) {
 634                 smbw_busy--;
 635                 return NULL;
 636         }
 637
 638         smbw_busy--;
 639
 640         return (DIR *)smbw_dir(fd);
 641 }
 642
 643 /*****************************************************
 644 read one entry from a directory
 645 *******************************************************/
 646 struct dirent *smbw_readdir(DIR *dirp)
 647 {
 648         struct smbw_dir *d = (struct smbw_dir *)dirp;
 649         static union {
 650                 char buf[DIRP_SIZE];
 651                 struct dirent de;
 652         } dbuf;
 653
 654         if (smbw_getdents(d->fd, &dbuf.de, DIRP_SIZE) > 0)
 655                 return &dbuf.de;
 656
 657         return NULL;
 658 }
 659
 660 /*****************************************************
 661 close a DIR*
 662 *******************************************************/
 663 int smbw_closedir(DIR *dirp)
 664 {
 665         struct smbw_dir *d = (struct smbw_dir *)dirp;
 666         return smbw_close(d->fd);
 667 }
 668
 669 /*****************************************************
 670 seek in a directory
 671 *******************************************************/
 672 void smbw_seekdir(DIR *dirp, off_t offset)
 673 {
 674         struct smbw_dir *d = (struct smbw_dir *)dirp;
 675         smbw_dir_lseek(d->fd,offset, SEEK_SET);
 676 }
 677
 678 /*****************************************************
 679 current loc in a directory
 680 *******************************************************/
 681 off_t smbw_telldir(DIR *dirp)
 682 {
 683         struct smbw_dir *d = (struct smbw_dir *)dirp;
 684         return smbw_dir_lseek(d->fd,0,SEEK_CUR);
 685 }
 686