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