]> arthur.barton.de Git - netatalk.git/blob - etc/afpd/volume.c
Whitespace and exclamation mark fixes
[netatalk.git] / etc / afpd / volume.c
1 /*
2  * Copyright (c) 1990,1993 Regents of The University of Michigan.
3  * All Rights Reserved.  See COPYRIGHT.
4  */
5
6 #ifdef HAVE_CONFIG_H
7 #include "config.h"
8 #endif /* HAVE_CONFIG_H */
9
10 #include <stdio.h>
11 #include <stdlib.h>
12 #include <ctype.h>
13 #include <pwd.h>
14 #include <grp.h>
15 #include <utime.h>
16 #include <errno.h>
17 #include <string.h>
18 #include <sys/param.h>
19 #include <sys/socket.h>
20 #include <netinet/in.h>
21 #include <arpa/inet.h>
22 #include <inttypes.h>
23 #include <time.h>
24
25 #include <atalk/dsi.h>
26 #include <atalk/adouble.h>
27 #include <atalk/afp.h>
28 #include <atalk/util.h>
29 #include <atalk/logger.h>
30 #include <atalk/vfs.h>
31 #include <atalk/uuid.h>
32 #include <atalk/ea.h>
33 #include <atalk/bstrlib.h>
34 #include <atalk/bstradd.h>
35 #include <atalk/ftw.h>
36 #include <atalk/globals.h>
37 #include <atalk/fce_api.h>
38 #include <atalk/errchk.h>
39 #include <atalk/iniparser.h>
40 #include <atalk/unix.h>
41 #include <atalk/netatalk_conf.h>
42 #include <atalk/server_ipc.h>
43
44 #ifdef CNID_DB
45 #include <atalk/cnid.h>
46 #endif /* CNID_DB*/
47
48 #include "directory.h"
49 #include "file.h"
50 #include "volume.h"
51 #include "unix.h"
52 #include "mangle.h"
53 #include "fork.h"
54 #include "hash.h"
55 #include "acls.h"
56
57 #define VOLPASSLEN  8
58
59 extern int afprun(int root, char *cmd, int *outfd);
60
61 /*!
62  * Read band-size info from Info.plist XML file of an TM sparsebundle
63  *
64  * @param path   (r) path to Info.plist file
65  * @return           band-size in bytes, -1 on error
66  */
67 static long long int get_tm_bandsize(const char *path)
68 {
69     EC_INIT;
70     FILE *file = NULL;
71     char buf[512];
72     long long int bandsize = -1;
73
74     EC_NULL_LOGSTR( file = fopen(path, "r"),
75                     "get_tm_bandsize(\"%s\"): %s",
76                     path, strerror(errno) );
77
78     while (fgets(buf, sizeof(buf), file) != NULL) {
79         if (strstr(buf, "band-size") == NULL)
80             continue;
81
82         if (fscanf(file, " <integer>%lld</integer>", &bandsize) != 1) {
83             LOG(log_error, logtype_afpd, "get_tm_bandsize(\"%s\"): can't parse band-size", path);
84             EC_FAIL;
85         }
86         break;
87     }
88
89 EC_CLEANUP:
90     if (file)
91         fclose(file);
92     LOG(log_debug, logtype_afpd, "get_tm_bandsize(\"%s\"): bandsize: %lld", path, bandsize);
93     return bandsize;
94 }
95
96 /*!
97  * Return number on entries in a directory
98  *
99  * @param path   (r) path to dir
100  * @return           number of entries, -1 on error
101  */
102 static long long int get_tm_bands(const char *path)
103 {
104     EC_INIT;
105     long long int count = 0;
106     DIR *dir = NULL;
107     const struct dirent *entry;
108
109     EC_NULL( dir = opendir(path) );
110
111     while ((entry = readdir(dir)) != NULL)
112         count++;
113     count -= 2; /* All OSens I'm aware of return "." and "..", so just substract them, avoiding string comparison in loop */
114         
115 EC_CLEANUP:
116     if (dir)
117         closedir(dir);
118     if (ret != 0)
119         return -1;
120     return count;
121 }
122
123 /*!
124  * Calculate used size of a TimeMachine volume
125  *
126  * This assumes that the volume is used only for TimeMachine.
127  *
128  * 1) readdir(path of volume)
129  * 2) for every element that matches regex "\(.*\)\.sparsebundle$" :
130  * 3) parse "\1.sparsebundle/Info.plist" and read the band-size XML key integer value
131  * 4) readdir "\1.sparsebundle/bands/" counting files
132  * 5) calculate used size as: (file_count - 1) * band-size
133  *
134  * The result of the calculation is returned in "volume->v_tm_used".
135  * "volume->v_appended" gets reset to 0.
136  * "volume->v_tm_cachetime" is updated with the current time from time(NULL).
137  *
138  * "volume->v_tm_used" is cached for TM_USED_CACHETIME seconds and updated by
139  * "volume->v_appended". The latter is increased by X every time the client
140  * appends X bytes to a file (in fork.c).
141  *
142  * @param vol     (rw) volume to calculate
143  * @return             0 on success, -1 on error
144  */
145 #define TM_USED_CACHETIME 60    /* cache for 60 seconds */
146 static int get_tm_used(struct vol * restrict vol)
147 {
148     EC_INIT;
149     long long int bandsize;
150     VolSpace used = 0;
151     bstring infoplist = NULL;
152     bstring bandsdir = NULL;
153     DIR *dir = NULL;
154     const struct dirent *entry;
155     const char *p;
156     long int links;
157     time_t now = time(NULL);
158
159     if (vol->v_tm_cachetime
160         && ((vol->v_tm_cachetime + TM_USED_CACHETIME) >= now)) {
161         if (vol->v_tm_used == -1)
162             EC_FAIL;
163         vol->v_tm_used += vol->v_appended;
164         vol->v_appended = 0;
165         LOG(log_debug, logtype_afpd, "getused(\"%s\"): cached: %" PRIu64 " bytes",
166             vol->v_path, vol->v_tm_used);
167         return 0;
168     }
169
170     vol->v_tm_cachetime = now;
171
172     EC_NULL( dir = opendir(vol->v_path) );
173
174     while ((entry = readdir(dir)) != NULL) {
175         if (((p = strstr(entry->d_name, "sparsebundle")) != NULL)
176             && (strlen(entry->d_name) == (p + strlen("sparsebundle") - entry->d_name))) {
177
178             EC_NULL_LOG( infoplist = bformat("%s/%s/%s", vol->v_path, entry->d_name, "Info.plist") );
179             
180             if ((bandsize = get_tm_bandsize(cfrombstr(infoplist))) == -1) {
181                 bdestroy(infoplist);
182                 continue;
183             }
184
185             EC_NULL_LOG( bandsdir = bformat("%s/%s/%s/", vol->v_path, entry->d_name, "bands") );
186
187             if ((links = get_tm_bands(cfrombstr(bandsdir))) == -1) {
188                 bdestroy(infoplist);
189                 bdestroy(bandsdir);
190                 continue;
191             }
192
193             used += (links - 1) * bandsize;
194             LOG(log_debug, logtype_afpd, "getused(\"%s\"): bands: %" PRIu64 " bytes",
195                 cfrombstr(bandsdir), used);
196         }
197     }
198
199     vol->v_tm_used = used;
200
201 EC_CLEANUP:
202     if (infoplist)
203         bdestroy(infoplist);
204     if (bandsdir)
205         bdestroy(bandsdir);
206     if (dir)
207         closedir(dir);
208
209     LOG(log_debug, logtype_afpd, "getused(\"%s\"): %" PRIu64 " bytes", vol->v_path, vol->v_tm_used);
210
211     EC_EXIT;
212 }
213
214 static int getvolspace(const AFPObj *obj, struct vol *vol,
215                        uint32_t *bfree, uint32_t *btotal,
216                        VolSpace *xbfree, VolSpace *xbtotal, uint32_t *bsize)
217 {
218     int         spaceflag, rc;
219     uint32_t   maxsize;
220 #ifndef NO_QUOTA_SUPPORT
221     VolSpace    qfree, qtotal;
222 #endif
223
224     spaceflag = AFPVOL_GVSMASK & vol->v_flags;
225     /* report up to 2GB if afp version is < 2.2 (4GB if not) */
226     maxsize = (obj->afp_version < 22) ? 0x7fffffffL : 0xffffffffL;
227
228 #ifdef AFS
229     if ( spaceflag == AFPVOL_NONE || spaceflag == AFPVOL_AFSGVS ) {
230         if ( afs_getvolspace( vol, xbfree, xbtotal, bsize ) == AFP_OK ) {
231             vol->v_flags = ( ~AFPVOL_GVSMASK & vol->v_flags ) | AFPVOL_AFSGVS;
232             goto getvolspace_done;
233         }
234     }
235 #endif
236
237     if (( rc = ustatfs_getvolspace( vol, xbfree, xbtotal, bsize)) != AFP_OK ) {
238         return( rc );
239     }
240
241 #ifndef NO_QUOTA_SUPPORT
242     if ( spaceflag == AFPVOL_NONE || spaceflag == AFPVOL_UQUOTA ) {
243         if ( uquota_getvolspace(obj, vol, &qfree, &qtotal, *bsize ) == AFP_OK ) {
244             vol->v_flags = ( ~AFPVOL_GVSMASK & vol->v_flags ) | AFPVOL_UQUOTA;
245             *xbfree = MIN(*xbfree, qfree);
246             *xbtotal = MIN(*xbtotal, qtotal);
247             goto getvolspace_done;
248         }
249     }
250 #endif
251     vol->v_flags = ( ~AFPVOL_GVSMASK & vol->v_flags ) | AFPVOL_USTATFS;
252
253 getvolspace_done:
254     if (vol->v_limitsize) {
255         if (get_tm_used(vol) != 0)
256             return AFPERR_MISC;
257
258         *xbtotal = MIN(*xbtotal, (vol->v_limitsize * 1024 * 1024));
259         *xbfree = MIN(*xbfree, *xbtotal < vol->v_tm_used ? 0 : *xbtotal - vol->v_tm_used);
260
261         LOG(log_debug, logtype_afpd,
262             "volparams: total: %" PRIu64 ", used: %" PRIu64 ", free: %" PRIu64 " bytes",
263             *xbtotal, vol->v_tm_used, *xbfree);
264     }
265
266     *bfree = MIN(*xbfree, maxsize);
267     *btotal = MIN(*xbtotal, maxsize);
268     return( AFP_OK );
269 }
270
271 /* -----------------------
272  * set volume creation date
273  * avoid duplicate, well at least it tries
274  */
275 static void vol_setdate(uint16_t id, struct adouble *adp, time_t date)
276 {
277     struct vol  *volume;
278     struct vol  *vol = getvolumes();
279
280     for ( volume = getvolumes(); volume; volume = volume->v_next ) {
281         if (volume->v_vid == id) {
282             vol = volume;
283         }
284         else if ((time_t)(AD_DATE_FROM_UNIX(date)) == volume->v_ctime) {
285             date = date+1;
286             volume = getvolumes(); /* restart */
287         }
288     }
289     vol->v_ctime = AD_DATE_FROM_UNIX(date);
290     ad_setdate(adp, AD_DATE_CREATE | AD_DATE_UNIX, date);
291 }
292
293 /* ----------------------- */
294 static int getvolparams(const AFPObj *obj, uint16_t bitmap, struct vol *vol, struct stat *st, char *buf, size_t *buflen)
295 {
296     struct adouble  ad;
297     int         bit = 0, isad = 1;
298     uint32_t       aint;
299     u_short     ashort;
300     uint32_t       bfree, btotal, bsize;
301     VolSpace            xbfree, xbtotal; /* extended bytes */
302     char        *data, *nameoff = NULL;
303     char                *slash;
304
305     LOG(log_debug, logtype_afpd, "getvolparams: Volume '%s'", vol->v_localname);
306
307     /* courtesy of jallison@whistle.com:
308      * For MacOS8.x support we need to create the
309      * .Parent file here if it doesn't exist. */
310
311     /* Convert adouble:v2 to adouble:ea on the fly */
312     (void)ad_convert(vol->v_path, st, vol, NULL);
313
314     ad_init(&ad, vol);
315     if (ad_open(&ad, vol->v_path, ADFLAGS_HF | ADFLAGS_DIR | ADFLAGS_RDWR | ADFLAGS_CREATE, 0666) != 0 ) {
316         isad = 0;
317         vol->v_ctime = AD_DATE_FROM_UNIX(st->st_mtime);
318
319     } else if (ad_get_MD_flags( &ad ) & O_CREAT) {
320         slash = strrchr( vol->v_path, '/' );
321         if(slash)
322             slash++;
323         else
324             slash = vol->v_path;
325         if (ad_getentryoff(&ad, ADEID_NAME)) {
326             ad_setentrylen( &ad, ADEID_NAME, strlen( slash ));
327             memcpy(ad_entry( &ad, ADEID_NAME ), slash,
328                    ad_getentrylen( &ad, ADEID_NAME ));
329         }
330         vol_setdate(vol->v_vid, &ad, st->st_mtime);
331         ad_flush(&ad);
332     }
333     else {
334         if (ad_getdate(&ad, AD_DATE_CREATE, &aint) < 0)
335             vol->v_ctime = AD_DATE_FROM_UNIX(st->st_mtime);
336         else
337             vol->v_ctime = aint;
338     }
339
340     if (( bitmap & ( (1<<VOLPBIT_BFREE)|(1<<VOLPBIT_BTOTAL) |
341                      (1<<VOLPBIT_XBFREE)|(1<<VOLPBIT_XBTOTAL) |
342                      (1<<VOLPBIT_BSIZE)) ) != 0 ) {
343         if ( getvolspace(obj, vol, &bfree, &btotal, &xbfree, &xbtotal,
344                           &bsize) != AFP_OK ) {
345             if ( isad ) {
346                 ad_close( &ad, ADFLAGS_HF );
347             }
348             return( AFPERR_PARAM );
349         }
350     }
351
352     data = buf;
353     while ( bitmap != 0 ) {
354         while (( bitmap & 1 ) == 0 ) {
355             bitmap = bitmap>>1;
356             bit++;
357         }
358
359         switch ( bit ) {
360         case VOLPBIT_ATTR :
361             ashort = 0;
362             /* check for read-only.
363              * NOTE: we don't actually set the read-only flag unless
364              *       it's passed in that way as it's possible to mount
365              *       a read-write filesystem under a read-only one. */
366             if ((vol->v_flags & AFPVOL_RO) ||
367                 ((utime(vol->v_path, NULL) < 0) && (errno == EROFS))) {
368                 ashort |= VOLPBIT_ATTR_RO;
369             }
370             /* prior 2.1 only VOLPBIT_ATTR_RO is defined */
371             if (obj->afp_version > 20) {
372                 if (vol->v_cdb != NULL && (vol->v_cdb->cnid_db_flags & CNID_FLAG_PERSISTENT))
373                     ashort |= VOLPBIT_ATTR_FILEID;
374                 ashort |= VOLPBIT_ATTR_CATSEARCH;
375
376                 if (obj->afp_version >= 30) {
377                     ashort |= VOLPBIT_ATTR_UTF8;
378                     if (vol->v_flags & AFPVOL_UNIX_PRIV)
379                         ashort |= VOLPBIT_ATTR_UNIXPRIV;
380                     if (vol->v_flags & AFPVOL_TM)
381                         ashort |= VOLPBIT_ATTR_TM;
382                     if (vol->v_flags & AFPVOL_NONETIDS)
383                         ashort |= VOLPBIT_ATTR_NONETIDS;
384                     if (obj->afp_version >= 32) {
385                         if (vol->v_vfs_ea)
386                             ashort |= VOLPBIT_ATTR_EXT_ATTRS;
387                         if (vol->v_flags & AFPVOL_ACLS)
388                             ashort |= VOLPBIT_ATTR_ACLS;
389                         if (vol->v_casefold & AFPVOL_CASESENS)
390                             ashort |= VOLPBIT_ATTR_CASESENS;
391                     }
392                 }
393             }
394             ashort = htons(ashort);
395             memcpy(data, &ashort, sizeof( ashort ));
396             data += sizeof( ashort );
397             break;
398
399         case VOLPBIT_SIG :
400             ashort = htons( AFPVOLSIG_DEFAULT );
401             memcpy(data, &ashort, sizeof( ashort ));
402             data += sizeof( ashort );
403             break;
404
405         case VOLPBIT_CDATE :
406             aint = vol->v_ctime;
407             memcpy(data, &aint, sizeof( aint ));
408             data += sizeof( aint );
409             break;
410
411         case VOLPBIT_MDATE :
412             if ( st->st_mtime > vol->v_mtime ) {
413                 vol->v_mtime = st->st_mtime;
414             }
415             aint = AD_DATE_FROM_UNIX(vol->v_mtime);
416             memcpy(data, &aint, sizeof( aint ));
417             data += sizeof( aint );
418             break;
419
420         case VOLPBIT_BDATE :
421             if (!isad ||  (ad_getdate(&ad, AD_DATE_BACKUP, &aint) < 0))
422                 aint = AD_DATE_START;
423             memcpy(data, &aint, sizeof( aint ));
424             data += sizeof( aint );
425             break;
426
427         case VOLPBIT_VID :
428             memcpy(data, &vol->v_vid, sizeof( vol->v_vid ));
429             data += sizeof( vol->v_vid );
430             break;
431
432         case VOLPBIT_BFREE :
433             bfree = htonl( bfree );
434             memcpy(data, &bfree, sizeof( bfree ));
435             data += sizeof( bfree );
436             break;
437
438         case VOLPBIT_BTOTAL :
439             btotal = htonl( btotal );
440             memcpy(data, &btotal, sizeof( btotal ));
441             data += sizeof( btotal );
442             break;
443
444 #ifndef NO_LARGE_VOL_SUPPORT
445         case VOLPBIT_XBFREE :
446             xbfree = hton64( xbfree );
447             memcpy(data, &xbfree, sizeof( xbfree ));
448             data += sizeof( xbfree );
449             break;
450
451         case VOLPBIT_XBTOTAL :
452             xbtotal = hton64( xbtotal );
453             memcpy(data, &xbtotal, sizeof( xbtotal ));
454             data += sizeof( xbfree );
455             break;
456 #endif /* ! NO_LARGE_VOL_SUPPORT */
457
458         case VOLPBIT_NAME :
459             nameoff = data;
460             data += sizeof( uint16_t );
461             break;
462
463         case VOLPBIT_BSIZE:  /* block size */
464             bsize = htonl(bsize);
465             memcpy(data, &bsize, sizeof(bsize));
466             data += sizeof(bsize);
467             break;
468
469         default :
470             if ( isad ) {
471                 ad_close( &ad, ADFLAGS_HF );
472             }
473             return( AFPERR_BITMAP );
474         }
475         bitmap = bitmap>>1;
476         bit++;
477     }
478     if ( nameoff ) {
479         ashort = htons( data - buf );
480         memcpy(nameoff, &ashort, sizeof( ashort ));
481         /* name is always in mac charset */
482         aint = ucs2_to_charset( vol->v_maccharset, vol->v_macname, data+1, AFPVOL_MACNAMELEN + 1);
483         if ( aint <= 0 ) {
484             *buflen = 0;
485             return AFPERR_MISC;
486         }
487
488         *data++ = aint;
489         data += aint;
490     }
491     if ( isad ) {
492         ad_close(&ad, ADFLAGS_HF);
493     }
494     *buflen = data - buf;
495     return( AFP_OK );
496 }
497
498 /* ------------------------- */
499 static int stat_vol(const AFPObj *obj, uint16_t bitmap, struct vol *vol, char *rbuf, size_t *rbuflen)
500 {
501     struct stat st;
502     int     ret;
503     size_t  buflen;
504
505     if ( stat( vol->v_path, &st ) < 0 ) {
506         *rbuflen = 0;
507         return( AFPERR_PARAM );
508     }
509     /* save the volume device number */
510     vol->v_dev = st.st_dev;
511
512     buflen = *rbuflen - sizeof( bitmap );
513     if (( ret = getvolparams(obj, bitmap, vol, &st,
514                               rbuf + sizeof( bitmap ), &buflen )) != AFP_OK ) {
515         *rbuflen = 0;
516         return( ret );
517     }
518     *rbuflen = buflen + sizeof( bitmap );
519     bitmap = htons( bitmap );
520     memcpy(rbuf, &bitmap, sizeof( bitmap ));
521     return( AFP_OK );
522
523 }
524
525 /* ------------------------------- */
526 int afp_getsrvrparms(AFPObj *obj, char *ibuf _U_, size_t ibuflen _U_, char *rbuf, size_t *rbuflen)
527 {
528     struct timeval  tv;
529     struct stat     st;
530     struct vol      *volume;
531     char    *data;
532     char        *namebuf;
533     int         vcnt;
534     size_t      len;
535     uint32_t    aint;
536
537     load_volumes(obj, lv_none);
538
539     data = rbuf + 5;
540     for ( vcnt = 0, volume = getvolumes(); volume && vcnt < 255; volume = volume->v_next ) {
541         if (!(volume->v_flags & AFPVOL_NOSTAT)) {
542             struct maccess ma;
543
544             if ( stat( volume->v_path, &st ) < 0 ) {
545                 LOG(log_info, logtype_afpd, "afp_getsrvrparms(%s): stat: %s",
546                     volume->v_path, strerror(errno) );
547                 continue;       /* can't access directory */
548             }
549             if (!S_ISDIR(st.st_mode)) {
550                 continue;       /* not a dir */
551             }
552             accessmode(obj, volume, volume->v_path, &ma, NULL, &st);
553             if ((ma.ma_user & (AR_UREAD | AR_USEARCH)) != (AR_UREAD | AR_USEARCH)) {
554                 continue;   /* no r-x access */
555             }
556         }
557
558         if (utf8_encoding(obj)) {
559             len = ucs2_to_charset_allocate(CH_UTF8_MAC, &namebuf, volume->v_u8mname);
560         } else {
561             len = ucs2_to_charset_allocate(obj->options.maccharset, &namebuf, volume->v_macname);
562         }
563
564         if (len == (size_t)-1)
565             continue;
566
567         /*
568          * There seems to be an undocumented limit on how big our reply can get
569          * before the client chokes and closes the connection.
570          * Testing with 10.8.4 found the limit at ~4600 bytes. Go figure. 
571          */
572         if (((data + len + 3) - rbuf) > 4600)
573             break;
574
575         /* set password bit if there's a volume password */
576         *data = (volume->v_password) ? AFPSRVR_PASSWD : 0;
577
578         *data++ |= 0; /* UNIX PRIVS BIT ..., OSX doesn't seem to use it, so we don't either */
579         *data++ = len;
580         memcpy(data, namebuf, len );
581         data += len;
582         free(namebuf);
583         vcnt++;
584     }
585
586     *rbuflen = data - rbuf;
587     data = rbuf;
588     if ( gettimeofday( &tv, NULL ) < 0 ) {
589         LOG(log_error, logtype_afpd, "afp_getsrvrparms: gettimeofday: %s", strerror(errno) );
590         *rbuflen = 0;
591         return AFPERR_PARAM;
592     }
593
594     aint = AD_DATE_FROM_UNIX(tv.tv_sec);
595     memcpy(data, &aint, sizeof( uint32_t));
596     data += sizeof( uint32_t);
597     *data = vcnt;
598     return( AFP_OK );
599 }
600
601 /* ------------------------- */
602 static int volume_codepage(AFPObj *obj, struct vol *volume)
603 {
604     struct charset_functions *charset;
605     /* Codepages */
606
607     if (!volume->v_volcodepage)
608         volume->v_volcodepage = strdup("UTF8");
609
610     if ( (charset_t) -1 == ( volume->v_volcharset = add_charset(volume->v_volcodepage)) ) {
611         LOG (log_error, logtype_afpd, "Setting codepage %s as volume codepage failed", volume->v_volcodepage);
612         return -1;
613     }
614
615     if ( NULL == (charset = find_charset_functions(volume->v_volcodepage)) || charset->flags & CHARSET_ICONV ) {
616         LOG (log_warning, logtype_afpd, "WARNING: volume encoding %s is *not* supported by netatalk, expect problems!", volume->v_volcodepage);
617     }
618
619     if (!volume->v_maccodepage)
620         volume->v_maccodepage = strdup(obj->options.maccodepage);
621
622     if ( (charset_t) -1 == ( volume->v_maccharset = add_charset(volume->v_maccodepage)) ) {
623         LOG (log_error, logtype_afpd, "Setting codepage %s as mac codepage failed", volume->v_maccodepage);
624         return -1;
625     }
626
627     if ( NULL == ( charset = find_charset_functions(volume->v_maccodepage)) || ! (charset->flags & CHARSET_CLIENT) ) {
628         LOG (log_error, logtype_afpd, "Fatal error: mac charset %s not supported", volume->v_maccodepage);
629         return -1;
630     }
631     volume->v_kTextEncoding = htonl(charset->kTextEncoding);
632     return 0;
633 }
634
635 /* ------------------------- */
636 static int volume_openDB(const AFPObj *obj, struct vol *volume)
637 {
638     int flags = 0;
639
640     if ((volume->v_flags & AFPVOL_NODEV)) {
641         flags |= CNID_FLAG_NODEV;
642     }
643
644     volume->v_cdb = cnid_open(volume, volume->v_cnidscheme, flags);
645
646     if ( ! volume->v_cdb && ! (flags & CNID_FLAG_MEMORY)) {
647         /* The first attempt failed and it wasn't yet an attempt to open in-memory */
648         LOG(log_error, logtype_afpd, "Can't open volume \"%s\" CNID backend \"%s\" ",
649             volume->v_path, volume->v_cnidscheme);
650         LOG(log_error, logtype_afpd, "Reopen volume %s using in memory temporary CNID DB.",
651             volume->v_path);
652         flags |= CNID_FLAG_MEMORY;
653         volume->v_cdb = cnid_open(volume, "tdb", flags);
654 #ifdef SERVERTEXT
655         /* kill ourself with SIGUSR2 aka msg pending */
656         if (volume->v_cdb) {
657             setmessage("Something wrong with the volume's CNID DB, using temporary CNID DB instead."
658                        "Check server messages for details!");
659             kill(getpid(), SIGUSR2);
660             /* deactivate cnid caching/storing in AppleDouble files */
661         }
662 #endif
663     }
664
665     return (!volume->v_cdb)?-1:0;
666 }
667
668 /*
669  * Send list of open volumes to afpd master via IPC
670  */
671 static void server_ipc_volumes(AFPObj *obj)
672 {
673     struct vol *volume, *vols;
674     volume = vols = getvolumes();
675     bstring openvolnames = bfromcstr("");
676     bool firstvol = true;
677
678     while (volume) {
679         if (volume->v_flags & AFPVOL_OPEN) {
680             if (!firstvol)
681                 bcatcstr(openvolnames, ", ");
682             else
683                 firstvol = false;
684             bcatcstr(openvolnames, volume->v_localname);
685         }
686         volume = volume->v_next;
687     }
688
689     ipc_child_write(obj->ipc_fd, IPC_VOLUMES, blength(openvolnames), bdata(openvolnames));
690     bdestroy(openvolnames);
691 }
692
693 /* -------------------------
694  * we are the user here
695  */
696 int afp_openvol(AFPObj *obj, char *ibuf, size_t ibuflen _U_, char *rbuf, size_t *rbuflen)
697 {
698     struct stat st;
699     char    *volname;
700
701     struct vol  *volume;
702     struct dir  *dir;
703     int     len, ret;
704     size_t  namelen;
705     uint16_t   bitmap;
706     char        *vol_uname;
707     char        *vol_mname = NULL;
708     char        *volname_tmp;
709
710     ibuf += 2;
711     memcpy(&bitmap, ibuf, sizeof( bitmap ));
712     bitmap = ntohs( bitmap );
713     ibuf += sizeof( bitmap );
714
715     *rbuflen = 0;
716     if (( bitmap & (1<<VOLPBIT_VID)) == 0 ) {
717         return AFPERR_BITMAP;
718     }
719
720     len = (unsigned char)*ibuf++;
721     volname = obj->oldtmp;
722
723     if ((volname_tmp = strchr(volname,'+')) != NULL)
724         volname = volname_tmp+1;
725
726     if (utf8_encoding(obj)) {
727         namelen = convert_string(CH_UTF8_MAC, CH_UCS2, ibuf, len, volname, sizeof(obj->oldtmp));
728     } else {
729         namelen = convert_string(obj->options.maccharset, CH_UCS2, ibuf, len, volname, sizeof(obj->oldtmp));
730     }
731
732     if ( namelen <= 0) {
733         return AFPERR_PARAM;
734     }
735
736     ibuf += len;
737     if ((len + 1) & 1) /* pad to an even boundary */
738         ibuf++;
739
740     load_volumes(obj, lv_none);
741
742     for ( volume = getvolumes(); volume; volume = volume->v_next ) {
743         if ( strcasecmp_w( (ucs2_t*) volname, volume->v_name ) == 0 ) {
744             break;
745         }
746     }
747
748     if ( volume == NULL ) {
749         return AFPERR_NOOBJ;
750     }
751
752     /* check for a volume password */
753     if (volume->v_password && strncmp(ibuf, volume->v_password, VOLPASSLEN)) {
754         return AFPERR_ACCESS;
755     }
756
757     if (( volume->v_flags & AFPVOL_OPEN  ) ) {
758         /* the volume is already open */
759         return stat_vol(obj, bitmap, volume, rbuf, rbuflen);
760     }
761
762     if (volume->v_root_preexec) {
763         if ((ret = afprun(1, volume->v_root_preexec, NULL)) && volume->v_root_preexec_close) {
764             LOG(log_error, logtype_afpd, "afp_openvol(%s): root preexec : %d", volume->v_path, ret );
765             return AFPERR_MISC;
766         }
767     }
768
769     if (volume->v_preexec) {
770         if ((ret = afprun(0, volume->v_preexec, NULL)) && volume->v_preexec_close) {
771             LOG(log_error, logtype_afpd, "afp_openvol(%s): preexec : %d", volume->v_path, ret );
772             return AFPERR_MISC;
773         }
774     }
775
776     if ( stat( volume->v_path, &st ) < 0 ) {
777         return AFPERR_PARAM;
778     }
779
780     if ( chdir( volume->v_path ) < 0 ) {
781         return AFPERR_PARAM;
782     }
783
784     if (volume_codepage(obj, volume) < 0) {
785         ret = AFPERR_MISC;
786         goto openvol_err;
787     }
788
789     /* initialize volume variables
790      * FIXME file size
791      */
792     if (utf8_encoding(obj)) {
793         volume->max_filename = UTF8FILELEN_EARLY;
794     }
795     else {
796         volume->max_filename = MACFILELEN;
797     }
798
799     volume->v_flags |= AFPVOL_OPEN;
800     volume->v_cdb = NULL;
801
802     if (utf8_encoding(obj)) {
803         len = convert_string_allocate(CH_UCS2, CH_UTF8_MAC, volume->v_u8mname, namelen, &vol_mname);
804     } else {
805         len = convert_string_allocate(CH_UCS2, obj->options.maccharset, volume->v_macname, namelen, &vol_mname);
806     }
807     if ( !vol_mname || len <= 0) {
808         ret = AFPERR_MISC;
809         goto openvol_err;
810     }
811
812     if ((vol_uname = strrchr(volume->v_path, '/')) == NULL)
813         vol_uname = volume->v_path;
814     else if (vol_uname[1] != '\0')
815         vol_uname++;
816
817     if ((dir = dir_new(vol_mname,
818                        vol_uname,
819                        volume,
820                        DIRDID_ROOT_PARENT,
821                        DIRDID_ROOT,
822                        bfromcstr(volume->v_path),
823                        &st)
824             ) == NULL) {
825         free(vol_mname);
826         LOG(log_error, logtype_afpd, "afp_openvol(%s): malloc: %s", volume->v_path, strerror(errno) );
827         ret = AFPERR_MISC;
828         goto openvol_err;
829     }
830     volume->v_root = dir;
831     curdir = dir;
832
833     if (volume_openDB(obj, volume) < 0) {
834         LOG(log_error, logtype_afpd, "Fatal error: cannot open CNID or invalid CNID backend for %s: %s",
835             volume->v_path, volume->v_cnidscheme);
836         ret = AFPERR_MISC;
837         goto openvol_err;
838     }
839
840     ret  = stat_vol(obj, bitmap, volume, rbuf, rbuflen);
841
842     if (ret == AFP_OK) {
843         /*
844          * If you mount a volume twice, the second time the trash appears on
845          * the desk-top.  That's because the Mac remembers the DID for the
846          * trash (even for volumes in different zones, on different servers).
847          * Just so this works better, we prime the DID cache with the trash,
848          * fixing the trash at DID 17.
849          * FIXME (RL): should it be done inside a CNID backend ? (always returning Trash DID when asked) ?
850          */
851         if ((volume->v_cdb->cnid_db_flags & CNID_FLAG_PERSISTENT)) {
852
853             /* FIXME find db time stamp */
854             if (cnid_getstamp(volume->v_cdb, volume->v_stamp, sizeof(volume->v_stamp)) < 0) {
855                 LOG (log_error, logtype_afpd,
856                      "afp_openvol(%s): Fatal error: Unable to get stamp value from CNID backend",
857                      volume->v_path);
858                 ret = AFPERR_MISC;
859                 goto openvol_err;
860             }
861         }
862
863         const char *msg;
864         if ((msg = atalk_iniparser_getstring(obj->iniconfig, volume->v_configname, "login message",  NULL)) != NULL)
865             setmessage(msg);
866
867         free(vol_mname);
868         server_ipc_volumes(obj);
869         return( AFP_OK );
870     }
871
872 openvol_err:
873     if (volume->v_root) {
874         dir_free( volume->v_root );
875         volume->v_root = NULL;
876     }
877
878     volume->v_flags &= ~AFPVOL_OPEN;
879     if (volume->v_cdb != NULL) {
880         cnid_close(volume->v_cdb);
881         volume->v_cdb = NULL;
882     }
883     free(vol_mname);
884     *rbuflen = 0;
885     return ret;
886 }
887
888 void closevol(const AFPObj *obj, struct vol *vol)
889 {
890     if (!vol)
891         return;
892
893     vol->v_flags &= ~AFPVOL_OPEN;
894
895     of_closevol(obj, vol);
896
897     dir_free( vol->v_root );
898     vol->v_root = NULL;
899     if (vol->v_cdb != NULL) {
900         cnid_close(vol->v_cdb);
901         vol->v_cdb = NULL;
902     }
903
904     if (vol->v_postexec) {
905         afprun(0, vol->v_postexec, NULL);
906     }
907     if (vol->v_root_postexec) {
908         afprun(1, vol->v_root_postexec, NULL);
909     }
910 }
911
912 /* ------------------------- */
913 void close_all_vol(const AFPObj *obj)
914 {
915     struct vol  *ovol;
916     curdir = NULL;
917     for ( ovol = getvolumes(); ovol; ovol = ovol->v_next ) {
918         if ( (ovol->v_flags & AFPVOL_OPEN) ) {
919             ovol->v_flags &= ~AFPVOL_OPEN;
920             closevol(obj, ovol);
921         }
922     }
923 }
924
925 /* ------------------------- */
926 int afp_closevol(AFPObj *obj, char *ibuf, size_t ibuflen _U_, char *rbuf _U_, size_t *rbuflen)
927 {
928     struct vol  *vol;
929     uint16_t   vid;
930
931     *rbuflen = 0;
932     ibuf += 2;
933     memcpy(&vid, ibuf, sizeof( vid ));
934     if (NULL == ( vol = getvolbyvid( vid )) ) {
935         return( AFPERR_PARAM );
936     }
937
938     (void)chdir("/");
939     curdir = NULL;
940     closevol(obj, vol);
941     server_ipc_volumes(obj);
942
943     return( AFP_OK );
944 }
945
946 /* --------------------------
947    poll if a volume is changed by other processes.
948    return
949    0 no attention msg sent
950    1 attention msg sent
951    -1 error (socket closed)
952
953    Note: if attention return -1 no packet has been
954    sent because the buffer is full, we don't care
955    either there's no reader or there's a lot of
956    traffic and another pollvoltime will follow
957 */
958 int  pollvoltime(AFPObj *obj)
959
960 {
961     struct vol       *vol;
962     struct timeval   tv;
963     struct stat      st;
964
965     if (!(obj->afp_version > 21 && obj->options.flags & OPTION_SERVERNOTIF))
966         return 0;
967
968     if ( gettimeofday( &tv, NULL ) < 0 )
969         return 0;
970
971     for ( vol = getvolumes(); vol; vol = vol->v_next ) {
972         if ( (vol->v_flags & AFPVOL_OPEN)  && vol->v_mtime + 30 < tv.tv_sec) {
973             if ( !stat( vol->v_path, &st ) && vol->v_mtime != st.st_mtime ) {
974                 vol->v_mtime = st.st_mtime;
975                 if (!obj->attention(obj->dsi, AFPATTN_NOTIFY | AFPATTN_VOLCHANGED))
976                     return -1;
977                 return 1;
978             }
979         }
980     }
981     return 0;
982 }
983
984 /* ------------------------- */
985 void setvoltime(AFPObj *obj, struct vol *vol)
986 {
987     struct timeval  tv;
988
989     if ( gettimeofday( &tv, NULL ) < 0 ) {
990         LOG(log_error, logtype_afpd, "setvoltime(%s): gettimeofday: %s", vol->v_path, strerror(errno) );
991         return;
992     }
993     if( utime( vol->v_path, NULL ) < 0 ) {
994         /* write of time failed ... probably a read only filesys,
995          * where no other users can interfere, so there's no issue here
996          */
997     }
998
999     /* a little granularity */
1000     if (vol->v_mtime < tv.tv_sec) {
1001         vol->v_mtime = tv.tv_sec;
1002         /* or finder doesn't update free space
1003          * AFP 3.2 and above clients seem to be ok without so many notification
1004          */
1005         if (obj->afp_version < 32 && obj->options.flags & OPTION_SERVERNOTIF) {
1006             obj->attention(obj->dsi, AFPATTN_NOTIFY | AFPATTN_VOLCHANGED);
1007         }
1008     }
1009 }
1010
1011 /* ------------------------- */
1012 int afp_getvolparams(AFPObj *obj, char *ibuf, size_t ibuflen _U_,char *rbuf, size_t *rbuflen)
1013 {
1014     struct vol  *vol;
1015     uint16_t   vid, bitmap;
1016
1017     ibuf += 2;
1018     memcpy(&vid, ibuf, sizeof( vid ));
1019     ibuf += sizeof( vid );
1020     memcpy(&bitmap, ibuf, sizeof( bitmap ));
1021     bitmap = ntohs( bitmap );
1022
1023     if (NULL == ( vol = getvolbyvid( vid )) ) {
1024         *rbuflen = 0;
1025         return( AFPERR_PARAM );
1026     }
1027
1028     return stat_vol(obj, bitmap, vol, rbuf, rbuflen);
1029 }
1030
1031 /* ------------------------- */
1032 int afp_setvolparams(AFPObj *obj _U_, char *ibuf, size_t ibuflen _U_, char *rbuf _U_,  size_t *rbuflen)
1033 {
1034     struct adouble ad;
1035     struct vol  *vol;
1036     uint16_t   vid, bitmap;
1037     uint32_t   aint;
1038
1039     ibuf += 2;
1040     *rbuflen = 0;
1041
1042     memcpy(&vid, ibuf, sizeof( vid ));
1043     ibuf += sizeof( vid );
1044     memcpy(&bitmap, ibuf, sizeof( bitmap ));
1045     bitmap = ntohs( bitmap );
1046     ibuf += sizeof(bitmap);
1047
1048     if (( vol = getvolbyvid( vid )) == NULL ) {
1049         return( AFPERR_PARAM );
1050     }
1051
1052     if ((vol->v_flags & AFPVOL_RO))
1053         return AFPERR_VLOCK;
1054
1055     /* we can only set the backup date. */
1056     if (bitmap != (1 << VOLPBIT_BDATE))
1057         return AFPERR_BITMAP;
1058
1059     ad_init(&ad, vol);
1060     if ( ad_open(&ad,  vol->v_path, ADFLAGS_HF | ADFLAGS_DIR | ADFLAGS_RDWR) < 0 ) {
1061         if (errno == EROFS)
1062             return AFPERR_VLOCK;
1063
1064         return AFPERR_ACCESS;
1065     }
1066
1067     memcpy(&aint, ibuf, sizeof(aint));
1068     ad_setdate(&ad, AD_DATE_BACKUP, aint);
1069     ad_flush(&ad);
1070     ad_close(&ad, ADFLAGS_HF);
1071     return( AFP_OK );
1072 }