etc/afpd/messages.c

   1 /*
   2  * $Id: messages.c,v 1.16.6.1.2.3 2003-10-30 05:57:44 bfernhomberg Exp $
   3  *
   4  * Copyright (c) 1997 Adrian Sun (asun@zoology.washington.edu)
   5  * All Rights Reserved.  See COPYRIGHT.
   6  */
   7
   8 #ifdef HAVE_CONFIG_H
   9 #include "config.h"
  10 #endif /* HAVE_CONFIG_H */
  11
  12 #ifdef HAVE_UNISTD_H
  13 #include <unistd.h>
  14 #endif /* HAVE_UNISTD_H */
  15 #include <stdio.h>
  16 #include <string.h>
  17 #include <errno.h>
  18 #include <stdlib.h>
  19 #include <atalk/afp.h>
  20 #include <atalk/dsi.h>
  21 #include <atalk/logger.h>
  22 #include "globals.h"
  23 #include "misc.h"
  24
  25
  26 #define MAXMESGSIZE 199
  27
  28 /* this is only used by afpd children, so it's okay. */
  29 static char servermesg[MAXPATHLEN] = "";
  30 static char localized_message[MAXPATHLEN] = "";
  31
  32 void setmessage(const char *message)
  33 {
  34     strncpy(servermesg, message, MAXMESGSIZE);
  35 }
  36
  37 void readmessage(obj)
  38 AFPObj *obj;
  39 {
  40     /* Read server message from file defined as SERVERTEXT */
  41 #ifdef SERVERTEXT
  42     FILE *message;
  43     char * filename;
  44     unsigned int i;
  45     int rc;
  46     static int c;
  47     uid_t euid;
  48     u_int32_t maxmsgsize;
  49
  50     maxmsgsize = (obj->proto == AFPPROTO_DSI)?MIN(MAX(((DSI*)obj->handle)->attn_quantum, MAXMESGSIZE),MAXPATHLEN):MAXMESGSIZE;
  51
  52     i=0;
  53     /* Construct file name SERVERTEXT/message.[pid] */
  54     if ( NULL == (filename=(char*) malloc(sizeof(SERVERTEXT)+15)) ) {
  55         LOG(log_error, logtype_afpd, "readmessage: malloc: %s", strerror(errno) );
  56         return;
  57     }
  58
  59     sprintf(filename, "%s/message.%d", SERVERTEXT, getpid());
  60
  61 #ifdef DEBUG
  62     LOG(log_debug, logtype_afpd, "Reading file %s ", filename);
  63 #endif /* DEBUG */
  64
  65     message=fopen(filename, "r");
  66     if (message==NULL) {
  67         LOG(log_info, logtype_afpd, "Unable to open file %s", filename);
  68         sprintf(filename, "%s/message", SERVERTEXT);
  69         message=fopen(filename, "r");
  70     }
  71
  72     /* if either message.pid or message exists */
  73     if (message!=NULL) {
  74         /* added while loop to get characters and put in servermesg */
  75         while ((( c=fgetc(message)) != EOF) && (i < (maxmsgsize - 1))) {
  76             if ( c == '\n')  c = ' ';
  77             servermesg[i++] = c;
  78         }
  79         servermesg[i] = 0;
  80
  81         /* cleanup */
  82         fclose(message);
  83
  84         /* Save effective uid and switch to root to delete file. */
  85         /* Delete will probably fail otherwise, but let's try anyways */
  86         euid = geteuid();
  87         if (seteuid(0) < 0) {
  88             LOG(log_error, logtype_afpd, "Could not switch back to root: %s",
  89                                 strerror(errno));
  90         }
  91
  92         if ( 0 < (rc = unlink(filename)) )
  93             LOG(log_error, logtype_afpd, "File '%s' could not be deleted", strerror(errno));
  94
  95         /* Drop privs again, failing this is very bad */
  96         if (seteuid(euid) < 0) {
  97             LOG(log_error, logtype_afpd, "Could not switch back to uid %d: %s", euid, strerror(errno));
  98         }
  99
 100         if (rc < 0) {
 101             LOG(log_error, logtype_afpd, "Error deleting %s: %s", filename, strerror(rc));
 102         }
 103 #ifdef DEBUG
 104         else {
 105             LOG(log_info, logtype_afpd, "Deleted %s", filename);
 106         }
 107
 108         LOG(log_info, logtype_afpd, "Set server message to \"%s\"", servermesg);
 109 #endif /* DEBUG */
 110     }
 111     free(filename);
 112 #endif /* SERVERTEXT */
 113 }
 114
 115 int afp_getsrvrmesg(obj, ibuf, ibuflen, rbuf, rbuflen)
 116 AFPObj *obj;
 117 char *ibuf, *rbuf;
 118 int ibuflen, *rbuflen;
 119 {
 120     char *message;
 121     u_int16_t type, bitmap;
 122     u_int32_t msgsize;
 123     size_t outlen;
 124
 125     msgsize = (obj->proto == AFPPROTO_DSI)?MAX(((DSI*)obj->handle)->attn_quantum, MAXMESGSIZE):MAXMESGSIZE;
 126
 127     memcpy(&type, ibuf + 2, sizeof(type));
 128     memcpy(&bitmap, ibuf + 4, sizeof(bitmap));
 129
 130     switch (ntohs(type)) {
 131     case AFPMESG_LOGIN: /* login */
 132         message = obj->options.loginmesg;
 133         break;
 134     case AFPMESG_SERVER: /* server */
 135         message = servermesg;
 136         break;
 137     default:
 138         *rbuflen = 0;
 139         return AFPERR_BITMAP;
 140     }
 141
 142     /* output format:
 143      * message type:   2 bytes
 144      * bitmap:         2 bytes
 145      * message length: 1 byte
 146      * message:        up to 199 bytes
 147      */
 148     memcpy(rbuf, &type, sizeof(type));
 149     rbuf += sizeof(type);
 150     memcpy(rbuf, &bitmap, sizeof(bitmap));
 151     rbuf += sizeof(bitmap);
 152     *rbuflen = strlen(message);
 153 #if 0
 154     if (*rbuflen > MAXMESGSIZE)
 155         *rbuflen = MAXMESGSIZE;
 156 #endif
 157     if (*rbuflen > msgsize)
 158         *rbuflen = msgsize;
 159     *rbuf++ = *rbuflen;
 160
 161     /* Convert the message to the macs codepage
 162      * according to AFP 3.1 specs page 200
 163      * bit 1 set in bitmap means Unicode ?= utf8
 164      * Never saw this in the wild yet             */
 165
 166     if ( (size_t)-1 == (outlen = convert_string(obj->options.unixcharset, ((ntohs(bitmap)) & 2)?CH_UTF8_MAC:obj->options.maccharset, message, *rbuflen, localized_message, msgsize)) )
 167         memcpy(rbuf, message, *rbuflen); /*FIXME*/
 168     else
 169         memcpy(rbuf, localized_message, outlen);
 170
 171     *rbuflen += 5;
 172
 173     return AFP_OK;
 174 }