]> arthur.barton.de Git - netatalk.git/blob - etc/afpd/afp_dsi.c
c9c3da92e2c8cc97282268393165dab09ce3267d
[netatalk.git] / etc / afpd / afp_dsi.c
1 /*
2  * Copyright (c) 1999 Adrian Sun (asun@zoology.washington.edu)
3  * Copyright (c) 1990,1993 Regents of The University of Michigan.
4  * All Rights Reserved.  See COPYRIGHT.
5  *
6  * modified from main.c. this handles afp over tcp.
7  */
8
9 #ifdef HAVE_CONFIG_H
10 #include "config.h"
11 #endif /* HAVE_CONFIG_H */
12
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <signal.h>
16 #include <string.h>
17 #include <errno.h>
18 #ifdef HAVE_UNISTD_H
19 #include <unistd.h>
20 #endif /* HAVE_UNISTD_H */
21 #include <sys/socket.h>
22 #include <sys/time.h>
23 #ifdef HAVE_SYS_STAT_H
24 #include <sys/stat.h>
25 #endif /* HAVE_SYS_STAT_H */
26 #include <netinet/in.h>
27 #include <arpa/inet.h>
28 #include <atalk/logger.h>
29 #include <setjmp.h>
30
31 #include <atalk/dsi.h>
32 #include <atalk/compat.h>
33 #include <atalk/util.h>
34 #include <atalk/uuid.h>
35
36 #include "globals.h"
37 #include "switch.h"
38 #include "auth.h"
39 #include "fork.h"
40 #include "dircache.h"
41
42 #ifdef FORCE_UIDGID
43 #warning UIDGID
44 #include "uid.h"
45 #endif /* FORCE_UIDGID */
46
47 /* 
48  * We generally pass this from afp_over_dsi to all afp_* funcs, so it should already be
49  * available everywhere. Unfortunately some funcs (eg acltoownermode) need acces to it
50  * but are deeply nested in the function chain with the caller already without acces to it.
51  * Changing this would require adding a reference to the caller which itself might be
52  * called in many places (eg acltoownermode is called from accessmode).
53  * The only sane way out is providing a copy of it here:
54  */
55 AFPObj *AFPobj = NULL;
56
57 typedef struct {
58     uint16_t DSIreqID;
59     uint8_t  AFPcommand;
60     uint32_t result;
61 } rc_elem_t;
62
63 /*
64  * AFP replay cache:
65  * - fix sized array
66  * - indexed just by taking DSIreqID mod REPLAYCACHE_SIZE
67  */
68 static rc_elem_t replaycache[REPLAYCACHE_SIZE];
69
70 static sigjmp_buf recon_jmp;
71 static void afp_dsi_close(AFPObj *obj)
72 {
73     DSI *dsi = obj->handle;
74
75     close(obj->ipc_fd);
76     obj->ipc_fd = -1;
77
78     /* we may have been called from a signal handler caught when afpd was running
79      * as uid 0, that's the wrong user for volume's prexec_close scripts if any,
80      * restore our login user
81      */
82     if (geteuid() != obj->uid) {
83         if (seteuid( obj->uid ) < 0) {
84             LOG(log_error, logtype_afpd, "can't seteuid(%u) back %s: uid: %u, euid: %u", 
85                 obj->uid, strerror(errno), getuid(), geteuid());
86             exit(EXITERR_SYS);
87         }
88     }
89
90     close_all_vol();
91     if (obj->logout)
92         (*obj->logout)();
93
94     LOG(log_note, logtype_afpd, "AFP statistics: %.2f KB read, %.2f KB written",
95         dsi->read_count/1024.0, dsi->write_count/1024.0);
96     log_dircache_stat();
97
98     dsi_close(dsi);
99 }
100
101 /* -------------------------------
102  * SIGTERM
103  * a little bit of code duplication. 
104  */
105 static void afp_dsi_die(int sig)
106 {
107     DSI *dsi = (DSI *)AFPobj->handle;
108
109     if (dsi->flags & DSI_RECONINPROG) {
110         /* Primary reconnect succeeded, got SIGTERM from afpd parent */
111         dsi->flags &= ~DSI_RECONINPROG;
112         return; /* this returns to afp_disconnect */
113     }
114
115     if (dsi->flags & DSI_DISCONNECTED) {
116         LOG(log_note, logtype_afpd, "Disconnected session terminating");
117         exit(0);
118     }
119
120     dsi_attention(AFPobj->handle, AFPATTN_SHUTDOWN);
121     afp_dsi_close(AFPobj);
122    if (sig) /* if no signal, assume dieing because logins are disabled &
123                 don't log it (maintenance mode)*/
124         LOG(log_info, logtype_afpd, "Connection terminated");
125     if (sig == SIGTERM || sig == SIGALRM) {
126         exit( 0 );
127     }
128     else {
129         exit(sig);
130     }
131 }
132
133 /* SIGURG handler (primary reconnect) */
134 static void afp_dsi_transfer_session(int sig _U_)
135 {
136     uint16_t dsiID;
137     int socket;
138     DSI *dsi = (DSI *)AFPobj->handle;
139
140     LOG(log_debug, logtype_afpd, "afp_dsi_transfer_session: got SIGURG, trying to receive session");
141
142     if (readt(AFPobj->ipc_fd, &dsiID, 2, 0, 2) != 2) {
143         LOG(log_error, logtype_afpd, "afp_dsi_transfer_session: couldn't receive DSI id, goodbye");
144         afp_dsi_close(AFPobj);
145         exit(EXITERR_SYS);
146     }
147
148     if ((socket = recv_fd(AFPobj->ipc_fd, 1)) == -1) {
149         LOG(log_error, logtype_afpd, "afp_dsi_transfer_session: couldn't receive session fd, goodbye");
150         afp_dsi_close(AFPobj);
151         exit(EXITERR_SYS);
152     }
153
154     LOG(log_debug, logtype_afpd, "afp_dsi_transfer_session: received socket fd: %i", socket);
155
156     dsi->proto_close(dsi);
157     dsi->socket = socket;
158     dsi->flags = DSI_RECONSOCKET;
159     dsi->datalen = 0;
160     dsi->eof = dsi->start = dsi->buffer;
161     dsi->in_write = 0;
162     dsi->header.dsi_requestID = dsiID;
163     dsi->header.dsi_command = DSIFUNC_CMD;
164
165     /*
166      * The session transfer happens in the middle of FPDisconnect old session, thus we
167      * have to send the reply now.
168      */
169     if (!dsi_cmdreply(dsi, AFP_OK)) {
170         LOG(log_error, logtype_afpd, "dsi_cmdreply: %s", strerror(errno) );
171         afp_dsi_close(AFPobj);
172         exit(EXITERR_CLNT);
173     }
174
175     LOG(log_note, logtype_afpd, "afp_dsi_transfer_session: succesfull primary reconnect");
176     /* 
177      * Now returning from this signal handler return to dsi_receive which should start
178      * reading/continuing from the connected socket that was passed via the parent from
179      * another session. The parent will terminate that session.
180      */
181     siglongjmp(recon_jmp, 1);
182 }
183
184 /* ------------------- */
185 static void afp_dsi_timedown(int sig _U_)
186 {
187     struct sigaction    sv;
188     struct itimerval    it;
189     DSI                 *dsi = (DSI *)AFPobj->handle;
190     dsi->flags |= DSI_DIE;
191     /* shutdown and don't reconnect. server going down in 5 minutes. */
192     setmessage("The server is going down for maintenance.");
193     if (dsi_attention(AFPobj->handle, AFPATTN_SHUTDOWN | AFPATTN_NORECONNECT |
194                   AFPATTN_MESG | AFPATTN_TIME(5)) < 0) {
195         DSI *dsi = (DSI *)AFPobj->handle;
196         dsi->down_request = 1;
197     }                  
198
199     it.it_interval.tv_sec = 0;
200     it.it_interval.tv_usec = 0;
201     it.it_value.tv_sec = 300;
202     it.it_value.tv_usec = 0;
203
204     if ( setitimer( ITIMER_REAL, &it, NULL ) < 0 ) {
205         LOG(log_error, logtype_afpd, "afp_timedown: setitimer: %s", strerror(errno) );
206         afp_dsi_die(EXITERR_SYS);
207     }
208     memset(&sv, 0, sizeof(sv));
209     sv.sa_handler = afp_dsi_die;
210     sigemptyset( &sv.sa_mask );
211     sigaddset(&sv.sa_mask, SIGHUP);
212     sigaddset(&sv.sa_mask, SIGTERM);
213     sv.sa_flags = SA_RESTART;
214     if ( sigaction( SIGALRM, &sv, NULL ) < 0 ) {
215         LOG(log_error, logtype_afpd, "afp_timedown: sigaction: %s", strerror(errno) );
216         afp_dsi_die(EXITERR_SYS);
217     }
218
219     /* ignore myself */
220     sv.sa_handler = SIG_IGN;
221     sigemptyset( &sv.sa_mask );
222     sv.sa_flags = SA_RESTART;
223     if ( sigaction( SIGUSR1, &sv, NULL ) < 0 ) {
224         LOG(log_error, logtype_afpd, "afp_timedown: sigaction SIGHUP: %s", strerror(errno) );
225         afp_dsi_die(EXITERR_SYS);
226     }
227 }
228
229 /* ---------------------------------
230  * SIGHUP reload configuration file
231  */
232 volatile int reload_request = 0;
233
234 static void afp_dsi_reload(int sig _U_)
235 {
236     reload_request = 1;
237 }
238
239 /* ---------------------------------
240  * SIGINT: enable max_debug LOGging
241  */
242 static volatile sig_atomic_t debug_request = 0;
243
244 static void afp_dsi_debug(int sig _U_)
245 {
246     debug_request = 1;
247 }
248
249 /* ---------------------- */
250 static void afp_dsi_getmesg (int sig _U_)
251 {
252     DSI *dsi = (DSI *)AFPobj->handle;
253
254     dsi->msg_request = 1;
255     if (dsi_attention(AFPobj->handle, AFPATTN_MESG | AFPATTN_TIME(5)) < 0)
256         dsi->msg_request = 2;
257 }
258
259 static void alarm_handler(int sig _U_)
260 {
261     int err;
262     DSI *dsi = (DSI *)AFPobj->handle;
263
264     /* we have to restart the timer because some libraries may use alarm() */
265     setitimer(ITIMER_REAL, &dsi->timer, NULL);
266
267     /* we got some traffic from the client since the previous timer tick. */
268     if ((dsi->flags & DSI_DATA)) {
269         dsi->flags &= ~DSI_DATA;
270         return;
271     }
272
273     dsi->tickle++;
274     LOG(log_maxdebug, logtype_afpd, "alarm: tickles: %u, flags: %s|%s|%s|%s|%s|%s|%s|%s|%s",
275         dsi->tickle,
276         (dsi->flags & DSI_DATA) ?         "DSI_DATA" : "-",
277         (dsi->flags & DSI_RUNNING) ?      "DSI_RUNNING" : "-",
278         (dsi->flags & DSI_SLEEPING) ?     "DSI_SLEEPING" : "-",
279         (dsi->flags & DSI_EXTSLEEP) ?     "DSI_EXTSLEEP" : "-",
280         (dsi->flags & DSI_DISCONNECTED) ? "DSI_DISCONNECTED" : "-",
281         (dsi->flags & DSI_DIE) ?          "DSI_DIE" : "-",
282         (dsi->flags & DSI_NOREPLY) ?      "DSI_NOREPLY" : "-",
283         (dsi->flags & DSI_RECONSOCKET) ?  "DSI_RECONSOCKET" : "-",
284         (dsi->flags & DSI_RECONINPROG) ?  "DSI_RECONINPROG" : "-");
285
286     if (dsi->flags & DSI_SLEEPING) {
287         if (dsi->tickle > AFPobj->options.sleep) {
288             LOG(log_note, logtype_afpd, "afp_alarm: sleep time ended");
289             afp_dsi_die(EXITERR_CLNT);
290         }
291         return;
292     } 
293
294     if (dsi->flags & DSI_DISCONNECTED) {
295         if (dsi->tickle > AFPobj->options.disconnected) {
296             LOG(log_error, logtype_afpd, "afp_alarm: reconnect timer expired, goodbye");
297             afp_dsi_die(EXITERR_CLNT);
298         }
299         return;
300     }
301
302     /* if we're in the midst of processing something, don't die. */        
303     if ( !(dsi->flags & DSI_RUNNING) && (dsi->tickle >= AFPobj->options.timeout)) {
304         LOG(log_error, logtype_afpd, "afp_alarm: child timed out, entering disconnected state");
305         dsi->proto_close(dsi);
306         dsi->flags |= DSI_DISCONNECTED;
307         return;
308     }
309
310     if ((err = pollvoltime(AFPobj)) == 0)
311         LOG(log_debug, logtype_afpd, "afp_alarm: sending DSI tickle");
312         err = dsi_tickle(AFPobj->handle);
313     if (err <= 0) {
314         LOG(log_error, logtype_afpd, "afp_alarm: connection problem, entering disconnected state");
315         dsi->proto_close(dsi);
316         dsi->flags |= DSI_DISCONNECTED;
317     }
318 }
319
320 /* ----------------- 
321    if dsi->in_write is set attention, tickle (and close?) msg
322    aren't sent. We don't care about tickle 
323 */
324 static void pending_request(DSI *dsi)
325 {
326     /* send pending attention */
327
328     /* read msg if any, it could be done in afp_getsrvrmesg */
329     if (dsi->msg_request) {
330         if (dsi->msg_request == 2) {
331             /* didn't send it in signal handler */
332             dsi_attention(AFPobj->handle, AFPATTN_MESG | AFPATTN_TIME(5));
333         }
334         dsi->msg_request = 0;
335         readmessage(AFPobj);
336     }
337     if (dsi->down_request) {
338         dsi->down_request = 0;
339         dsi_attention(AFPobj->handle, AFPATTN_SHUTDOWN | AFPATTN_NORECONNECT |
340                   AFPATTN_MESG | AFPATTN_TIME(5));
341     }
342 }
343
344 /* -------------------------------------------
345  afp over dsi. this never returns. 
346 */
347 void afp_over_dsi(AFPObj *obj)
348 {
349     DSI *dsi = (DSI *) obj->handle;
350     int rc_idx;
351     u_int32_t err, cmd;
352     u_int8_t function;
353     struct sigaction action;
354
355     AFPobj = obj;
356     obj->exit = afp_dsi_die;
357     obj->reply = (int (*)()) dsi_cmdreply;
358     obj->attention = (int (*)(void *, AFPUserBytes)) dsi_attention;
359     dsi->tickle = 0;
360
361     memset(&action, 0, sizeof(action));
362
363     /* install SIGHUP */
364     action.sa_handler = afp_dsi_reload;
365     sigemptyset( &action.sa_mask );
366     sigaddset(&action.sa_mask, SIGALRM);
367     sigaddset(&action.sa_mask, SIGTERM);
368     sigaddset(&action.sa_mask, SIGUSR1);
369     sigaddset(&action.sa_mask, SIGINT);
370     sigaddset(&action.sa_mask, SIGUSR2);
371     action.sa_flags = SA_RESTART;
372     if ( sigaction( SIGHUP, &action, NULL ) < 0 ) {
373         LOG(log_error, logtype_afpd, "afp_over_dsi: sigaction: %s", strerror(errno) );
374         afp_dsi_die(EXITERR_SYS);
375     }
376
377     /* install SIGURG */
378     action.sa_handler = afp_dsi_transfer_session;
379     sigemptyset( &action.sa_mask );
380     sigaddset(&action.sa_mask, SIGALRM);
381     sigaddset(&action.sa_mask, SIGTERM);
382     sigaddset(&action.sa_mask, SIGUSR1);
383     sigaddset(&action.sa_mask, SIGINT);
384     sigaddset(&action.sa_mask, SIGUSR2);
385     action.sa_flags = SA_RESTART;
386     if ( sigaction( SIGURG, &action, NULL ) < 0 ) {
387         LOG(log_error, logtype_afpd, "afp_over_dsi: sigaction: %s", strerror(errno) );
388         afp_dsi_die(EXITERR_SYS);
389     }
390
391     /* install SIGTERM */
392     action.sa_handler = afp_dsi_die;
393     sigemptyset( &action.sa_mask );
394     sigaddset(&action.sa_mask, SIGALRM);
395     sigaddset(&action.sa_mask, SIGHUP);
396     sigaddset(&action.sa_mask, SIGUSR1);
397     sigaddset(&action.sa_mask, SIGINT);
398     sigaddset(&action.sa_mask, SIGUSR2);
399     action.sa_flags = SA_RESTART;
400     if ( sigaction( SIGTERM, &action, NULL ) < 0 ) {
401         LOG(log_error, logtype_afpd, "afp_over_dsi: sigaction: %s", strerror(errno) );
402         afp_dsi_die(EXITERR_SYS);
403     }
404
405     /* Added for server message support */
406     action.sa_handler = afp_dsi_getmesg;
407     sigemptyset( &action.sa_mask );
408     sigaddset(&action.sa_mask, SIGALRM);
409     sigaddset(&action.sa_mask, SIGTERM);
410     sigaddset(&action.sa_mask, SIGUSR1);
411     sigaddset(&action.sa_mask, SIGHUP);
412     sigaddset(&action.sa_mask, SIGINT);
413     action.sa_flags = SA_RESTART;
414     if ( sigaction( SIGUSR2, &action, NULL) < 0 ) {
415         LOG(log_error, logtype_afpd, "afp_over_dsi: sigaction: %s", strerror(errno) );
416         afp_dsi_die(EXITERR_SYS);
417     }
418
419     /*  SIGUSR1 - set down in 5 minutes  */
420     action.sa_handler = afp_dsi_timedown;
421     sigemptyset( &action.sa_mask );
422     sigaddset(&action.sa_mask, SIGALRM);
423     sigaddset(&action.sa_mask, SIGHUP);
424     sigaddset(&action.sa_mask, SIGTERM);
425     sigaddset(&action.sa_mask, SIGINT);
426     sigaddset(&action.sa_mask, SIGUSR2);
427     action.sa_flags = SA_RESTART;
428     if ( sigaction( SIGUSR1, &action, NULL) < 0 ) {
429         LOG(log_error, logtype_afpd, "afp_over_dsi: sigaction: %s", strerror(errno) );
430         afp_dsi_die(EXITERR_SYS);
431     }
432
433     /*  SIGINT - enable max_debug LOGging to /tmp/afpd.PID.XXXXXX */
434     action.sa_handler = afp_dsi_debug;
435     sigfillset( &action.sa_mask );
436     action.sa_flags = SA_RESTART;
437     if ( sigaction( SIGINT, &action, NULL) < 0 ) {
438         LOG(log_error, logtype_afpd, "afp_over_dsi: sigaction: %s", strerror(errno) );
439         afp_dsi_die(EXITERR_SYS);
440     }
441
442 #ifndef DEBUGGING
443     /* tickle handler */
444     action.sa_handler = alarm_handler;
445     sigemptyset(&action.sa_mask);
446     sigaddset(&action.sa_mask, SIGHUP);
447     sigaddset(&action.sa_mask, SIGTERM);
448     sigaddset(&action.sa_mask, SIGUSR1);
449     sigaddset(&action.sa_mask, SIGINT);
450     sigaddset(&action.sa_mask, SIGUSR2);
451     action.sa_flags = SA_RESTART;
452     if ((sigaction(SIGALRM, &action, NULL) < 0) ||
453             (setitimer(ITIMER_REAL, &dsi->timer, NULL) < 0)) {
454         afp_dsi_die(EXITERR_SYS);
455     }
456 #endif /* DEBUGGING */
457
458     if (dircache_init(obj->options.dircachesize) != 0)
459         afp_dsi_die(EXITERR_SYS);
460
461     /* set TCP snd/rcv buf */
462     if (obj->options.tcp_rcvbuf) {
463         if (setsockopt(dsi->socket,
464                        SOL_SOCKET,
465                        SO_RCVBUF,
466                        &obj->options.tcp_rcvbuf,
467                        sizeof(obj->options.tcp_rcvbuf)) != 0) {
468             LOG(log_error, logtype_dsi, "afp_over_dsi: setsockopt(SO_RCVBUF): %s", strerror(errno));
469         }
470     }
471     if (obj->options.tcp_sndbuf) {
472         if (setsockopt(dsi->socket,
473                        SOL_SOCKET,
474                        SO_SNDBUF,
475                        &obj->options.tcp_sndbuf,
476                        sizeof(obj->options.tcp_sndbuf)) != 0) {
477             LOG(log_error, logtype_dsi, "afp_over_dsi: setsockopt(SO_SNDBUF): %s", strerror(errno));
478         }
479     }
480
481     /* get stuck here until the end */
482     while (1) {
483         if (sigsetjmp(recon_jmp, 1) != 0)
484             /* returning from SIGALARM handler for a primary reconnect */
485             continue;
486
487         /* Blocking read on the network socket */
488         cmd = dsi_receive(dsi);
489
490         if (cmd == 0) {
491             /* cmd == 0 is the error condition */
492             if (dsi->flags & DSI_RECONSOCKET) {
493                 /* we just got a reconnect so we immediately try again to receive on the new fd */
494                 dsi->flags &= ~DSI_RECONSOCKET;
495                 continue;
496             }
497             /* Some error on the client connection, enter disconnected state */
498             dsi->flags |= DSI_DISCONNECTED;
499
500             /* the client sometimes logs out (afp_logout) but doesn't close the DSI session */
501             if (dsi->flags & DSI_AFP_LOGGED_OUT) {
502                 afp_dsi_close(obj);
503                 exit(0);
504             }
505
506             pause(); /* gets interrupted by SIGALARM or SIGURG tickle */
507             continue; /* continue receiving until disconnect timer expires
508                        * or a primary reconnect succeeds  */
509         }
510
511         if (!(dsi->flags & DSI_EXTSLEEP) && (dsi->flags & DSI_SLEEPING)) {
512             LOG(log_debug, logtype_afpd, "afp_over_dsi: got data, ending normal sleep");
513             dsi->flags &= ~DSI_SLEEPING;
514             dsi->tickle = 0;
515         }
516
517         if (reload_request) {
518             reload_request = 0;
519             load_volumes(AFPobj);
520         }
521
522         /* The first SIGINT enables debugging, the next restores the config */
523         if (debug_request) {
524             static int debugging = 0;
525             debug_request = 0;
526
527             dircache_dump();
528             uuidcache_dump();
529
530             if (debugging) {
531                 if (obj->options.logconfig)
532                     setuplog(obj->options.logconfig);
533                 else
534                     setuplog("default log_note");
535                 debugging = 0;
536             } else {
537                 char logstr[50];
538                 debugging = 1;
539                 sprintf(logstr, "default log_maxdebug /tmp/afpd.%u.XXXXXX", getpid());
540                 setuplog(logstr);
541             }
542         }
543
544
545         dsi->flags |= DSI_DATA;
546         dsi->tickle = 0;
547
548         switch(cmd) {
549
550         case DSIFUNC_CLOSE:
551             LOG(log_debug, logtype_afpd, "DSI: close session request");
552             afp_dsi_close(obj);
553             LOG(log_note, logtype_afpd, "done");
554             exit(0);
555
556         case DSIFUNC_TICKLE:
557             dsi->flags &= ~DSI_DATA; /* thats no data in the sense we use it in alarm_handler */
558             LOG(log_debug, logtype_afpd, "DSI: client tickle");
559             /* timer is not every 30 seconds anymore, so we don't get killed on the client side. */
560             if ((dsi->flags & DSI_DIE))
561                 dsi_tickle(dsi);
562             break;
563
564         case DSIFUNC_CMD:
565 #ifdef AFS
566             if ( writtenfork ) {
567                 if ( flushfork( writtenfork ) < 0 ) {
568                     LOG(log_error, logtype_afpd, "main flushfork: %s", strerror(errno) );
569                 }
570                 writtenfork = NULL;
571             }
572 #endif /* AFS */
573
574             function = (u_char) dsi->commands[0];
575
576             /* AFP replay cache */
577             rc_idx = dsi->clientID % REPLAYCACHE_SIZE;
578             LOG(log_debug, logtype_afpd, "DSI request ID: %u", dsi->clientID);
579
580             if (replaycache[rc_idx].DSIreqID == dsi->clientID
581                 && replaycache[rc_idx].AFPcommand == function) {
582                 LOG(log_note, logtype_afpd, "AFP Replay Cache match: id: %u / cmd: %s",
583                     dsi->clientID, AfpNum2name(function));
584                 err = replaycache[rc_idx].result;
585             /* AFP replay cache end */
586             } else {
587                 /* send off an afp command. in a couple cases, we take advantage
588                  * of the fact that we're a stream-based protocol. */
589                 if (afp_switch[function]) {
590                     dsi->datalen = DSI_DATASIZ;
591                     dsi->flags |= DSI_RUNNING;
592
593                     LOG(log_debug, logtype_afpd, "<== Start AFP command: %s", AfpNum2name(function));
594
595                     err = (*afp_switch[function])(obj,
596                                                   (char *)&dsi->commands, dsi->cmdlen,
597                                                   (char *)&dsi->data, &dsi->datalen);
598
599                     LOG(log_debug, logtype_afpd, "==> Finished AFP command: %s -> %s",
600                         AfpNum2name(function), AfpErr2name(err));
601
602                     dir_free_invalid_q();
603
604 #ifdef FORCE_UIDGID
605                     /* bring everything back to old euid, egid */
606                     if (obj->force_uid)
607                         restore_uidgid ( &obj->uidgid );
608 #endif /* FORCE_UIDGID */
609                     dsi->flags &= ~DSI_RUNNING;
610
611                     /* Add result to the AFP replay cache */
612                     replaycache[rc_idx].DSIreqID = dsi->clientID;
613                     replaycache[rc_idx].AFPcommand = function;
614                     replaycache[rc_idx].result = err;
615                 } else {
616                     LOG(log_error, logtype_afpd, "bad function %X", function);
617                     dsi->datalen = 0;
618                     err = AFPERR_NOOP;
619                 }
620             }
621
622             /* single shot toggle that gets set by dsi_readinit. */
623             if (dsi->flags & DSI_NOREPLY) {
624                 dsi->flags &= ~DSI_NOREPLY;
625                 break;
626             }
627
628             if (!dsi_cmdreply(dsi, err)) {
629                 LOG(log_error, logtype_afpd, "dsi_cmdreply(%d): %s", dsi->socket, strerror(errno) );
630                 dsi->flags |= DSI_DISCONNECTED;
631             }
632             break;
633
634         case DSIFUNC_WRITE: /* FPWrite and FPAddIcon */
635             function = (u_char) dsi->commands[0];
636             if ( afp_switch[ function ] != NULL ) {
637                 dsi->datalen = DSI_DATASIZ;
638                 dsi->flags |= DSI_RUNNING;
639
640                 LOG(log_debug, logtype_afpd, "<== Start AFP command: %s", AfpNum2name(function));
641
642                 err = (*afp_switch[function])(obj,
643                                               (char *)&dsi->commands, dsi->cmdlen,
644                                               (char *)&dsi->data, &dsi->datalen);
645
646                 LOG(log_debug, logtype_afpd, "==> Finished AFP command: %s -> %s",
647                     AfpNum2name(function), AfpErr2name(err));
648
649                 dsi->flags &= ~DSI_RUNNING;
650 #ifdef FORCE_UIDGID
651                 /* bring everything back to old euid, egid */
652                 if (obj->force_uid)
653                     restore_uidgid ( &obj->uidgid );
654 #endif /* FORCE_UIDGID */
655             } else {
656                 LOG(log_error, logtype_afpd, "(write) bad function %x", function);
657                 dsi->datalen = 0;
658                 err = AFPERR_NOOP;
659             }
660
661             if (!dsi_wrtreply(dsi, err)) {
662                 LOG(log_error, logtype_afpd, "dsi_wrtreply: %s", strerror(errno) );
663                 dsi->flags |= DSI_DISCONNECTED;
664             }
665             break;
666
667         case DSIFUNC_ATTN: /* attention replies */
668             break;
669
670             /* error. this usually implies a mismatch of some kind
671              * between server and client. if things are correct,
672              * we need to flush the rest of the packet if necessary. */
673         default:
674             LOG(log_info, logtype_afpd,"afp_dsi: spurious command %d", cmd);
675             dsi_writeinit(dsi, dsi->data, DSI_DATASIZ);
676             dsi_writeflush(dsi);
677             break;
678         }
679         pending_request(dsi);
680     }
681
682     /* error */
683     afp_dsi_die(EXITERR_CLNT);
684 }