etc/afpd/messages.c

   1 /*
   2  * Copyright (c) 1997 Adrian Sun (asun@zoology.washington.edu)
   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 <unistd.h>
  11 #include <stdio.h>
  12 #include <string.h>
  13 #include <errno.h>
  14 #include <stdlib.h>
  15 #include <atalk/afp.h>
  16 #include <atalk/dsi.h>
  17 #include <atalk/util.h>
  18 #include <atalk/unix.h>
  19 #include <atalk/logger.h>
  20 #include <atalk/globals.h>
  21
  22 #include "misc.h"
  23
  24 #define MAXMESGSIZE 199
  25
  26 /* this is only used by afpd children, so it's okay. */
  27 static char servermesg[MAXPATHLEN] = "";
  28 static char localized_message[MAXPATHLEN] = "";
  29
  30 /*!
  31  * Copy AFP message to message buffer
  32  * @param message (r) message to send
  33  * @returns 0 if this message is being set the first time, return 1 if the preceeding
  34  *          message was the same
  35  */
  36 int setmessage(const char *message)
  37 {
  38     if (strncmp(message, servermesg, MAXMESGSIZE) == 0)
  39         return 1;
  40     strlcpy(servermesg, message, MAXMESGSIZE);
  41     return 0;
  42 }
  43
  44 void readmessage(AFPObj *obj)
  45 {
  46     /* Read server message from file defined as SERVERTEXT */
  47 #ifdef SERVERTEXT
  48     FILE *message;
  49     char * filename;
  50     unsigned int i;
  51     int rc;
  52     static int c;
  53     uint32_t maxmsgsize;
  54
  55     maxmsgsize = MIN(MAX(obj->dsi->attn_quantum, MAXMESGSIZE), MAXPATHLEN);
  56
  57     i=0;
  58     /* Construct file name SERVERTEXT/message.[pid] */
  59     if ( NULL == (filename=(char*) malloc(sizeof(SERVERTEXT)+15)) ) {
  60         LOG(log_error, logtype_afpd, "readmessage: malloc: %s", strerror(errno) );
  61         return;
  62     }
  63
  64     sprintf(filename, "%s/message.%d", SERVERTEXT, getpid());
  65
  66 #ifdef DEBUG
  67     LOG(log_debug9, logtype_afpd, "Reading file %s ", filename);
  68 #endif
  69
  70     message=fopen(filename, "r");
  71     if (message==NULL) {
  72         /* try without the process id */
  73         sprintf(filename, "%s/message", SERVERTEXT);
  74         message=fopen(filename, "r");
  75     }
  76
  77     /* if either message.pid or message exists */
  78     if (message!=NULL) {
  79         /* added while loop to get characters and put in servermesg */
  80         while ((( c=fgetc(message)) != EOF) && (i < (maxmsgsize - 1))) {
  81             if ( c == '\n')  c = ' ';
  82             servermesg[i++] = c;
  83         }
  84         servermesg[i] = 0;
  85
  86         /* cleanup */
  87         fclose(message);
  88
  89         become_root();
  90
  91         if ((rc = unlink(filename)) != 0)
  92             LOG(log_error, logtype_afpd, "File '%s' could not be deleted", strerror(errno));
  93
  94         unbecome_root();
  95
  96         if (rc < 0) {
  97             LOG(log_error, logtype_afpd, "Error deleting %s: %s", filename, strerror(rc));
  98         }
  99 #ifdef DEBUG
 100         else {
 101             LOG(log_debug9, logtype_afpd, "Deleted %s", filename);
 102         }
 103
 104         LOG(log_debug9, logtype_afpd, "Set server message to \"%s\"", servermesg);
 105 #endif
 106     }
 107     free(filename);
 108 #endif /* SERVERTEXT */
 109 }
 110
 111 int afp_getsrvrmesg(AFPObj *obj, char *ibuf, size_t ibuflen _U_, char *rbuf, size_t *rbuflen)
 112 {
 113     char *message;
 114     uint16_t type, bitmap;
 115     uint16_t msgsize;
 116     size_t outlen = 0;
 117     size_t msglen = 0;
 118     int utf8 = 0;
 119
 120     *rbuflen = 0;
 121
 122     msgsize = MAX(obj->dsi->attn_quantum, MAXMESGSIZE);
 123
 124     memcpy(&type, ibuf + 2, sizeof(type));
 125     memcpy(&bitmap, ibuf + 4, sizeof(bitmap));
 126
 127     message = servermesg;
 128     switch (ntohs(type)) {
 129     case AFPMESG_LOGIN: /* login */
 130         /* at least TIGER loses server messages
 131          * if it receives a server msg attention before
 132          * it has asked the login msg...
 133          * Workaround: concatenate the two if any, ugly.
 134          */
 135         if (obj->options.loginmesg) {
 136             if (*message)
 137                 strlcat(message, " - ", MAXMESGSIZE);
 138             strlcat(message, obj->options.loginmesg, MAXMESGSIZE);
 139         }
 140         break;
 141     case AFPMESG_SERVER: /* server */
 142         break;
 143     default:
 144         return AFPERR_BITMAP;
 145     }
 146
 147     /* output format:
 148      * message type:   2 bytes
 149      * bitmap:         2 bytes
 150      * message length: 1 byte ( 2 bytes for utf8)
 151      * message:        up to 199 bytes (dsi attn_quantum for utf8)
 152      */
 153     memcpy(rbuf, &type, sizeof(type));
 154     rbuf += sizeof(type);
 155     *rbuflen += sizeof(type);
 156     memcpy(rbuf, &bitmap, sizeof(bitmap));
 157     rbuf += sizeof(bitmap);
 158     *rbuflen += sizeof(bitmap);
 159
 160     utf8 = ntohs(bitmap) & 2;
 161     msglen = strlen(message);
 162     if (msglen > msgsize)
 163         msglen = msgsize;
 164
 165     if (msglen) {
 166         if ( (size_t)-1 == (outlen = convert_string(obj->options.unixcharset, utf8?CH_UTF8_MAC:obj->options.maccharset,
 167                                                     message, msglen, localized_message, msgsize)) )
 168         {
 169             memcpy(rbuf+((utf8)?2:1), message, msglen); /*FIXME*/
 170             outlen = msglen;
 171         }
 172         else
 173         {
 174             memcpy(rbuf+((utf8)?2:1), localized_message, outlen);
 175         }
 176     }
 177
 178     if ( utf8 ) {
 179         /* UTF8 message, 2 byte length */
 180         msgsize = htons(outlen);
 181         memcpy(rbuf, &msgsize, sizeof(msgsize));
 182         *rbuflen += sizeof(msgsize);
 183     }
 184     else {
 185         *rbuf = outlen;
 186         *rbuflen += 1;
 187     }
 188     *rbuflen += outlen;
 189 //    *message = 0;
 190     return AFP_OK;
 191 }