]> arthur.barton.de Git - netatalk.git/blob - etc/afpd/status.c
Whitespace and exclamation mark fixes
[netatalk.git] / etc / afpd / status.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 <stdbool.h>
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <unistd.h>
14 #include <ctype.h>
15 #include <string.h>
16 #include <time.h>
17 #include <sys/types.h>
18 #include <sys/stat.h>
19 #include <fcntl.h>
20 #include <sys/socket.h>
21 #include <atalk/logger.h>
22
23 #if 0
24 #ifdef BSD4_4
25 #include <sys/param.h>
26 #ifndef HAVE_GETHOSTID
27 #include <sys/sysctl.h>
28 #endif /* HAVE_GETHOSTID */
29 #endif /* BSD4_4 */
30 #endif
31
32 #include <arpa/inet.h>
33
34 #ifdef HAVE_KERBEROS
35 #ifdef HAVE_KRB5_KRB5_H
36 #include <krb5/krb5.h>
37 #else
38 #include <krb5.h>
39 #endif /* HAVE_KRB5_KRB5_H */
40 #endif /* HAVE_KERBEROS */
41
42 #include <atalk/dsi.h>
43 #include <atalk/unicode.h>
44 #include <atalk/util.h>
45 #include <atalk/globals.h>
46
47 #include "status.h"
48 #include "afp_config.h"
49 #include "icon.h"
50 #include "uam_auth.h"
51
52 static   size_t maxstatuslen = 0;
53
54 static int uam_gss_enabled()
55 {
56     /* XXX: must be a better way to find out if uam_gss is active */
57     return auth_uamfind(UAM_SERVER_LOGIN_EXT,
58                         "Client Krb v2",
59                         sizeof("Client Krb v2")) != NULL;
60 }
61
62 static void status_flags(char *data,
63                          const int notif,
64                          const int ipok,
65                          const unsigned char passwdbits,
66                          const int dirsrvcs,
67                          int flags)
68 {
69     uint16_t           status;
70
71     status = AFPSRVRINFO_COPY
72            | AFPSRVRINFO_SRVSIGNATURE
73            | AFPSRVRINFO_SRVMSGS
74            | AFPSRVRINFO_FASTBOZO
75            | AFPSRVRINFO_SRVUTF8
76            | AFPSRVRINFO_EXTSLEEP;
77
78     if (passwdbits & PASSWD_SET) /* some uams may not allow this. */
79         status |= AFPSRVRINFO_PASSWD;
80     if (passwdbits & PASSWD_NOSAVE)
81         status |= AFPSRVRINFO_NOSAVEPASSWD;
82     if (ipok) /* only advertise tcp/ip if we have a valid address */
83         status |= AFPSRVRINFO_TCPIP;
84     if (notif) /* Default is yes */
85         status |= AFPSRVRINFO_SRVNOTIFY;
86     if (dirsrvcs)
87         status |= AFPSRVRINFO_SRVRDIR;
88     if (flags & OPTION_UUID)
89         status |= AFPSRVRINFO_UUID;
90
91     status = htons(status);
92     memcpy(data + AFPSTATUS_FLAGOFF, &status, sizeof(status));
93 }
94
95 static int status_server(char *data, const char *server, const struct afp_options *options)
96 {
97     char                *start = data;
98     char                *Obj;
99     char                buf[32];
100     uint16_t           status;
101     size_t              len;
102
103     /* make room for all offsets before server name */
104     data += AFPSTATUS_PRELEN;
105
106     /* extract the obj part of the server */
107     Obj = (char *) server;
108     if ((size_t)-1 == (len = convert_string( 
109                            options->unixcharset, options->maccharset, 
110                            Obj, -1, buf, sizeof(buf))) ) {
111         len = MIN(strlen(Obj), 31);
112         *data++ = len;
113         memcpy( data, Obj, len );
114         LOG ( log_error, logtype_afpd, "Could not set servername, using fallback");
115     } else {
116         *data++ = len;
117         memcpy( data, buf, len );
118     }
119     if ((len + 1) & 1) /* pad server name and length byte to even boundary */
120         len++;
121     data += len;
122
123     /* make room for signature and net address offset. save location of
124      * signature offset. we're also making room for directory names offset
125      * and the utf-8 server name offset.
126      *
127      * NOTE: technically, we don't need to reserve space for the
128      * signature and net address offsets if they're not going to be
129      * used. as there are no offsets after them, it doesn't hurt to
130      * have them specified though. so, we just do that to simplify
131      * things.  
132      *
133      * NOTE2: AFP3.1 Documentation states that the directory names offset
134      * is a required feature, even though it can be set to zero.
135      */
136     len = data - start;
137     status = htons(len + AFPSTATUS_POSTLEN);
138     memcpy(start + AFPSTATUS_MACHOFF, &status, sizeof(status));
139     return len; /* return the offset to beginning of signature offset */
140 }
141
142 static void status_machine(char *data)
143 {
144     char                *start = data;
145     uint16_t           status;
146     int                 len;
147 #ifdef AFS
148     const char          *machine = "afs";
149 #else /* !AFS */
150     const char          *machine = "Netatalk%s";
151 #endif /* AFS */
152     char buf[AFPSTATUS_MACHLEN+1];
153
154     memcpy(&status, start + AFPSTATUS_MACHOFF, sizeof(status));
155     data += ntohs( status );
156
157     if ((strlen(machine) + strlen(VERSION)) <= AFPSTATUS_MACHLEN) {
158         len = snprintf(buf, AFPSTATUS_MACHLEN + 1, machine, VERSION);
159     } else {
160         if (strlen(VERSION) > AFPSTATUS_MACHLEN) {
161             len = snprintf(buf, AFPSTATUS_MACHLEN + 1, VERSION);
162         } else {
163             (void)snprintf(buf, AFPSTATUS_MACHLEN + 1, machine, "");
164             (void)snprintf(buf + AFPSTATUS_MACHLEN - strlen(VERSION),
165                            strlen(VERSION) + 1,
166                            VERSION);
167             len = AFPSTATUS_MACHLEN;
168         }
169     }
170
171     *data++ = len;
172     memcpy( data, buf, len );
173     data += len;
174
175     status = htons(data - start);
176     memcpy(start + AFPSTATUS_VERSOFF, &status, sizeof(status));
177 }
178
179 /* server signature is a 16-byte quantity */
180 static uint16_t status_signature(char *data, int *servoffset,
181                                   const struct afp_options *options)
182 {
183     char                 *status;
184     uint16_t            offset, sigoff;
185
186     status = data;
187
188     /* get server signature offset */
189     memcpy(&offset, data + *servoffset, sizeof(offset));
190     sigoff = offset = ntohs(offset);
191
192     /* jump to server signature offset */
193     data += offset;
194
195     memset(data, 0, 16);
196     memcpy(data, options->signature, 16);
197     data += 16;
198
199     /* calculate net address offset */
200     *servoffset += sizeof(offset);
201     offset = htons(data - status);
202     memcpy(status + *servoffset, &offset, sizeof(offset));
203     return sigoff;
204 }
205
206 static size_t status_netaddress(char *data, int *servoffset,
207                              const DSI *dsi,
208                              const struct afp_options *options)
209 {
210     char               *begin;
211     uint16_t          offset;
212     size_t             addresses_len = 0;
213
214     begin = data;
215
216     /* get net address offset */
217     memcpy(&offset, data + *servoffset, sizeof(offset));
218     data += ntohs(offset);
219
220     /* format:
221        Address count (byte) 
222        len (byte = sizeof(length + address type + address) 
223        address type (byte, ip address = 0x01, ip + port = 0x02,
224                            ddp address = 0x03, fqdn = 0x04) 
225        address (up to 254 bytes, ip = 4, ip + port = 6, ddp = 4)
226        */
227
228     /* number of addresses. this currently screws up if we have a dsi
229        connection, but we don't have the ip address. to get around this,
230        we turn off the status flag for tcp/ip. */
231     *data++ = ((options->fqdn && dsi)? 1 : 0) + (dsi ? 1 : 0) +
232               (((options->flags & OPTION_ANNOUNCESSH) && options->fqdn && dsi)? 1 : 0);
233
234     /* ip address */
235     if (dsi) {
236         if (dsi->server.ss_family == AF_INET) { /* IPv4 */
237             const struct sockaddr_in *inaddr = (struct sockaddr_in *)&dsi->server;
238             if (inaddr->sin_port == htons(DSI_AFPOVERTCP_PORT)) {
239                 *data++ = 6; /* length */
240                 *data++ = 0x01; /* basic ip address */
241                 memcpy(data, &inaddr->sin_addr.s_addr,
242                        sizeof(inaddr->sin_addr.s_addr));
243                 data += sizeof(inaddr->sin_addr.s_addr);
244                 addresses_len += 7;
245             } else {
246                 /* ip address + port */
247                 *data++ = 8;
248                 *data++ = 0x02; /* ip address with port */
249                 memcpy(data, &inaddr->sin_addr.s_addr,
250                        sizeof(inaddr->sin_addr.s_addr));
251                 data += sizeof(inaddr->sin_addr.s_addr);
252                 memcpy(data, &inaddr->sin_port, sizeof(inaddr->sin_port));
253                 data += sizeof(inaddr->sin_port);
254                 addresses_len += 9;
255             }
256         } else { /* IPv6 */
257             const struct sockaddr_in6 *sa6 = (struct sockaddr_in6 *)&dsi->server;
258             if (sa6->sin6_port == htons(DSI_AFPOVERTCP_PORT)) {
259                 *data++ = 18; /* length */
260                 *data++ = 6; /* type */
261                 memcpy(data, &sa6->sin6_addr.s6_addr, sizeof(sa6->sin6_addr.s6_addr));
262                 data += sizeof(sa6->sin6_addr.s6_addr);
263                 addresses_len += 19;
264             } else {
265                 /* ip address + port */
266                 *data++ = 20; /* length */
267                 *data++ = 7; /* type*/
268                 memcpy(data, &sa6->sin6_addr.s6_addr, sizeof(sa6->sin6_addr.s6_addr));
269                 data += sizeof(sa6->sin6_addr.s6_addr);
270                 memcpy(data, &sa6->sin6_port, sizeof(sa6->sin6_port));
271                 data += sizeof(sa6->sin6_port);
272                 addresses_len += 21;
273             }
274
275         }
276     }
277
278     /* handle DNS names */
279     if (options->fqdn && dsi) {
280         size_t len = strlen(options->fqdn);
281         if ( len + 2 + addresses_len < maxstatuslen - offset) {
282             *data++ = len +2;
283             *data++ = 0x04;
284             memcpy(data, options->fqdn, len);
285             data += len;
286             addresses_len += len+2;
287         }
288
289         /* Annouce support for SSH tunneled AFP session, 
290          * this feature is available since 10.3.2.
291          * According to the specs (AFP 3.1 p.225) this should
292          * be an IP+Port style value, but it only works with 
293          * a FQDN. OSX Server uses FQDN as well.
294          */
295         if ( len + 2 + addresses_len < maxstatuslen - offset) {
296             if (options->flags & OPTION_ANNOUNCESSH) {
297                 *data++ = len +2;
298                 *data++ = 0x05;
299                 memcpy(data, options->fqdn, len);
300                 data += len;
301             }
302         }
303     }
304
305     /* calculate/store Directory Services Names offset */
306     offset = htons(data - begin); 
307     *servoffset += sizeof(offset);
308     memcpy(begin + *servoffset, &offset, sizeof(offset));
309
310     /* return length of buffer */
311     return (data - begin);
312 }
313
314
315 /**
316  * DirectoryNamesCount offset: uint16_t
317  * ...
318  * DirectoryNamesCount: uint8_t
319  * DirectoryNames: list of UTF-8 Pascal strings (uint8_t + char[1,255])
320  */
321 static size_t status_directorynames(char *data,
322                                     int *diroffset,
323                                     const DSI *dsi _U_,
324                                     const struct afp_options *options)
325 {
326     char *begin = data;
327     uint16_t offset;
328
329     memcpy(&offset, begin + *diroffset, sizeof(offset));
330     offset = ntohs(offset);
331     data += offset;
332
333     if (!uam_gss_enabled() || !options->k5principal) {
334         *data = 0;
335         data++;
336     } else {
337         memcpy(data, options->k5principal, options->k5principal_buflen);
338         data += options->k5principal_buflen;
339     }
340
341     /* Calculate and store offset for UTF8ServerName */
342     *diroffset += sizeof(uint16_t);
343     offset = htons(data - begin);
344     memcpy(begin + *diroffset, &offset, sizeof(uint16_t));
345
346     /* return length of buffer */
347     return (data - begin);
348 }
349
350 static size_t status_utf8servername(char *data, int *nameoffset,
351                                  const DSI *dsi _U_,
352                                  const struct afp_options *options)
353 {
354     uint16_t namelen;
355     size_t len;
356     char *begin = data;
357     uint16_t offset;
358
359     memcpy(&offset, data + *nameoffset, sizeof(offset));
360     offset = ntohs(offset);
361     data += offset;
362
363     LOG(log_info, logtype_afpd, "servername: %s", options->hostname);
364
365     if ((len = convert_string(options->unixcharset,
366                               CH_UTF8_MAC, 
367                               options->hostname,
368                               -1,
369                               data + sizeof(namelen),
370                               maxstatuslen-offset)) == (size_t)-1) {
371         LOG(log_error, logtype_afpd, "Could not set utf8 servername");
372
373         /* set offset to 0 */
374         memset(begin + *nameoffset, 0, sizeof(offset));
375         data = begin + offset;
376     } else {
377         namelen = htons(len);
378         memcpy( data, &namelen, sizeof(namelen));
379         data += sizeof(namelen);
380         data += len;
381         offset = htons(offset);
382         memcpy(begin + *nameoffset, &offset, sizeof(uint16_t));
383     }
384
385     /* return length of buffer */
386     return (data - begin);
387
388 }
389
390 /* returns actual offset to signature */
391 static void status_icon(char *data, const unsigned char *icondata,
392                         const int iconlen, const int sigoffset)
393 {
394     char                *start = data;
395     char                *sigdata = data + sigoffset;
396     uint16_t            ret, status;
397
398     memcpy(&status, start + AFPSTATUS_ICONOFF, sizeof(status));
399     if ( icondata == NULL ) {
400         ret = status;
401         memset(start + AFPSTATUS_ICONOFF, 0, sizeof(status));
402     } else {
403         data += ntohs( status );
404         memcpy( data, icondata, iconlen);
405         data += iconlen;
406         ret = htons(data - start);
407     }
408
409     /* put in signature offset */
410     if (sigoffset)
411         memcpy(sigdata, &ret, sizeof(ret));
412 }
413
414 /* ---------------------
415  */
416 void status_init(AFPObj *obj, DSI *dsi)
417 {
418     char *status = dsi->status;
419     size_t statuslen;
420     int c, sigoff, ipok = 0;
421     const struct afp_options *options = &obj->options;
422
423     maxstatuslen = sizeof(dsi->status);
424
425     if (dsi->server.ss_family == AF_INET) { /* IPv4 */
426         const struct sockaddr_in *sa4 = (struct sockaddr_in *)&dsi->server;
427         ipok = sa4->sin_addr.s_addr ? 1 : 0;
428     } else { /* IPv6 */
429         const struct sockaddr_in6 *sa6 = (struct sockaddr_in6 *)&dsi->server;
430         for (int i=0; i<16; i++) {
431             if (sa6->sin6_addr.s6_addr[i]) {
432                 ipok = 1;
433                 break;
434             }
435         }
436     }
437
438     /*
439      * These routines must be called in order -- earlier calls
440      * set the offsets for later calls.
441      *
442      * using structs is a bad idea, but that's what the original code
443      * does. solaris, in fact, will segfault with them. so, now
444      * we just use the powers of #defines and memcpy.
445      *
446      * reply block layout (offsets are 16-bit quantities):
447      * machine type offset -> AFP version count offset ->
448      * UAM count offset -> vol icon/mask offset -> flags ->
449      *
450      * server name [padded to even boundary] -> signature offset ->
451      * network address offset ->
452      *
453      * at the appropriate offsets:
454      * machine type, afp versions, uams, server signature
455      * (16-bytes), network addresses, volume icon/mask 
456      */
457
458     status_flags(status,
459                  options->flags & OPTION_SERVERNOTIF,
460                  (options->fqdn || ipok),
461                  options->passwdbits, 
462                  1,
463                  options->flags);
464     /* returns offset to signature offset */
465     c = status_server(status, options->hostname, options);
466     status_machine(status);
467     status_versions(status, dsi);
468     status_uams(status, options->uamlist);
469     status_icon(status, icon, sizeof(icon), c);
470
471     sigoff = status_signature(status, &c, options);
472     /* c now contains the offset where the netaddress offset lives */
473
474     status_netaddress(status, &c, dsi, options);
475     /* c now contains the offset where the Directory Names Count offset lives */
476
477     statuslen = status_directorynames(status, &c, dsi, options);
478     /* c now contains the offset where the UTF-8 ServerName offset lives */
479
480     if ( statuslen < maxstatuslen) 
481         statuslen = status_utf8servername(status, &c, dsi, options);
482
483     dsi->signature = status + sigoff;
484     dsi->statuslen = statuslen;
485 }
486
487 /* set_signature()                                                    */
488 /*                                                                    */
489 /* If found in conf file, use it.                                     */
490 /* If not found in conf file, genarate and append in conf file.       */
491 /* If conf file don't exist, create and genarate.                     */
492 /* If cannot open conf file, use one-time signature.                  */
493 /* If signature = xxxxx, use it.                                      */
494
495 void set_signature(struct afp_options *options) {
496     int fd, i;
497     struct stat tmpstat;
498     char *servername_conf;
499     int header = 0;
500     char buf[1024], *p;
501     FILE *fp = NULL;
502     size_t len;
503     char *server_tmp;
504     
505     server_tmp = options->hostname;
506     len = strlen(options->signatureopt);
507     if (len == 0) {
508         goto server_signature_auto;   /* default */
509     } else if (len < 3) {
510         LOG(log_warning, logtype_afpd, "WARNING: signature string %s is very short!", options->signatureopt);
511         goto server_signature_user;
512     } else if (len > 16) {
513         LOG(log_warning, logtype_afpd, "WARNING: signature string %s is very long!", options->signatureopt);
514         len = 16;
515         goto server_signature_user;
516     } else {
517         LOG(log_info, logtype_afpd, "signature string is %s.", options->signatureopt);
518         goto server_signature_user;
519     }
520     
521 server_signature_user:
522     
523     /* Signature is defined in afp.conf */
524     memset(options->signature, 0, 16);
525     memcpy(options->signature, options->signatureopt, len);
526     goto server_signature_done;
527     
528 server_signature_auto:
529     
530     /* Signature type is auto, using afp_signature.conf */
531     if (!stat(options->sigconffile, &tmpstat)) {                /* conf file exists? */
532         if ((fp = fopen(options->sigconffile, "r")) != NULL) {  /* read open? */
533             /* scan in the conf file */
534             while (fgets(buf, sizeof(buf), fp) != NULL) { 
535                 p = buf;
536                 while (p && isblank(*p))
537                     p++;
538                 if (!p || (*p == '#') || (*p == '\n'))
539                     continue;                             /* invalid line */
540                 if (*p == '"') {
541                     p++;
542                     if ((servername_conf = strtok( p, "\"" )) == NULL)
543                         continue;                         /* syntax error: invalid quoted servername */
544                 } else {
545                     if ((servername_conf = strtok( p, " \t" )) == NULL)
546                         continue;                         /* syntax error: invalid servername */
547                 }
548                 p = strchr(p, '\0');
549                 p++;
550                 if (*p == '\0')
551                     continue;                             /* syntax error: missing signature */
552                 
553                 if (strcmp(server_tmp, servername_conf))
554                     continue;                             /* another servername */
555                 
556                 while (p && isblank(*p))
557                     p++;
558                 if ( 16 == sscanf(p, "%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX%2hhX",
559                                   &options->signature[ 0], &options->signature[ 1],
560                                   &options->signature[ 2], &options->signature[ 3],
561                                   &options->signature[ 4], &options->signature[ 5],
562                                   &options->signature[ 6], &options->signature[ 7],
563                                   &options->signature[ 8], &options->signature[ 9],
564                                   &options->signature[10], &options->signature[11],
565                                   &options->signature[12], &options->signature[13],
566                                   &options->signature[14], &options->signature[15]
567                          )) {
568                     fclose(fp);
569                     goto server_signature_done;                 /* found in conf file */
570                 }
571             }
572             if ((fp = freopen(options->sigconffile, "a+", fp)) != NULL) { /* append because not found */
573                 fseek(fp, 0L, SEEK_END);
574                 if(ftell(fp) == 0) {                     /* size = 0 */
575                     header = 1;
576                     goto server_signature_random;
577                 } else {
578                     fseek(fp, -1L, SEEK_END);
579                     if(fgetc(fp) != '\n') fputc('\n', fp); /* last char is \n? */
580                     goto server_signature_random;
581                 }                    
582             } else {
583                 LOG(log_error, logtype_afpd, "ERROR: Cannot write in %s (%s). Using one-time signature.",
584                     options->sigconffile, strerror(errno));
585                 goto server_signature_random;
586             }
587         } else {
588             LOG(log_error, logtype_afpd, "ERROR: Cannot read %s (%s). Using one-time signature.",
589                 options->sigconffile, strerror(errno));
590             goto server_signature_random;
591         }
592     } else {                                                          /* conf file don't exist */
593         if (( fd = creat(options->sigconffile, 0644 )) < 0 ) {
594             LOG(log_error, logtype_afpd, "ERROR: Cannot create %s (%s). Using one-time signature.",
595                 options->sigconffile, strerror(errno));
596             goto server_signature_random;
597         }
598         if (( fp = fdopen( fd, "w" )) == NULL ) {
599             LOG(log_error, logtype_afpd, "ERROR: Cannot fdopen %s (%s). Using one-time signature.",
600                 options->sigconffile, strerror(errno));
601             close(fd);
602             goto server_signature_random;
603         }
604         header = 1;
605         goto server_signature_random;
606     }
607     
608 server_signature_random:
609     
610     /* generate signature from random number */
611     randombytes(options->signature, 16);
612
613     if (fp && header) { /* conf file is created or size=0 */
614         fprintf(fp, "# DON'T TOUCH NOR COPY THOUGHTLESSLY!\n");
615         fprintf(fp, "# This file is auto-generated by afpd.\n");
616         fprintf(fp, "# \n");
617         fprintf(fp, "# ServerSignature is unique identifier used to prevent logging on to\n");
618         fprintf(fp, "# the same server twice.\n");
619         fprintf(fp, "# \n");
620         fprintf(fp, "# If setting \"signature = xxxxx\" in afp.conf, this file is not used.\n\n");
621     }
622     
623     if (fp) {
624         fprintf(fp, "\"%s\"\t", server_tmp);
625         for (i=0 ; i<16 ; i++) {
626             fprintf(fp, "%02X", (options->signature)[i]);
627         }
628         fprintf(fp, "%s", "\n");
629         fclose(fp);
630     }
631     
632 server_signature_done:
633     
634     /* retrun */
635     LOG(log_info, logtype_afpd,
636         "signature is %02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X",
637         (options->signature)[ 0], (options->signature)[ 1],
638         (options->signature)[ 2], (options->signature)[ 3],
639         (options->signature)[ 4], (options->signature)[ 5],
640         (options->signature)[ 6], (options->signature)[ 7],
641         (options->signature)[ 8], (options->signature)[ 9],
642         (options->signature)[10], (options->signature)[11],
643         (options->signature)[12], (options->signature)[13],
644         (options->signature)[14], (options->signature)[15]);
645     
646     return;
647 }
648
649 /* this is the same as asp/dsi_getstatus */
650 int afp_getsrvrinfo(AFPObj *obj, char *ibuf _U_, size_t ibuflen _U_, char *rbuf, size_t *rbuflen)
651 {
652     memcpy(rbuf, obj->dsi->status, obj->dsi->statuslen);
653     *rbuflen = obj->dsi->statuslen;
654
655     return AFP_OK;
656 }