2 * Copyright (c) 1990,1993 Regents of The University of Michigan.
3 * Copyright (c) 2010 Frank Lahm
5 * All Rights Reserved. See COPYRIGHT.
10 #endif /* HAVE_CONFIG_H */
15 #include <sys/param.h>
16 #include <sys/socket.h>
19 #include <atalk/dsi.h>
20 #include <atalk/afp.h>
21 #include <atalk/adouble.h>
22 #include <atalk/logger.h>
23 #include <atalk/util.h>
24 #include <atalk/cnid.h>
25 #include <atalk/bstradd.h>
26 #include <atalk/globals.h>
27 #include <atalk/netatalk_conf.h>
32 #include "directory.h"
37 struct ofork *writtenfork;
40 static int getforkparams(const AFPObj *obj, struct ofork *ofork, uint16_t bitmap, char *buf, size_t *buflen)
50 /* can only get the length of the opened fork */
51 if ( ( (bitmap & ((1<<FILPBIT_DFLEN) | (1<<FILPBIT_EXTDFLEN)))
52 && (ofork->of_flags & AFPFORK_RSRC))
54 ( (bitmap & ((1<<FILPBIT_RFLEN) | (1<<FILPBIT_EXTRFLEN)))
55 && (ofork->of_flags & AFPFORK_DATA))) {
56 return( AFPERR_BITMAP );
59 if (! AD_META_OPEN(ofork->of_ad)) {
66 dir = dirlookup(vol, ofork->of_did);
68 if (NULL == (path.m_name = utompath(vol, of_name(ofork), dir->d_did, utf8_encoding(obj)))) {
69 return( AFPERR_MISC );
71 path.u_name = of_name(ofork);
74 if ( bitmap & ( (1<<FILPBIT_DFLEN) | (1<<FILPBIT_EXTDFLEN) |
75 (1<<FILPBIT_FNUM) | (1 << FILPBIT_CDATE) |
76 (1 << FILPBIT_MDATE) | (1 << FILPBIT_BDATE))) {
77 if ( ad_data_fileno( ofork->of_ad ) <= 0 ) {
78 /* 0 is for symlink */
79 if (movecwd(vol, dir) < 0)
80 return( AFPERR_NOOBJ );
81 if ( lstat( path.u_name, st ) < 0 )
82 return( AFPERR_NOOBJ );
84 if ( fstat( ad_data_fileno( ofork->of_ad ), st ) < 0 ) {
85 return( AFPERR_BITMAP );
89 return getmetadata(obj, vol, bitmap, &path, dir, buf, buflen, adp );
92 static off_t get_off_t(char **ibuf, int is64)
98 memcpy(&temp, *ibuf, sizeof( temp ));
99 ret = ntohl(temp); /* ntohl is unsigned */
100 *ibuf += sizeof(temp);
103 memcpy(&temp, *ibuf, sizeof( temp ));
104 *ibuf += sizeof(temp);
105 ret = ntohl(temp)| (ret << 32);
108 ret = (int)ret; /* sign extend */
113 static int set_off_t(off_t offset, char *rbuf, int is64)
120 temp = htonl(offset >> 32);
121 memcpy(rbuf, &temp, sizeof( temp ));
122 rbuf += sizeof(temp);
123 ret = sizeof( temp );
124 offset &= 0xffffffff;
126 temp = htonl(offset);
127 memcpy(rbuf, &temp, sizeof( temp ));
128 ret += sizeof( temp );
133 static int is_neg(int is64, off_t val)
135 if (val < 0 || (sizeof(off_t) == 8 && !is64 && (val & 0x80000000U)))
140 static int sum_neg(int is64, off_t offset, off_t reqcount)
142 if (is_neg(is64, offset +reqcount) )
147 static int fork_setmode(const AFPObj *obj, struct adouble *adp, int eid, int access, int ofrefnum)
158 if (obj->options.flags & OPTION_SHARE_RESERV) {
159 shmd.f_access = (access & OPENACC_RD ? F_RDACC : 0) | (access & OPENACC_WR ? F_WRACC : 0);
160 if (shmd.f_access == 0)
161 /* we must give an access mode, otherwise fcntl will complain */
162 shmd.f_access = F_RDACC;
163 shmd.f_deny = (access & OPENACC_DRD ? F_RDDNY : F_NODNY) | (access & OPENACC_DWR) ? F_WRDNY : 0;
164 shmd.f_id = ofrefnum;
166 int fd = (eid == ADEID_DFORK) ? ad_data_fileno(adp) : ad_reso_fileno(adp);
168 if (fd != -1 && fd != AD_SYMLINK && fcntl(fd, F_SHARE, &shmd) != 0) {
169 LOG(log_debug, logtype_afpd, "fork_setmode: fcntl: %s", strerror(errno));
176 if (! (access & (OPENACC_WR | OPENACC_RD | OPENACC_DWR | OPENACC_DRD))) {
177 return ad_lock(adp, eid, ADLOCK_RD | ADLOCK_FILELOCK, AD_FILELOCK_OPEN_NONE, 1, ofrefnum);
180 if ((access & (OPENACC_RD | OPENACC_DRD))) {
181 if ((readset = ad_testlock(adp, eid, AD_FILELOCK_OPEN_RD)) <0)
183 if ((denyreadset = ad_testlock(adp, eid, AD_FILELOCK_DENY_RD)) <0)
186 if ((access & OPENACC_RD) && denyreadset) {
190 if ((access & OPENACC_DRD) && readset) {
194 /* boolean logic is not enough, because getforkmode is not always telling the
197 if ((access & OPENACC_RD)) {
198 ret = ad_lock(adp, eid, ADLOCK_RD | ADLOCK_FILELOCK, AD_FILELOCK_OPEN_RD, 1, ofrefnum);
202 if ((access & OPENACC_DRD)) {
203 ret = ad_lock(adp, eid, ADLOCK_RD | ADLOCK_FILELOCK, AD_FILELOCK_DENY_RD, 1, ofrefnum);
208 /* ------------same for writing -------------- */
209 if ((access & (OPENACC_WR | OPENACC_DWR))) {
210 if ((writeset = ad_testlock(adp, eid, AD_FILELOCK_OPEN_WR)) <0)
212 if ((denywriteset = ad_testlock(adp, eid, AD_FILELOCK_DENY_WR)) <0)
215 if ((access & OPENACC_WR) && denywriteset) {
219 if ((access & OPENACC_DWR) && writeset) {
223 if ((access & OPENACC_WR)) {
224 ret = ad_lock(adp, eid, ADLOCK_RD | ADLOCK_FILELOCK, AD_FILELOCK_OPEN_WR, 1, ofrefnum);
228 if ((access & OPENACC_DWR)) {
229 ret = ad_lock(adp, eid, ADLOCK_RD | ADLOCK_FILELOCK, AD_FILELOCK_DENY_WR, 1, ofrefnum);
238 /* ----------------------- */
239 int afp_openfork(AFPObj *obj _U_, char *ibuf, size_t ibuflen _U_, char *rbuf, size_t *rbuflen)
243 struct ofork *ofork, *opened;
244 struct adouble *adsame = NULL;
246 int ret, adflags, eid;
248 uint16_t vid, bitmap, access, ofrefnum;
249 char fork, *path, *upath;
257 memcpy(&vid, ibuf, sizeof( vid ));
261 if (NULL == ( vol = getvolbyvid( vid ))) {
262 return( AFPERR_PARAM );
265 memcpy(&did, ibuf, sizeof( did ));
266 ibuf += sizeof( int );
268 if (NULL == ( dir = dirlookup( vol, did ))) {
272 memcpy(&bitmap, ibuf, sizeof( bitmap ));
273 bitmap = ntohs( bitmap );
274 ibuf += sizeof( bitmap );
275 memcpy(&access, ibuf, sizeof( access ));
276 access = ntohs( access );
277 ibuf += sizeof( access );
279 if ((vol->v_flags & AFPVOL_RO) && (access & OPENACC_WR)) {
283 if (NULL == ( s_path = cname( vol, dir, &ibuf ))) {
284 return get_afp_errno(AFPERR_PARAM);
287 if (*s_path->m_name == '\0') {
289 return AFPERR_BADTYPE;
292 /* stat() data fork st is set because it's not a dir */
293 switch ( s_path->st_errno ) {
299 return (access & OPENACC_WR) ? AFPERR_LOCK : AFPERR_ACCESS;
301 LOG(log_error, logtype_afpd, "afp_openfork(%s): %s", s_path->m_name, strerror(errno));
305 upath = s_path->u_name;
306 path = s_path->m_name;
309 if (!vol_unix_priv(vol)) {
310 if (check_access(obj, vol, upath, access ) < 0) {
311 return AFPERR_ACCESS;
314 if (file_access(obj, vol, s_path, access ) < 0) {
315 return AFPERR_ACCESS;
319 if ((opened = of_findname(s_path))) {
320 adsame = opened->of_ad;
323 adflags = ADFLAGS_SETSHRMD;
325 if (fork == OPENFORK_DATA) {
327 adflags |= ADFLAGS_DF;
330 adflags |= ADFLAGS_RF;
331 if (!(access & OPENACC_WR))
332 adflags |= ADFLAGS_NORF;
335 if (access & OPENACC_WR) {
336 adflags |= ADFLAGS_RDWR;
337 if (fork != OPENFORK_DATA)
339 * We only try to create the resource
340 * fork if the user wants to open it for write acess.
342 adflags |= ADFLAGS_CREATE;
344 adflags |= ADFLAGS_RDONLY;
347 if ((ofork = of_alloc(vol, curdir, path, &ofrefnum, eid, adsame, st)) == NULL)
350 LOG(log_debug, logtype_afpd, "afp_openfork(\"%s\", %s, %s)",
351 fullpathname(s_path->u_name),
352 (fork == OPENFORK_DATA) ? "data" : "reso",
353 !(access & OPENACC_WR) ? "O_RDONLY" : "O_RDWR");
357 /* First ad_open(), opens data or ressource fork */
358 if (ad_open(ofork->of_ad, upath, adflags, 0666) < 0) {
371 ret = AFPERR_BADTYPE;
374 LOG(log_error, logtype_afpd, "afp_openfork(%s): ad_open: %s", s_path->m_name, strerror(errno) );
381 * Create metadata if we open rw, otherwise only open existing metadata
383 if (access & OPENACC_WR) {
384 adflags = ADFLAGS_HF | ADFLAGS_RDWR | ADFLAGS_CREATE;
386 adflags = ADFLAGS_HF | ADFLAGS_RDONLY;
389 if (ad_open(ofork->of_ad, upath, adflags, 0666) == 0) {
390 ofork->of_flags |= AFPFORK_META;
395 /* no metadata? We don't care! */
404 ret = AFPERR_BADTYPE;
407 LOG(log_error, logtype_afpd, "afp_openfork(%s): ad_open: %s", s_path->m_name, strerror(errno) );
413 if ((adflags & ADFLAGS_RF) && (ad_get_RF_flags( ofork->of_ad) & O_CREAT)) {
414 if (ad_setname(ofork->of_ad, path)) {
415 ad_flush( ofork->of_ad );
419 if ((ret = getforkparams(obj, ofork, bitmap, rbuf + 2 * sizeof(int16_t), &buflen)) != AFP_OK) {
420 ad_close( ofork->of_ad, adflags | ADFLAGS_SETSHRMD);
424 *rbuflen = buflen + 2 * sizeof( uint16_t );
425 bitmap = htons( bitmap );
426 memcpy(rbuf, &bitmap, sizeof( uint16_t ));
427 rbuf += sizeof( uint16_t );
429 /* check WriteInhibit bit if we have a ressource fork
430 * the test is done here, after some Mac trafic capture
432 if (ad_meta_fileno(ofork->of_ad) != -1) { /* META */
433 ad_getattr(ofork->of_ad, &bshort);
434 if ((bshort & htons(ATTRBIT_NOWRITE)) && (access & OPENACC_WR)) {
435 ad_close( ofork->of_ad, adflags | ADFLAGS_SETSHRMD);
438 memcpy(rbuf, &ofrefnum, sizeof(ofrefnum));
439 return(AFPERR_OLOCK);
444 * synchronization locks:
447 /* don't try to lock non-existent rforks. */
448 if ((eid == ADEID_DFORK)
449 || (ad_reso_fileno(ofork->of_ad) != -1)
450 || (ofork->of_ad->ad_vers == AD_VERSION_EA)) {
451 ret = fork_setmode(obj, ofork->of_ad, eid, access, ofrefnum);
452 /* can we access the fork? */
454 ofork->of_flags |= AFPFORK_ERROR;
456 ad_close( ofork->of_ad, adflags | ADFLAGS_SETSHRMD);
459 case EAGAIN: /* return data anyway */
463 memcpy(rbuf, &ofrefnum, sizeof(ofrefnum));
464 return( AFPERR_DENYCONF );
468 LOG(log_error, logtype_afpd, "afp_openfork(%s): ad_lock: %s", s_path->m_name, strerror(ret) );
469 return( AFPERR_PARAM );
472 if ((access & OPENACC_WR))
473 ofork->of_flags |= AFPFORK_ACCWR;
475 /* the file may be open read only without ressource fork */
476 if ((access & OPENACC_RD))
477 ofork->of_flags |= AFPFORK_ACCRD;
479 LOG(log_debug, logtype_afpd, "afp_openfork(\"%s\"): fork: %" PRIu16,
480 fullpathname(s_path->m_name), ofork->of_refnum);
482 memcpy(rbuf, &ofrefnum, sizeof(ofrefnum));
488 return (access & OPENACC_WR) ? AFPERR_LOCK : AFPERR_ACCESS;
492 int afp_setforkparams(AFPObj *obj, char *ibuf, size_t ibuflen, char *rbuf _U_, size_t *rbuflen)
496 uint16_t ofrefnum, bitmap;
504 memcpy(&ofrefnum, ibuf, sizeof( ofrefnum ));
505 ibuf += sizeof( ofrefnum );
507 memcpy(&bitmap, ibuf, sizeof(bitmap));
508 bitmap = ntohs(bitmap);
509 ibuf += sizeof( bitmap );
512 if (NULL == ( ofork = of_find( ofrefnum )) ) {
513 LOG(log_error, logtype_afpd, "afp_setforkparams: of_find(%d) could not locate fork", ofrefnum );
514 return( AFPERR_PARAM );
517 if (ofork->of_vol->v_flags & AFPVOL_RO)
520 if ((ofork->of_flags & AFPFORK_ACCWR) == 0)
521 return AFPERR_ACCESS;
523 if ( ofork->of_flags & AFPFORK_DATA) {
525 } else if (ofork->of_flags & AFPFORK_RSRC) {
530 if ( ( (bitmap & ( (1<<FILPBIT_DFLEN) | (1<<FILPBIT_EXTDFLEN) ))
531 && eid == ADEID_RFORK
533 ( (bitmap & ( (1<<FILPBIT_RFLEN) | (1<<FILPBIT_EXTRFLEN) ))
534 && eid == ADEID_DFORK)) {
535 return AFPERR_BITMAP;
539 if ((bitmap & ( (1<<FILPBIT_EXTDFLEN) | (1<<FILPBIT_EXTRFLEN) ))) {
540 if (obj->afp_version >= 30) {
544 return AFPERR_BITMAP;
547 if (ibuflen < 2+ sizeof(ofrefnum) + sizeof(bitmap) + is64 +4)
548 return AFPERR_PARAM ;
550 size = get_off_t(&ibuf, is64);
553 return AFPERR_PARAM; /* Some MacOS don't return an error they just don't change the size! */
556 if (bitmap == (1<<FILPBIT_DFLEN) || bitmap == (1<<FILPBIT_EXTDFLEN)) {
557 st_size = ad_size(ofork->of_ad, eid);
559 if (st_size > size &&
560 ad_tmplock(ofork->of_ad, eid, ADLOCK_WR, size, st_size -size, ofork->of_refnum) < 0)
561 goto afp_setfork_err;
563 err = ad_dtruncate( ofork->of_ad, size );
565 ad_tmplock(ofork->of_ad, eid, ADLOCK_CLR, size, st_size -size, ofork->of_refnum);
567 goto afp_setfork_err;
568 } else if (bitmap == (1<<FILPBIT_RFLEN) || bitmap == (1<<FILPBIT_EXTRFLEN)) {
569 ad_refresh(NULL, ofork->of_ad );
571 st_size = ad_size(ofork->of_ad, eid);
573 if (st_size > size &&
574 ad_tmplock(ofork->of_ad, eid, ADLOCK_WR, size, st_size -size, ofork->of_refnum) < 0) {
575 goto afp_setfork_err;
578 err = ad_rtruncate(ofork->of_ad, size);
580 ad_tmplock(ofork->of_ad, eid, ADLOCK_CLR, size, st_size -size, ofork->of_refnum);
582 goto afp_setfork_err;
584 if (ad_flush( ofork->of_ad ) < 0) {
585 LOG(log_error, logtype_afpd, "afp_setforkparams(%s): ad_flush: %s", of_name(ofork), strerror(errno) );
589 return AFPERR_BITMAP;
592 if ( flushfork( ofork ) < 0 ) {
593 LOG(log_error, logtype_afpd, "afp_setforkparams(%s): flushfork: %s", of_name(ofork), strerror(errno) );
608 return AFPERR_ACCESS;
612 LOG(log_error, logtype_afpd, "afp_setforkparams: DISK FULL");
620 /* for this to work correctly, we need to check for locks before each
621 * read and write. that's most easily handled by always doing an
622 * appropriate check before each ad_read/ad_write. other things
623 * that can change files like truncate are handled internally to those
626 #define ENDBIT(a) ((a) & 0x80)
627 #define UNLOCKBIT(a) ((a) & 0x01)
630 /* ---------------------- */
631 static int byte_lock(AFPObj *obj _U_, char *ibuf, size_t ibuflen _U_, char *rbuf, size_t *rbuflen, int is64)
634 off_t offset, length;
642 /* figure out parameters */
644 flags = *ibuf; /* first bit = endflag, lastbit = lockflag */
646 memcpy(&ofrefnum, ibuf, sizeof(ofrefnum));
647 ibuf += sizeof(ofrefnum);
649 if (NULL == ( ofork = of_find( ofrefnum )) ) {
650 LOG(log_error, logtype_afpd, "byte_lock: of_find(%d) could not locate fork", ofrefnum );
651 return( AFPERR_PARAM );
654 if ( ofork->of_flags & AFPFORK_DATA) {
656 } else if (ofork->of_flags & AFPFORK_RSRC) {
661 offset = get_off_t(&ibuf, is64);
662 length = get_off_t(&ibuf, is64);
665 length = BYTELOCK_MAX;
666 else if (!length || is_neg(is64, length)) {
668 } else if ((length >= AD_FILELOCK_BASE) && -1 == (ad_reso_fileno(ofork->of_ad))) { /* HF ?*/
673 offset += ad_size(ofork->of_ad, eid);
674 /* FIXME what do we do if file size > 2 GB and
675 it's not byte_lock_ext?
678 if (offset < 0) /* error if we have a negative offset */
681 /* if the file is a read-only file, we use read locks instead of
682 * write locks. that way, we can prevent anyone from initiating
684 lockop = UNLOCKBIT(flags) ? ADLOCK_CLR : ADLOCK_WR;
685 if (ad_lock(ofork->of_ad, eid, lockop, offset, length,
686 ofork->of_refnum) < 0) {
690 return UNLOCKBIT(flags) ? AFPERR_NORANGE : AFPERR_LOCK;
696 return UNLOCKBIT(flags) ? AFPERR_NORANGE : AFPERR_RANGEOVR;
704 *rbuflen = set_off_t (offset, rbuf, is64);
708 /* --------------------------- */
709 int afp_bytelock(AFPObj *obj, char *ibuf, size_t ibuflen, char *rbuf, size_t *rbuflen)
711 return byte_lock ( obj, ibuf, ibuflen, rbuf, rbuflen , 0);
714 /* --------------------------- */
715 int afp_bytelock_ext(AFPObj *obj, char *ibuf, size_t ibuflen, char *rbuf, size_t *rbuflen)
717 return byte_lock ( obj, ibuf, ibuflen, rbuf, rbuflen , 1);
723 * Read *rbuflen bytes from fork at offset
725 * @param ofork (r) fork handle
726 * @param eid (r) data fork or ressource fork entry id
727 * @param offset (r) offset
728 * @param rbuf (r) data buffer
729 * @param rbuflen (rw) in: number of bytes to read, out: bytes read
731 * @return AFP status code
733 static int read_file(const struct ofork *ofork, int eid, off_t offset, char *rbuf, size_t *rbuflen)
739 cc = ad_read(ofork->of_ad, eid, offset, rbuf, *rbuflen);
741 LOG(log_error, logtype_afpd, "afp_read(%s): ad_read: %s", of_name(ofork), strerror(errno) );
743 return( AFPERR_PARAM );
747 if ((size_t)cc < *rbuflen)
752 static int read_fork(AFPObj *obj, char *ibuf, size_t ibuflen _U_, char *rbuf, size_t *rbuflen, int is64)
756 off_t offset, saveoff, reqcount, savereqcount, size;
761 /* we break the AFP spec here by not supporting nlmask and nlchar anymore */
764 memcpy(&ofrefnum, ibuf, sizeof( ofrefnum ));
765 ibuf += sizeof( u_short );
767 if (NULL == ( ofork = of_find( ofrefnum )) ) {
768 LOG(log_error, logtype_afpd, "afp_read: of_find(%d) could not locate fork", ofrefnum );
773 if ((ofork->of_flags & AFPFORK_ACCRD) == 0) {
778 if ( ofork->of_flags & AFPFORK_DATA) {
780 } else if (ofork->of_flags & AFPFORK_RSRC) {
782 } else { /* fork wasn't opened. this should never really happen. */
787 offset = get_off_t(&ibuf, is64);
788 reqcount = get_off_t(&ibuf, is64);
790 /* zero request count */
796 /* reqcount isn't always truthful. we need to deal with that. */
797 size = ad_size(ofork->of_ad, eid);
799 LOG(log_debug, logtype_afpd,
800 "afp_read(fork: %" PRIu16 " [%s], off: %" PRIu64 ", len: %" PRIu64 ", size: %" PRIu64 ")",
801 ofork->of_refnum, (ofork->of_flags & AFPFORK_DATA) ? "data" : "reso", offset, reqcount, size);
808 /* subtract off the offset */
809 if (reqcount + offset > size) {
810 reqcount = size - offset;
814 savereqcount = reqcount;
817 if (reqcount < 0 || offset < 0) {
822 if (obj->options.flags & OPTION_AFP_READ_LOCK) {
823 if (ad_tmplock(ofork->of_ad, eid, ADLOCK_RD, offset, reqcount, ofork->of_refnum) < 0) {
830 if (!(eid == ADEID_DFORK && ad_data_fileno(ofork->of_ad) == AD_SYMLINK) &&
831 !(obj->options.flags & OPTION_NOSENDFILE)) {
832 int fd = ad_readfile_init(ofork->of_ad, eid, &offset, 0);
833 if (dsi_stream_read_file(dsi, fd, offset, reqcount, err) < 0) {
834 LOG(log_error, logtype_afpd, "afp_read(%s): ad_readfile: %s",
835 of_name(ofork), strerror(errno));
842 *rbuflen = MIN(reqcount, *rbuflen);
844 err = read_file(ofork, eid, offset, rbuf, rbuflen);
848 LOG(log_debug, logtype_afpd,
849 "afp_read(name: \"%s\", offset: %jd, reqcount: %jd): got %jd bytes from file",
850 of_name(ofork), (intmax_t)offset, (intmax_t)reqcount, (intmax_t)*rbuflen);
855 * dsi_readinit() returns size of next read buffer. by this point,
856 * we know that we're sending some data. if we fail, something
859 if ((cc = dsi_readinit(dsi, rbuf, *rbuflen, reqcount, err)) < 0)
863 while (*rbuflen > 0) {
864 cc = read_file(ofork, eid, offset, rbuf, rbuflen);
869 /* dsi_read() also returns buffer size of next allocation */
870 cc = dsi_read(dsi, rbuf, *rbuflen); /* send it off */
879 LOG(log_error, logtype_afpd, "afp_read(%s): %s", of_name(ofork), strerror(errno));
881 if (obj->options.flags & OPTION_AFP_READ_LOCK)
882 ad_tmplock(ofork->of_ad, eid, ADLOCK_CLR, saveoff, savereqcount, ofork->of_refnum);
883 obj->exit(EXITERR_CLNT);
886 if (obj->options.flags & OPTION_AFP_READ_LOCK)
887 ad_tmplock(ofork->of_ad, eid, ADLOCK_CLR, saveoff, savereqcount, ofork->of_refnum);
895 /* ---------------------- */
896 int afp_read(AFPObj *obj, char *ibuf, size_t ibuflen, char *rbuf, size_t *rbuflen)
898 return read_fork(obj, ibuf, ibuflen, rbuf, rbuflen, 0);
901 /* ---------------------- */
902 int afp_read_ext(AFPObj *obj, char *ibuf, size_t ibuflen, char *rbuf, size_t *rbuflen)
904 return read_fork(obj, ibuf, ibuflen, rbuf, rbuflen, 1);
907 /* ---------------------- */
908 int afp_flush(AFPObj *obj _U_, char *ibuf, size_t ibuflen _U_, char *rbuf _U_, size_t *rbuflen)
916 memcpy(&vid, ibuf, sizeof(vid));
917 if (NULL == ( vol = getvolbyvid( vid )) ) {
918 return( AFPERR_PARAM );
925 int afp_flushfork(AFPObj *obj _U_, char *ibuf, size_t ibuflen _U_, char *rbuf _U_, size_t *rbuflen)
932 memcpy(&ofrefnum, ibuf, sizeof( ofrefnum ));
934 if (NULL == ( ofork = of_find( ofrefnum )) ) {
935 LOG(log_error, logtype_afpd, "afp_flushfork: of_find(%d) could not locate fork", ofrefnum );
936 return( AFPERR_PARAM );
939 LOG(log_debug, logtype_afpd, "afp_flushfork(fork: %s)",
940 (ofork->of_flags & AFPFORK_DATA) ? "d" : "r");
942 if ( flushfork( ofork ) < 0 ) {
943 LOG(log_error, logtype_afpd, "afp_flushfork(%s): %s", of_name(ofork), strerror(errno) );
951 There is a lot to tell about fsync, fdatasync, F_FULLFSYNC.
952 fsync(2) on OSX is implemented differently than on other platforms.
953 see: http://mirror.linux.org.au/pub/linux.conf.au/2007/video/talks/278.pdf.
955 int afp_syncfork(AFPObj *obj _U_, char *ibuf, size_t ibuflen _U_, char *rbuf _U_, size_t *rbuflen)
963 memcpy(&ofrefnum, ibuf, sizeof(ofrefnum));
964 ibuf += sizeof( ofrefnum );
966 if (NULL == ( ofork = of_find( ofrefnum )) ) {
967 LOG(log_error, logtype_afpd, "afpd_syncfork: of_find(%d) could not locate fork", ofrefnum );
968 return( AFPERR_PARAM );
971 LOG(log_debug, logtype_afpd, "afp_syncfork(fork: %s)",
972 (ofork->of_flags & AFPFORK_DATA) ? "d" : "r");
974 if ( flushfork( ofork ) < 0 ) {
975 LOG(log_error, logtype_afpd, "flushfork(%s): %s", of_name(ofork), strerror(errno) );
982 /* this is very similar to closefork */
983 int flushfork(struct ofork *ofork)
987 int err = 0, doflush = 0;
989 if ( ad_data_fileno( ofork->of_ad ) != -1 &&
990 fsync( ad_data_fileno( ofork->of_ad )) < 0 ) {
991 LOG(log_error, logtype_afpd, "flushfork(%s): dfile(%d) %s",
992 of_name(ofork), ad_data_fileno(ofork->of_ad), strerror(errno) );
996 if ( ad_reso_fileno( ofork->of_ad ) != -1 && /* HF */
997 (ofork->of_flags & AFPFORK_RSRC)) {
999 /* read in the rfork length */
1000 ad_refresh(NULL, ofork->of_ad);
1002 /* set the date if we're dirty */
1003 if ((ofork->of_flags & AFPFORK_DIRTY) && !gettimeofday(&tv, NULL)) {
1004 ad_setdate(ofork->of_ad, AD_DATE_MODIFY|AD_DATE_UNIX, tv.tv_sec);
1005 ofork->of_flags &= ~AFPFORK_DIRTY;
1009 /* flush the header */
1010 if (doflush && ad_flush(ofork->of_ad) < 0)
1013 if (fsync( ad_reso_fileno( ofork->of_ad )) < 0)
1017 LOG(log_error, logtype_afpd, "flushfork(%s): hfile(%d) %s",
1018 of_name(ofork), ad_reso_fileno(ofork->of_ad), strerror(errno) );
1024 int afp_closefork(AFPObj *obj, char *ibuf, size_t ibuflen _U_, char *rbuf _U_, size_t *rbuflen)
1026 struct ofork *ofork;
1031 memcpy(&ofrefnum, ibuf, sizeof( ofrefnum ));
1033 if (NULL == ( ofork = of_find( ofrefnum )) ) {
1034 LOG(log_error, logtype_afpd, "afp_closefork: of_find(%d) could not locate fork", ofrefnum );
1035 return( AFPERR_PARAM );
1038 LOG(log_debug, logtype_afpd, "afp_closefork(fork: %" PRIu16 " [%s])",
1039 ofork->of_refnum, (ofork->of_flags & AFPFORK_DATA) ? "data" : "rsrc");
1041 if (of_closefork(obj, ofork) < 0 ) {
1042 LOG(log_error, logtype_afpd, "afp_closefork(%s): of_closefork: %s", of_name(ofork), strerror(errno) );
1043 return( AFPERR_PARAM );
1050 static ssize_t write_file(struct ofork *ofork, int eid,
1051 off_t offset, char *rbuf,
1057 LOG(log_debug, logtype_afpd, "write_file(off: %ju, size: %zu)",
1058 (uintmax_t)offset, rbuflen);
1060 if (( cc = ad_write(ofork->of_ad, eid, offset, 0,
1061 rbuf, rbuflen)) < 0 ) {
1066 LOG(log_error, logtype_afpd, "write_file: DISK FULL");
1067 return( AFPERR_DFULL );
1069 return AFPERR_ACCESS;
1071 LOG(log_error, logtype_afpd, "afp_write(%s): ad_write: %s", of_name(ofork), strerror(errno) );
1072 return( AFPERR_PARAM );
1081 * FPWrite. NOTE: on an error, we always use afp_write_err as
1082 * the client may have sent us a bunch of data that's not reflected
1083 * in reqcount et al.
1085 static int write_fork(AFPObj *obj, char *ibuf, size_t ibuflen _U_, char *rbuf, size_t *rbuflen, int is64)
1087 struct ofork *ofork;
1088 off_t offset, saveoff, reqcount, oldsize, newsize;
1089 int endflag, eid, err = AFP_OK;
1092 DSI *dsi = obj->dsi;
1093 char *rcvbuf = dsi->buffer;
1094 size_t rcvbuflen = dsi->dsireadbuf * dsi->server_quantum;
1096 /* figure out parameters */
1098 endflag = ENDBIT(*ibuf);
1100 memcpy(&ofrefnum, ibuf, sizeof( ofrefnum ));
1101 ibuf += sizeof( ofrefnum );
1103 offset = get_off_t(&ibuf, is64);
1104 reqcount = get_off_t(&ibuf, is64);
1106 if (NULL == ( ofork = of_find( ofrefnum )) ) {
1107 LOG(log_error, logtype_afpd, "afp_write: of_find(%d) could not locate fork", ofrefnum );
1112 LOG(log_debug, logtype_afpd, "afp_write(fork: %" PRIu16 " [%s], off: %" PRIu64 ", size: %" PRIu64 ")",
1113 ofork->of_refnum, (ofork->of_flags & AFPFORK_DATA) ? "data" : "reso", offset, reqcount);
1115 if ((ofork->of_flags & AFPFORK_ACCWR) == 0) {
1116 err = AFPERR_ACCESS;
1121 writtenfork = ofork;
1124 if ( ofork->of_flags & AFPFORK_DATA) {
1126 } else if (ofork->of_flags & AFPFORK_RSRC) {
1129 err = AFPERR_ACCESS; /* should never happen */
1133 oldsize = ad_size(ofork->of_ad, eid);
1137 /* handle bogus parameters */
1138 if (reqcount < 0 || offset < 0) {
1143 newsize = ((offset + reqcount) > oldsize) ? (offset + reqcount) : oldsize;
1145 /* offset can overflow on 64-bit capable filesystems.
1146 * report disk full if that's going to happen. */
1147 if (sum_neg(is64, offset, reqcount)) {
1148 LOG(log_error, logtype_afpd, "write_fork: DISK FULL");
1153 if (!reqcount) { /* handle request counts of 0 */
1155 *rbuflen = set_off_t (offset, rbuf, is64);
1160 if (obj->options.flags & OPTION_AFP_READ_LOCK) {
1161 if (ad_tmplock(ofork->of_ad, eid, ADLOCK_WR, saveoff, reqcount, ofork->of_refnum) < 0) {
1167 /* find out what we have already */
1168 cc = dsi_writeinit(dsi, rcvbuf, rcvbuflen);
1170 if (!cc || (cc = write_file(ofork, eid, offset, rcvbuf, cc)) < 0) {
1171 dsi_writeflush(dsi);
1173 if (obj->options.flags & OPTION_AFP_READ_LOCK)
1174 ad_tmplock(ofork->of_ad, eid, ADLOCK_CLR, saveoff, reqcount, ofork->of_refnum);
1180 #if 0 /*def HAVE_SENDFILE_WRITE*/
1181 if ((cc = ad_writefile(ofork->of_ad, eid, dsi->socket, offset, dsi->datasize)) < 0) {
1189 LOG(log_error, logtype_afpd, "afp_write: ad_writefile: %s", strerror(errno) );
1190 goto afp_write_loop;
1192 dsi_writeflush(dsi);
1194 if (obj->options.flags & OPTION_AFP_READ_LOCK)
1195 ad_tmplock(ofork->of_ad, eid, ADLOCK_CLR, saveoff, reqcount, ofork->of_refnum);
1200 goto afp_write_done;
1201 #endif /* 0, was HAVE_SENDFILE_WRITE */
1203 /* loop until everything gets written. currently
1204 * dsi_write handles the end case by itself. */
1205 while ((cc = dsi_write(dsi, rcvbuf, rcvbuflen))) {
1207 if ((cc = write_file(ofork, eid, offset, rcvbuf, cc)) < 0) {
1208 dsi_writeflush(dsi);
1210 if (obj->options.flags & OPTION_AFP_READ_LOCK)
1211 ad_tmplock(ofork->of_ad, eid, ADLOCK_CLR, saveoff, reqcount, ofork->of_refnum);
1215 LOG(log_debug, logtype_afpd, "afp_write: wrote: %jd, offset: %jd",
1216 (intmax_t)cc, (intmax_t)offset);
1221 if (obj->options.flags & OPTION_AFP_READ_LOCK)
1222 ad_tmplock(ofork->of_ad, eid, ADLOCK_CLR, saveoff, reqcount, ofork->of_refnum);
1223 if ( ad_meta_fileno( ofork->of_ad ) != -1 ) /* META */
1224 ofork->of_flags |= AFPFORK_DIRTY;
1226 /* we have modified any fork, remember until close_fork */
1227 ofork->of_flags |= AFPFORK_MODIFIED;
1229 /* update write count */
1230 ofork->of_vol->v_appended += (newsize > oldsize) ? (newsize - oldsize) : 0;
1232 *rbuflen = set_off_t (offset, rbuf, is64);
1236 dsi_writeinit(obj->dsi, rbuf, *rbuflen);
1237 dsi_writeflush(obj->dsi);
1239 if (err != AFP_OK) {
1245 /* ---------------------------- */
1246 int afp_write(AFPObj *obj, char *ibuf, size_t ibuflen, char *rbuf, size_t *rbuflen)
1248 return write_fork(obj, ibuf, ibuflen, rbuf, rbuflen, 0);
1251 /* ----------------------------
1252 * FIXME need to deal with SIGXFSZ signal
1254 int afp_write_ext(AFPObj *obj, char *ibuf, size_t ibuflen, char *rbuf, size_t *rbuflen)
1256 return write_fork(obj, ibuf, ibuflen, rbuf, rbuflen, 1);
1259 /* ---------------------------- */
1260 int afp_getforkparams(AFPObj *obj, char *ibuf, size_t ibuflen _U_, char *rbuf, size_t *rbuflen)
1262 struct ofork *ofork;
1264 uint16_t ofrefnum, bitmap;
1267 memcpy(&ofrefnum, ibuf, sizeof( ofrefnum ));
1268 ibuf += sizeof( ofrefnum );
1269 memcpy(&bitmap, ibuf, sizeof( bitmap ));
1270 bitmap = ntohs( bitmap );
1271 ibuf += sizeof( bitmap );
1274 if (NULL == ( ofork = of_find( ofrefnum )) ) {
1275 LOG(log_error, logtype_afpd, "afp_getforkparams: of_find(%d) could not locate fork", ofrefnum );
1276 return( AFPERR_PARAM );
1279 if (AD_META_OPEN(ofork->of_ad)) {
1280 if ( ad_refresh(NULL, ofork->of_ad ) < 0 ) {
1281 LOG(log_error, logtype_afpd, "getforkparams(%s): ad_refresh: %s", of_name(ofork), strerror(errno) );
1282 return( AFPERR_PARAM );
1286 if (AFP_OK != (ret = getforkparams(obj, ofork, bitmap, rbuf + sizeof( u_short ), &buflen ))) {
1290 *rbuflen = buflen + sizeof( u_short );
1291 bitmap = htons( bitmap );
1292 memcpy(rbuf, &bitmap, sizeof( bitmap ));