etc/afpd/catsearch.c

   1 /*
   2  * Netatalk 2002 (c)
   3  * Copyright (C) 1990, 1993 Regents of The University of Michigan
   4  * Copyright (C) 2010 Frank Lahm
   5  * All Rights Reserved. See COPYRIGHT
   6  */
   7
   8
   9 /*
  10  * This file contains FPCatSearch implementation. FPCatSearch performs
  11  * file/directory search based on specified criteria. It is used by client
  12  * to perform fast searches on (propably) big volumes. So, it has to be
  13  * pretty fast.
  14  *
  15  * This implementation bypasses most of adouble/libatalk stuff as long as
  16  * possible and does a standard filesystem search. It calls higher-level
  17  * libatalk/afpd functions only when it is really needed, mainly while
  18  * returning some non-UNIX information or filtering by non-UNIX criteria.
  19  *
  20  * Initial version written by Rafal Lewczuk <rlewczuk@pronet.pl>
  21  *
  22  * Starting with Netatalk 2.2 searching by name criteria utilizes the
  23  * CNID database in conjunction with an enhanced cnid_dbd. This requires
  24  * the use of cnidscheme:dbd for the searched volume, the new functionality
  25  * is not built into cnidscheme:cdb.
  26  */
  27
  28 #ifdef HAVE_CONFIG_H
  29 #include "config.h"
  30 #endif /* HAVE_CONFIG_H */
  31
  32 #include <sys/types.h>
  33 #include <sys/stat.h>
  34 #include <unistd.h>
  35 #include <stdio.h>
  36 #include <stdlib.h>
  37 #include <errno.h>
  38 #include <ctype.h>
  39 #include <string.h>
  40 #include <time.h>
  41 #include <string.h>
  42 #include <sys/file.h>
  43 #include <netinet/in.h>
  44
  45 #include <atalk/afp.h>
  46 #include <atalk/adouble.h>
  47 #include <atalk/logger.h>
  48 #include <atalk/cnid.h>
  49 #include <atalk/cnid_bdb_private.h>
  50 #include <atalk/util.h>
  51 #include <atalk/bstradd.h>
  52 #include <atalk/unicode.h>
  53 #include <atalk/globals.h>
  54 #include <atalk/netatalk_conf.h>
  55
  56 #include "desktop.h"
  57 #include "directory.h"
  58 #include "dircache.h"
  59 #include "file.h"
  60 #include "volume.h"
  61 #include "filedir.h"
  62 #include "fork.h"
  63
  64
  65 struct finderinfo {
  66         uint32_t f_type;
  67         uint32_t creator;
  68         uint16_t attrs;    /* File attributes (high 8 bits)*/
  69         uint16_t label;    /* Label (low 8 bits )*/
  70         char reserved[22]; /* Unknown (at least for now...) */
  71 };
  72
  73 typedef char packed_finder[ADEDLEN_FINDERI];
  74
  75 /* Known attributes:
  76  * 0x04 - has a custom icon
  77  * 0x20 - name/icon is locked
  78  * 0x40 - is invisible
  79  * 0x80 - is alias
  80  *
  81  * Known labels:
  82  * 0x02 - project 2
  83  * 0x04 - project 1
  84  * 0x06 - personal
  85  * 0x08 - cool
  86  * 0x0a - in progress
  87  * 0x0c - hot
  88  * 0x0e - essential
  89  */
  90
  91 /* This is our search-criteria structure. */
  92 struct scrit {
  93         uint32_t rbitmap;          /* Request bitmap - which values should we check ? */
  94         uint16_t fbitmap, dbitmap; /* file & directory bitmap - which values should we return ? */
  95         uint16_t attr;             /* File attributes */
  96         time_t cdate;               /* Creation date */
  97         time_t mdate;               /* Last modification date */
  98         time_t bdate;               /* Last backup date */
  99         uint32_t pdid;             /* Parent DID */
 100     uint16_t offcnt;           /* Offspring count */
 101         struct finderinfo finfo;    /* Finder info */
 102         char lname[64];             /* Long name */
 103         char utf8name[514];         /* UTF8 or UCS2 name */ /* for convert_charset dest_len parameter +2 */
 104 };
 105
 106 /*
 107  * Directory tree search is recursive by its nature. But AFP specification
 108  * requires FPCatSearch to pause after returning n results and be able to
 109  * resume the search later. So we have to do recursive search using flat
 110  * (iterative) algorithm and remember all directories to look into in an
 111  * stack-like structure. The structure below is one item of directory stack.
 112  *
 113  */
 114 struct dsitem {
 115     cnid_t ds_did;         /* CNID of this directory           */
 116     int    ds_checked;     /* Have we checked this directory ? */
 117 };
 118
 119
 120 #define DS_BSIZE 128
 121 static int save_cidx = -1; /* Saved index of currently scanned directory. */
 122 static struct dsitem *dstack = NULL; /* Directory stack data... */
 123 static int dssize = 0;               /* Directory stack (allocated) size... */
 124 static int dsidx = 0;                /* First free item index... */
 125 static struct scrit c1, c2;          /* search criteria */
 126
 127 /* Clears directory stack. */
 128 static void clearstack(void)
 129 {
 130         save_cidx = -1;
 131         while (dsidx > 0) {
 132                 dsidx--;
 133         }
 134 }
 135
 136 /* Puts new item onto directory stack. */
 137 static int addstack(char *uname, struct dir *dir, int pidx)
 138 {
 139         struct dsitem *ds;
 140     struct dsitem *tmpds = NULL;
 141
 142         /* check if we have some space on stack... */
 143         if (dsidx >= dssize) {
 144                 dssize += DS_BSIZE;
 145                 tmpds = realloc(dstack, dssize * sizeof(struct dsitem));
 146                 if (tmpds == NULL) {
 147             clearstack();
 148             free(dstack);
 149                         return -1;
 150         }
 151         dstack = tmpds;
 152         }
 153
 154         /* Put new element. Allocate and copy lname and path. */
 155         ds = dstack + dsidx++;
 156         ds->ds_did = dir->d_did;
 157         ds->ds_checked = 0;
 158         return 0;
 159 }
 160
 161 /* Removes checked items from top of directory stack. Returns index of the first unchecked elements or -1. */
 162 static int reducestack(void)
 163 {
 164         int r;
 165         if (save_cidx != -1) {
 166                 r = save_cidx;
 167                 save_cidx = -1;
 168                 return r;
 169         }
 170
 171         while (dsidx > 0) {
 172                 if (dstack[dsidx-1].ds_checked) {
 173                         dsidx--;
 174 //                      free(dstack[dsidx].path);
 175                 } else
 176                         return dsidx - 1;
 177         }
 178         return -1;
 179 }
 180
 181 /* Looks up for an opened adouble structure, opens resource fork of selected file.
 182  * FIXME What about noadouble?
 183 */
 184 static struct adouble *adl_lkup(struct vol *vol, struct path *path, struct adouble *adp)
 185 {
 186         static struct adouble ad;
 187
 188         struct ofork *of;
 189         int isdir;
 190
 191         if (adp)
 192             return adp;
 193
 194         isdir  = S_ISDIR(path->st.st_mode);
 195
 196         if (!isdir && (of = of_findname(vol, path))) {
 197                 adp = of->of_ad;
 198         } else {
 199                 ad_init(&ad, vol);
 200                 adp = &ad;
 201         }
 202
 203     if ( ad_metadata( path->u_name, ((isdir) ? ADFLAGS_DIR : 0), adp) < 0 ) {
 204         adp = NULL; /* FIXME without resource fork adl_lkup will be call again */
 205     }
 206
 207         return adp;
 208 }
 209
 210 /* -------------------- */
 211 static struct finderinfo *unpack_buffer(struct finderinfo *finfo, char *buffer)
 212 {
 213         memcpy(&finfo->f_type,  buffer +FINDERINFO_FRTYPEOFF, sizeof(finfo->f_type));
 214         memcpy(&finfo->creator, buffer +FINDERINFO_FRCREATOFF, sizeof(finfo->creator));
 215         memcpy(&finfo->attrs,   buffer +FINDERINFO_FRFLAGOFF, sizeof(finfo->attrs));
 216         memcpy(&finfo->label,   buffer +FINDERINFO_FRFLAGOFF, sizeof(finfo->label));
 217         finfo->attrs &= 0xff00; /* high 8 bits */
 218         finfo->label &= 0xff;   /* low 8 bits */
 219
 220         return finfo;
 221 }
 222
 223 /* -------------------- */
 224 static struct finderinfo *
 225 unpack_finderinfo(struct vol *vol, struct path *path, struct adouble **adp, struct finderinfo *finfo, int islnk)
 226 {
 227         packed_finder  buf;
 228         void           *ptr;
 229
 230     *adp = adl_lkup(vol, path, *adp);
 231         ptr = get_finderinfo(vol, path->u_name, *adp, &buf,islnk);
 232         return unpack_buffer(finfo, ptr);
 233 }
 234
 235 /* -------------------- */
 236 #define CATPBIT_PARTIAL 31
 237 /* Criteria checker. This function returns a 2-bit value. */
 238 /* bit 0 means if checked file meets given criteria. */
 239 /* bit 1 means if it is a directory and we should descent into it. */
 240 /* uname - UNIX name
 241  * fname - our fname (translated to UNIX)
 242  * cidx - index in directory stack
 243  */
 244 static int crit_check(struct vol *vol, struct path *path) {
 245         int result = 0;
 246         uint16_t attr, flags = CONV_PRECOMPOSE;
 247         struct finderinfo *finfo = NULL, finderinfo;
 248         struct adouble *adp = NULL;
 249         time_t c_date, b_date;
 250         uint32_t ac_date, ab_date;
 251         static char convbuf[514]; /* for convert_charset dest_len parameter +2 */
 252         size_t len;
 253     int islnk;
 254     islnk=S_ISLNK(path->st.st_mode);
 255
 256         if (S_ISDIR(path->st.st_mode)) {
 257                 if (!c1.dbitmap)
 258                         return 0;
 259         }
 260         else {
 261                 if (!c1.fbitmap)
 262                         return 0;
 263
 264                 /* compute the Mac name
 265                  * first try without id (it's slow to find it)
 266                  * An other option would be to call get_id in utompath but
 267                  * we need to pass parent dir
 268                 */
 269         if (!(path->m_name = utompath(vol, path->u_name, 0 , utf8_encoding(vol->v_obj)) )) {
 270                 /*retry with the right id */
 271
 272                 cnid_t id;
 273
 274                 adp = adl_lkup(vol, path, adp);
 275                 id = get_id(vol, adp, &path->st, path->d_dir->d_did, path->u_name, strlen(path->u_name));
 276                 if (!id) {
 277                         /* FIXME */
 278                         return 0;
 279                 }
 280                 /* save the id for getfilparm */
 281                 path->id = id;
 282                 if (!(path->m_name = utompath(vol, path->u_name, id , utf8_encoding(vol->v_obj)))) {
 283                         return 0;
 284                 }
 285         }
 286         }
 287
 288         /* Kind of optimization:
 289          * -- first check things we've already have - filename
 290          * -- last check things we get from ad_open()
 291          * FIXME strmcp strstr (icase)
 292          */
 293
 294         /* Check for filename */
 295         if ((c1.rbitmap & (1<<DIRPBIT_LNAME))) {
 296                 if ( (size_t)(-1) == (len = convert_string(vol->v_maccharset, CH_UCS2, path->m_name, -1, convbuf, 512)) )
 297                         goto crit_check_ret;
 298
 299                 if ((c1.rbitmap & (1<<CATPBIT_PARTIAL))) {
 300                         if (strcasestr_w( (ucs2_t*) convbuf, (ucs2_t*) c1.lname) == NULL)
 301                                 goto crit_check_ret;
 302                 } else
 303                         if (strcasecmp_w((ucs2_t*) convbuf, (ucs2_t*) c1.lname) != 0)
 304                                 goto crit_check_ret;
 305         }
 306
 307         if ((c1.rbitmap & (1<<FILPBIT_PDINFO))) {
 308                 if ( (size_t)(-1) == (len = convert_charset( CH_UTF8_MAC, CH_UCS2, CH_UTF8, path->m_name, strlen(path->m_name), convbuf, 512, &flags))) {
 309                         goto crit_check_ret;
 310                 }
 311
 312                 if (c1.rbitmap & (1<<CATPBIT_PARTIAL)) {
 313                         if (strcasestr_w((ucs2_t *) convbuf, (ucs2_t*)c1.utf8name) == NULL)
 314                                 goto crit_check_ret;
 315                 } else
 316                         if (strcasecmp_w((ucs2_t *)convbuf, (ucs2_t*)c1.utf8name) != 0)
 317                                 goto crit_check_ret;
 318         }
 319
 320
 321         /* FIXME */
 322         if ((unsigned)c2.mdate > 0x7fffffff)
 323                 c2.mdate = 0x7fffffff;
 324         if ((unsigned)c2.cdate > 0x7fffffff)
 325                 c2.cdate = 0x7fffffff;
 326         if ((unsigned)c2.bdate > 0x7fffffff)
 327                 c2.bdate = 0x7fffffff;
 328
 329         /* Check for modification date */
 330         if ((c1.rbitmap & (1<<DIRPBIT_MDATE))) {
 331                 if (path->st.st_mtime < c1.mdate || path->st.st_mtime > c2.mdate)
 332                         goto crit_check_ret;
 333         }
 334
 335         /* Check for creation date... */
 336         if ((c1.rbitmap & (1<<DIRPBIT_CDATE))) {
 337                 c_date = path->st.st_mtime;
 338                 adp = adl_lkup(vol, path, adp);
 339                 if (adp && ad_getdate(adp, AD_DATE_CREATE, &ac_date) >= 0)
 340                     c_date = AD_DATE_TO_UNIX(ac_date);
 341
 342                 if (c_date < c1.cdate || c_date > c2.cdate)
 343                         goto crit_check_ret;
 344         }
 345
 346         /* Check for backup date... */
 347         if ((c1.rbitmap & (1<<DIRPBIT_BDATE))) {
 348                 b_date = path->st.st_mtime;
 349                 adp = adl_lkup(vol, path, adp);
 350                 if (adp && ad_getdate(adp, AD_DATE_BACKUP, &ab_date) >= 0)
 351                         b_date = AD_DATE_TO_UNIX(ab_date);
 352
 353                 if (b_date < c1.bdate || b_date > c2.bdate)
 354                         goto crit_check_ret;
 355         }
 356
 357         /* Check attributes */
 358         if ((c1.rbitmap & (1<<DIRPBIT_ATTR)) && c2.attr != 0) {
 359                 if ((adp = adl_lkup(vol, path, adp))) {
 360                         ad_getattr(adp, &attr);
 361                         if ((attr & c2.attr) != c1.attr)
 362                                 goto crit_check_ret;
 363                 } else
 364                         goto crit_check_ret;
 365         }
 366
 367         /* Check file type ID */
 368         if ((c1.rbitmap & (1<<DIRPBIT_FINFO)) && c2.finfo.f_type != 0) {
 369             finfo = unpack_finderinfo(vol, path, &adp, &finderinfo,islnk);
 370                 if (finfo->f_type != c1.finfo.f_type)
 371                         goto crit_check_ret;
 372         }
 373
 374         /* Check creator ID */
 375         if ((c1.rbitmap & (1<<DIRPBIT_FINFO)) && c2.finfo.creator != 0) {
 376                 if (!finfo) {
 377                         finfo = unpack_finderinfo(vol, path, &adp, &finderinfo,islnk);
 378                 }
 379                 if (finfo->creator != c1.finfo.creator)
 380                         goto crit_check_ret;
 381         }
 382
 383         /* Check finder info attributes */
 384         if ((c1.rbitmap & (1<<DIRPBIT_FINFO)) && c2.finfo.attrs != 0) {
 385                 if (!finfo) {
 386                         finfo = unpack_finderinfo(vol, path, &adp, &finderinfo,islnk);
 387                 }
 388
 389                 if ((finfo->attrs & c2.finfo.attrs) != c1.finfo.attrs)
 390                         goto crit_check_ret;
 391         }
 392
 393         /* Check label */
 394         if ((c1.rbitmap & (1<<DIRPBIT_FINFO)) && c2.finfo.label != 0) {
 395                 if (!finfo) {
 396                         finfo = unpack_finderinfo(vol, path, &adp, &finderinfo,islnk);
 397                 }
 398                 if ((finfo->label & c2.finfo.label) != c1.finfo.label)
 399                         goto crit_check_ret;
 400         }
 401         /* FIXME: Attributes check ! */
 402
 403         /* All criteria are met. */
 404         result |= 1;
 405 crit_check_ret:
 406         if (adp != NULL)
 407                 ad_close(adp, ADFLAGS_HF);
 408         return result;
 409 }
 410
 411 /* ------------------------------ */
 412 static int rslt_add (const AFPObj *obj, struct vol *vol, struct path *path, char **buf, int ext)
 413 {
 414
 415         char            *p = *buf;
 416         int             ret;
 417         size_t          tbuf =0;
 418         uint16_t        resultsize;
 419         int             isdir = S_ISDIR(path->st.st_mode);
 420
 421         /* Skip resultsize */
 422         if (ext) {
 423                 p += sizeof(resultsize);
 424         }
 425         else {
 426                 p++;
 427         }
 428         *p++ = isdir ? FILDIRBIT_ISDIR : FILDIRBIT_ISFILE;    /* IsDir ? */
 429
 430         if (ext) {
 431                 *p++ = 0;                  /* Pad */
 432         }
 433
 434         if ( isdir ) {
 435         ret = getdirparams(obj, vol, c1.dbitmap, path, path->d_dir, p , &tbuf );
 436         }
 437         else {
 438             /* FIXME slow if we need the file ID, we already know it, done ? */
 439                 ret = getfilparams (obj, vol, c1.fbitmap, path, path->d_dir, p, &tbuf, 0);
 440         }
 441
 442         if ( ret != AFP_OK )
 443                 return 0;
 444
 445         /* Make sure entry length is even */
 446         if ((tbuf & 1)) {
 447            *p++ = 0;
 448            tbuf++;
 449         }
 450
 451         if (ext) {
 452                 resultsize = htons(tbuf);
 453                 memcpy ( *buf, &resultsize, sizeof(resultsize) );
 454                 *buf += tbuf + 4;
 455         }
 456         else {
 457                 **buf = tbuf;
 458                 *buf += tbuf + 2;
 459         }
 460
 461         return 1;
 462 }
 463
 464 #define VETO_STR \
 465         "./../.AppleDouble/.AppleDB/Network Trash Folder/TheVolumeSettingsFolder/TheFindByContentFolder/.AppleDesktop/.Parent/"
 466
 467 /*!
 468  * This function performs a filesystem search
 469  *
 470  * Uses globals c1, c2, the search criteria
 471  *
 472  * @param vol       (r)  volume we are searching on ...
 473  * @param dir       (rw) directory we are starting from ...
 474  * @param rmatches  (r)  maximum number of matches we can return
 475  * @param pos       (r)  position we've stopped recently
 476  * @param rbuf      (w)  output buffer
 477  * @param nrecs     (w)  number of matches
 478  * @param rsize     (w)  length of data written to output buffer
 479  * @param ext       (r)  extended search flag
 480  */
 481 #define NUM_ROUNDS 200
 482 static int catsearch(const AFPObj *obj,
 483                      struct vol *vol,
 484                      struct dir *dir,
 485                      int rmatches,
 486                      uint32_t *pos,
 487                      char *rbuf,
 488                      uint32_t *nrecs,
 489                      int *rsize,
 490                      int ext)
 491 {
 492     static uint32_t cur_pos;    /* Saved position index (ID) - used to remember "position" across FPCatSearch calls */
 493     static DIR *dirpos;                  /* UNIX structure describing currently opened directory. */
 494     struct dir *currentdir;      /* struct dir of current directory */
 495         int cidx, r;
 496         struct dirent *entry;
 497         int result = AFP_OK;
 498         int ccr;
 499     struct path path;
 500         char *vpath = vol->v_path;
 501         char *rrbuf = rbuf;
 502     time_t start_time;
 503     int num_rounds = NUM_ROUNDS;
 504     int cwd = -1;
 505     int error;
 506     int unlen;
 507
 508         if (*pos != 0 && *pos != cur_pos) {
 509                 result = AFPERR_CATCHNG;
 510                 goto catsearch_end;
 511         }
 512
 513         /* FIXME: Category "offspring count ! */
 514
 515
 516         /* We need to initialize all mandatory structures/variables and change working directory appropriate... */
 517         if (*pos == 0) {
 518                 clearstack();
 519                 if (dirpos != NULL) {
 520                         closedir(dirpos);
 521                         dirpos = NULL;
 522                 }
 523
 524                 if (addstack(vpath, dir, -1) == -1) {
 525                         result = AFPERR_MISC;
 526                         goto catsearch_end;
 527                 }
 528                 /* FIXME: Sometimes DID is given by client ! (correct this one above !) */
 529         }
 530
 531         /* Save current path */
 532     if ((cwd = open(".", O_RDONLY)) < 0) {
 533         result = AFPERR_MISC;
 534         goto catsearch_end;
 535     }
 536
 537         /* So we are beginning... */
 538     start_time = time(NULL);
 539
 540         while ((cidx = reducestack()) != -1) {
 541         if ((currentdir = dirlookup(vol, dstack[cidx].ds_did)) == NULL) {
 542             result = AFPERR_MISC;
 543             goto catsearch_end;
 544         }
 545         LOG(log_debug, logtype_afpd, "catsearch: current struct dir: \"%s\"", cfrombstr(currentdir->d_fullpath));
 546
 547                 error = movecwd(vol, currentdir);
 548
 549                 if (!error && dirpos == NULL)
 550                         dirpos = opendir(".");
 551
 552                 if (dirpos == NULL)
 553                         dirpos = opendir(cfrombstr(currentdir->d_fullpath));
 554
 555                 if (error || dirpos == NULL) {
 556                         switch (errno) {
 557                         case EACCES:
 558                                 dstack[cidx].ds_checked = 1;
 559                                 continue;
 560                         case EMFILE:
 561                         case ENFILE:
 562                         case ENOENT:
 563                                 result = AFPERR_NFILE;
 564                                 break;
 565                         case ENOMEM:
 566                         case ENOTDIR:
 567                         default:
 568                                 result = AFPERR_MISC;
 569                         } /* switch (errno) */
 570                         goto catsearch_end;
 571                 }
 572
 573
 574                 while ((entry = readdir(dirpos)) != NULL) {
 575                         (*pos)++;
 576
 577                         if (!check_dirent(vol, entry->d_name))
 578                            continue;
 579
 580             LOG(log_debug, logtype_afpd, "catsearch(\"%s\"): dirent: \"%s\"",
 581                 cfrombstr(currentdir->d_fullpath), entry->d_name);
 582
 583                         memset(&path, 0, sizeof(path));
 584                         path.u_name = entry->d_name;
 585                         if (of_stat(vol, &path) != 0) {
 586                                 switch (errno) {
 587                                 case EACCES:
 588                                 case ELOOP:
 589                                 case ENOENT:
 590                                         continue;
 591                                 case ENOTDIR:
 592                                 case EFAULT:
 593                                 case ENOMEM:
 594                                 case ENAMETOOLONG:
 595                                 default:
 596                                         result = AFPERR_MISC;
 597                                         goto catsearch_end;
 598                                 }
 599                         }
 600             switch (S_IFMT & path.st.st_mode) {
 601             case S_IFDIR:
 602                                 /* here we can short cut
 603                                    ie if in the same loop the parent dir wasn't in the cache
 604                                    ALL dirsearch_byname will fail.
 605                                 */
 606                 unlen = strlen(path.u_name);
 607                 path.d_dir = dircache_search_by_name(vol,
 608                                                      currentdir,
 609                                                      path.u_name,
 610                                                      unlen);
 611                 if (path.d_dir == NULL) {
 612                         /* path.m_name is set by adddir */
 613                     if ((path.d_dir = dir_add(vol,
 614                                               currentdir,
 615                                               &path,
 616                                               unlen)) == NULL) {
 617                                                 result = AFPERR_MISC;
 618                                                 goto catsearch_end;
 619                                         }
 620                 }
 621                 path.m_name = cfrombstr(path.d_dir->d_m_name);
 622
 623                                 if (addstack(path.u_name, path.d_dir, cidx) == -1) {
 624                                         result = AFPERR_MISC;
 625                                         goto catsearch_end;
 626                                 }
 627                 break;
 628             case S_IFREG:
 629                 path.d_dir = currentdir;
 630                 break;
 631             default:
 632                 continue;
 633             }
 634
 635                         ccr = crit_check(vol, &path);
 636
 637                         /* bit 0 means that criteria has been met */
 638                         if ((ccr & 1)) {
 639                                 r = rslt_add (obj, vol, &path, &rrbuf, ext);
 640
 641                                 if (r == 0) {
 642                                         result = AFPERR_MISC;
 643                                         goto catsearch_end;
 644                                 }
 645                                 *nrecs += r;
 646                                 /* Number of matches limit */
 647                                 if (--rmatches == 0)
 648                                         goto catsearch_pause;
 649                                 /* Block size limit */
 650                                 if (rrbuf - rbuf >= 448)
 651                                         goto catsearch_pause;
 652                         }
 653                         /* MacOS 9 doesn't like servers executing commands longer than few seconds */
 654                         if (--num_rounds <= 0) {
 655                             if (start_time != time(NULL)) {
 656                                         result=AFP_OK;
 657                                         goto catsearch_pause;
 658                             }
 659                             num_rounds = NUM_ROUNDS;
 660                         }
 661                 } /* while ((entry=readdir(dirpos)) != NULL) */
 662                 closedir(dirpos);
 663                 dirpos = NULL;
 664                 dstack[cidx].ds_checked = 1;
 665         } /* while (current_idx = reducestack()) != -1) */
 666
 667         /* We have finished traversing our tree. Return EOF here. */
 668         result = AFPERR_EOF;
 669         goto catsearch_end;
 670
 671 catsearch_pause:
 672         cur_pos = *pos;
 673         save_cidx = cidx;
 674
 675 catsearch_end: /* Exiting catsearch: error condition */
 676         *rsize = rrbuf - rbuf;
 677     if (cwd != -1) {
 678         if ((fchdir(cwd)) != 0) {
 679             LOG(log_debug, logtype_afpd, "error chdiring back: %s", strerror(errno));
 680         }
 681         close(cwd);
 682     }
 683         return result;
 684 } /* catsearch() */
 685
 686 /*!
 687  * This function performs a CNID db search
 688  *
 689  * Uses globals c1, c2, the search criteria
 690  *
 691  * @param vol       (r)  volume we are searching on ...
 692  * @param dir       (rw) directory we are starting from ...
 693  * @param uname     (r)  UNIX name of object to search
 694  * @param rmatches  (r)  maximum number of matches we can return
 695  * @param pos       (r)  position we've stopped recently
 696  * @param rbuf      (w)  output buffer
 697  * @param nrecs     (w)  number of matches
 698  * @param rsize     (w)  length of data written to output buffer
 699  * @param ext       (r)  extended search flag
 700  */
 701 static int catsearch_db(const AFPObj *obj,
 702                         struct vol *vol,
 703                         struct dir *dir,
 704                         const char *uname,
 705                         int rmatches,
 706                         uint32_t *pos,
 707                         char *rbuf,
 708                         uint32_t *nrecs,
 709                         int *rsize,
 710                         int ext)
 711 {
 712     static char resbuf[DBD_MAX_SRCH_RSLTS * sizeof(cnid_t)];
 713     static uint32_t cur_pos;
 714     static int num_matches;
 715     int ccr ,r;
 716         int result = AFP_OK;
 717     struct path path;
 718         char *rrbuf = rbuf;
 719     char buffer[MAXPATHLEN +2];
 720     uint16_t flags = CONV_TOLOWER;
 721
 722     LOG(log_debug, logtype_afpd, "catsearch_db(req pos: %u): {pos: %u, name: %s}",
 723         *pos, cur_pos, uname);
 724
 725         if (*pos != 0 && *pos != cur_pos) {
 726                 result = AFPERR_CATCHNG;
 727                 goto catsearch_end;
 728         }
 729
 730     if (cur_pos == 0 || *pos == 0) {
 731         if (convert_charset(vol->v_volcharset,
 732                             vol->v_volcharset,
 733                             vol->v_maccharset,
 734                             uname,
 735                             strlen(uname),
 736                             buffer,
 737                             MAXPATHLEN,
 738                             &flags) == (size_t)-1) {
 739             LOG(log_error, logtype_afpd, "catsearch_db: conversion error");
 740             result = AFPERR_MISC;
 741             goto catsearch_end;
 742         }
 743
 744         LOG(log_debug, logtype_afpd, "catsearch_db: %s", buffer);
 745
 746         AFP_CNID_START("cnid_find");
 747         num_matches = cnid_find(vol->v_cdb,
 748                                 buffer,
 749                                 strlen(uname),
 750                                 resbuf,
 751                                 sizeof(resbuf));
 752         AFP_CNID_DONE();
 753         if (num_matches == -1) {
 754             result = AFPERR_MISC;
 755             goto catsearch_end;
 756         }
 757     }
 758
 759         while (cur_pos < num_matches) {
 760         char *name;
 761         cnid_t cnid, did;
 762         char resolvebuf[12 + MAXPATHLEN + 1];
 763         struct dir *dir;
 764
 765         /* Next CNID to process from buffer */
 766         memcpy(&cnid, resbuf + cur_pos * sizeof(cnid_t), sizeof(cnid_t));
 767         did = cnid;
 768
 769         AFP_CNID_START("cnid_resolve");
 770         name = cnid_resolve(vol->v_cdb, &did, resolvebuf, 12 + MAXPATHLEN + 1);
 771         AFP_CNID_DONE();
 772         if (name == NULL)
 773             goto next;
 774
 775         LOG(log_debug, logtype_afpd, "catsearch_db: {pos: %u, name:%s, cnid: %u}",
 776             cur_pos, name, ntohl(cnid));
 777         if ((dir = dirlookup(vol, did)) == NULL)
 778             goto next;
 779         if (movecwd(vol, dir) < 0 )
 780             goto next;
 781
 782         memset(&path, 0, sizeof(path));
 783         path.u_name = name;
 784         path.m_name = utompath(vol, name, cnid, utf8_encoding(vol->v_obj));
 785
 786         if (of_stat(vol, &path) != 0) {
 787             switch (errno) {
 788             case EACCES:
 789             case ELOOP:
 790                 goto next;
 791             case ENOENT:
 792
 793             default:
 794                 result = AFPERR_MISC;
 795                 goto catsearch_end;
 796             }
 797         }
 798         /* For files path.d_dir is the parent dir, for dirs its the dir itself */
 799         if (S_ISDIR(path.st.st_mode))
 800             if ((dir = dirlookup(vol, cnid)) == NULL)
 801                 goto next;
 802         path.d_dir = dir;
 803
 804         LOG(log_maxdebug, logtype_afpd,"catsearch_db: dir: %s, cwd: %s, name: %s",
 805             cfrombstr(dir->d_fullpath), getcwdpath(), path.u_name);
 806
 807         /* At last we can check the search criteria */
 808         ccr = crit_check(vol, &path);
 809         if ((ccr & 1)) {
 810             LOG(log_debug, logtype_afpd,"catsearch_db: match: %s/%s",
 811                 getcwdpath(), path.u_name);
 812             /* bit 1 means that criteria has been met */
 813             r = rslt_add(obj, vol, &path, &rrbuf, ext);
 814             if (r == 0) {
 815                 result = AFPERR_MISC;
 816                 goto catsearch_end;
 817             }
 818             *nrecs += r;
 819             /* Number of matches limit */
 820             if (--rmatches == 0)
 821                 goto catsearch_pause;
 822             /* Block size limit */
 823             if (rrbuf - rbuf >= 448)
 824                 goto catsearch_pause;
 825         }
 826     next:
 827         cur_pos++;
 828     } /* while */
 829
 830         /* finished */
 831         result = AFPERR_EOF;
 832     cur_pos = 0;
 833         goto catsearch_end;
 834
 835 catsearch_pause:
 836     *pos = cur_pos;
 837
 838 catsearch_end: /* Exiting catsearch: error condition */
 839         *rsize = rrbuf - rbuf;
 840     LOG(log_debug, logtype_afpd, "catsearch_db(req pos: %u): {pos: %u}", *pos, cur_pos);
 841         return result;
 842 }
 843
 844 /* -------------------------- */
 845 static int catsearch_afp(AFPObj *obj _U_, char *ibuf, size_t ibuflen,
 846                   char *rbuf, size_t *rbuflen, int ext)
 847 {
 848     struct vol *vol;
 849     uint16_t   vid;
 850     uint16_t   spec_len;
 851     uint32_t   rmatches, reserved;
 852     uint32_t    catpos[4];
 853     uint32_t   pdid = 0;
 854     int ret, rsize;
 855     uint32_t nrecs = 0;
 856     unsigned char *spec1, *spec2, *bspec1, *bspec2;
 857     size_t      len;
 858     uint16_t    namelen;
 859     uint16_t    flags;
 860     char            tmppath[256];
 861     char        *uname;
 862
 863     *rbuflen = 0;
 864
 865     /* min header size */
 866     if (ibuflen < 32) {
 867         return AFPERR_PARAM;
 868     }
 869
 870     memset(&c1, 0, sizeof(c1));
 871     memset(&c2, 0, sizeof(c2));
 872
 873     ibuf += 2;
 874     memcpy(&vid, ibuf, sizeof(vid));
 875     ibuf += sizeof(vid);
 876
 877     if ((vol = getvolbyvid(vid)) == NULL) {
 878         return AFPERR_PARAM;
 879     }
 880
 881     memcpy(&rmatches, ibuf, sizeof(rmatches));
 882     rmatches = ntohl(rmatches);
 883     ibuf += sizeof(rmatches);
 884
 885     /* FIXME: (rl) should we check if reserved == 0 ? */
 886     ibuf += sizeof(reserved);
 887
 888     memcpy(catpos, ibuf, sizeof(catpos));
 889     ibuf += sizeof(catpos);
 890
 891     memcpy(&c1.fbitmap, ibuf, sizeof(c1.fbitmap));
 892     c1.fbitmap = c2.fbitmap = ntohs(c1.fbitmap);
 893     ibuf += sizeof(c1.fbitmap);
 894
 895     memcpy(&c1.dbitmap, ibuf, sizeof(c1.dbitmap));
 896     c1.dbitmap = c2.dbitmap = ntohs(c1.dbitmap);
 897     ibuf += sizeof(c1.dbitmap);
 898
 899     memcpy(&c1.rbitmap, ibuf, sizeof(c1.rbitmap));
 900     c1.rbitmap = c2.rbitmap = ntohl(c1.rbitmap);
 901     ibuf += sizeof(c1.rbitmap);
 902
 903     if (! (c1.fbitmap || c1.dbitmap)) {
 904             return AFPERR_BITMAP;
 905     }
 906
 907     if ( ext) {
 908         memcpy(&spec_len, ibuf, sizeof(spec_len));
 909         spec_len = ntohs(spec_len);
 910     }
 911     else {
 912         /* with catsearch only name and parent id are allowed */
 913         c1.fbitmap &= (1<<FILPBIT_LNAME) | (1<<FILPBIT_PDID);
 914         c1.dbitmap &= (1<<DIRPBIT_LNAME) | (1<<DIRPBIT_PDID);
 915         spec_len = *(unsigned char*)ibuf;
 916     }
 917
 918     /* Parse file specifications */
 919     spec1 = (unsigned char*)ibuf;
 920     spec2 = (unsigned char*)ibuf + spec_len + 2;
 921
 922     spec1 += 2;
 923     spec2 += 2;
 924
 925     bspec1 = spec1;
 926     bspec2 = spec2;
 927     /* File attribute bits... */
 928     if (c1.rbitmap & (1 << FILPBIT_ATTR)) {
 929             memcpy(&c1.attr, spec1, sizeof(c1.attr));
 930             spec1 += sizeof(c1.attr);
 931             memcpy(&c2.attr, spec2, sizeof(c2.attr));
 932             spec2 += sizeof(c1.attr);
 933     }
 934
 935     /* Parent DID */
 936     if (c1.rbitmap & (1 << FILPBIT_PDID)) {
 937             memcpy(&c1.pdid, spec1, sizeof(pdid));
 938             spec1 += sizeof(c1.pdid);
 939             memcpy(&c2.pdid, spec2, sizeof(pdid));
 940             spec2 += sizeof(c2.pdid);
 941     } /* FIXME: PDID - do we demarshall this argument ? */
 942
 943     /* Creation date */
 944     if (c1.rbitmap & (1 << FILPBIT_CDATE)) {
 945             memcpy(&c1.cdate, spec1, sizeof(c1.cdate));
 946             spec1 += sizeof(c1.cdate);
 947             c1.cdate = AD_DATE_TO_UNIX(c1.cdate);
 948             memcpy(&c2.cdate, spec2, sizeof(c2.cdate));
 949             spec2 += sizeof(c1.cdate);
 950             ibuf += sizeof(c1.cdate);;
 951             c2.cdate = AD_DATE_TO_UNIX(c2.cdate);
 952     }
 953
 954     /* Modification date */
 955     if (c1.rbitmap & (1 << FILPBIT_MDATE)) {
 956             memcpy(&c1.mdate, spec1, sizeof(c1.mdate));
 957             c1.mdate = AD_DATE_TO_UNIX(c1.mdate);
 958             spec1 += sizeof(c1.mdate);
 959             memcpy(&c2.mdate, spec2, sizeof(c2.mdate));
 960             c2.mdate = AD_DATE_TO_UNIX(c2.mdate);
 961             spec2 += sizeof(c1.mdate);
 962     }
 963
 964     /* Backup date */
 965     if (c1.rbitmap & (1 << FILPBIT_BDATE)) {
 966             memcpy(&c1.bdate, spec1, sizeof(c1.bdate));
 967             spec1 += sizeof(c1.bdate);
 968             c1.bdate = AD_DATE_TO_UNIX(c1.bdate);
 969             memcpy(&c2.bdate, spec2, sizeof(c2.bdate));
 970             spec2 += sizeof(c2.bdate);
 971             c1.bdate = AD_DATE_TO_UNIX(c2.bdate);
 972     }
 973
 974     /* Finder info */
 975     if (c1.rbitmap & (1 << FILPBIT_FINFO)) {
 976         packed_finder buf;
 977
 978             memcpy(buf, spec1, sizeof(buf));
 979             unpack_buffer(&c1.finfo, buf);
 980             spec1 += sizeof(buf);
 981
 982             memcpy(buf, spec2, sizeof(buf));
 983             unpack_buffer(&c2.finfo, buf);
 984             spec2 += sizeof(buf);
 985     } /* Finder info */
 986
 987     if ((c1.rbitmap & (1 << DIRPBIT_OFFCNT)) != 0) {
 988         /* Offspring count - only directories */
 989                 if (c1.fbitmap == 0) {
 990                 memcpy(&c1.offcnt, spec1, sizeof(c1.offcnt));
 991                 spec1 += sizeof(c1.offcnt);
 992                 c1.offcnt = ntohs(c1.offcnt);
 993                 memcpy(&c2.offcnt, spec2, sizeof(c2.offcnt));
 994                 spec2 += sizeof(c2.offcnt);
 995                 c2.offcnt = ntohs(c2.offcnt);
 996                 }
 997                 else if (c1.dbitmap == 0) {
 998                         /* ressource fork length */
 999                 }
1000                 else {
1001                 return AFPERR_BITMAP;  /* error */
1002                 }
1003     } /* Offspring count/ressource fork length */
1004
1005     /* Long name */
1006     if (c1.rbitmap & (1 << FILPBIT_LNAME)) {
1007         /* Get the long filename */
1008                 memcpy(tmppath, bspec1 + spec1[1] + 1, (bspec1 + spec1[1])[0]);
1009                 tmppath[(bspec1 + spec1[1])[0]]= 0;
1010                 len = convert_string ( vol->v_maccharset, CH_UCS2, tmppath, -1, c1.lname, sizeof(c1.lname));
1011         if (len == (size_t)(-1))
1012             return AFPERR_PARAM;
1013
1014 #if 0
1015                 /* FIXME: do we need it ? It's always null ! */
1016                 memcpy(c2.lname, bspec2 + spec2[1] + 1, (bspec2 + spec2[1])[0]);
1017                 c2.lname[(bspec2 + spec2[1])[0]]= 0;
1018 #endif
1019     }
1020         /* UTF8 Name */
1021     if (c1.rbitmap & (1 << FILPBIT_PDINFO)) {
1022
1023                 /* offset */
1024                 memcpy(&namelen, spec1, sizeof(namelen));
1025                 namelen = ntohs (namelen);
1026
1027                 spec1 = bspec1+namelen+4; /* Skip Unicode Hint */
1028
1029                 /* length */
1030                 memcpy(&namelen, spec1, sizeof(namelen));
1031                 namelen = ntohs (namelen);
1032                 if (namelen > UTF8FILELEN_EARLY)  /* Safeguard */
1033                         namelen = UTF8FILELEN_EARLY;
1034
1035                 memcpy (c1.utf8name, spec1+2, namelen);
1036                 c1.utf8name[namelen] = 0;
1037         if ((uname = mtoupath(vol, c1.utf8name, 0, utf8_encoding(obj))) == NULL)
1038             return AFPERR_PARAM;
1039
1040                 /* convert charset */
1041                 flags = CONV_PRECOMPOSE;
1042                 len = convert_charset(CH_UTF8_MAC, CH_UCS2, CH_UTF8, c1.utf8name, namelen, c1.utf8name, 512, &flags);
1043         if (len == (size_t)(-1))
1044             return AFPERR_PARAM;
1045     }
1046
1047     /* Call search */
1048     *rbuflen = 24;
1049     if ((c1.rbitmap & (1 << FILPBIT_PDINFO))
1050         && !(c1.rbitmap & (1<<CATPBIT_PARTIAL))
1051         && (strcmp(vol->v_cnidscheme, "dbd") == 0)
1052         && (vol->v_flags & AFPVOL_SEARCHDB))
1053         /* we've got a name and it's a dbd volume, so search CNID database */
1054         ret = catsearch_db(obj, vol, vol->v_root, uname, rmatches, &catpos[0], rbuf+24, &nrecs, &rsize, ext);
1055     else
1056         /* perform a slow filesystem tree search */
1057         ret = catsearch(obj, vol, vol->v_root, rmatches, &catpos[0], rbuf+24, &nrecs, &rsize, ext);
1058
1059     memcpy(rbuf, catpos, sizeof(catpos));
1060     rbuf += sizeof(catpos);
1061
1062     c1.fbitmap = htons(c1.fbitmap);
1063     memcpy(rbuf, &c1.fbitmap, sizeof(c1.fbitmap));
1064     rbuf += sizeof(c1.fbitmap);
1065
1066     c1.dbitmap = htons(c1.dbitmap);
1067     memcpy(rbuf, &c1.dbitmap, sizeof(c1.dbitmap));
1068     rbuf += sizeof(c1.dbitmap);
1069
1070     nrecs = htonl(nrecs);
1071     memcpy(rbuf, &nrecs, sizeof(nrecs));
1072     rbuf += sizeof(nrecs);
1073     *rbuflen += rsize;
1074
1075     return ret;
1076 } /* catsearch_afp */
1077
1078 /* -------------------------- */
1079 int afp_catsearch (AFPObj *obj, char *ibuf, size_t ibuflen,
1080                   char *rbuf, size_t *rbuflen)
1081 {
1082         return catsearch_afp( obj, ibuf, ibuflen, rbuf, rbuflen, 0);
1083 }
1084
1085
1086 int afp_catsearch_ext (AFPObj *obj, char *ibuf, size_t ibuflen,
1087                   char *rbuf, size_t *rbuflen)
1088 {
1089         return catsearch_afp( obj, ibuf, ibuflen, rbuf, rbuflen, 1);
1090 }
1091
1092 /* FIXME: we need a clean separation between afp stubs and 'real' implementation */
1093 /* (so, all buffer packing/unpacking should be done in stub, everything else
1094    should be done in other functions) */