citadel/modules/image/serv_image.c

   1 /*
   2  * Copyright (c) 1987-2021 by the citadel.org team
   3  *
   4  * This program is open source software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  */
  14
  15 #include "ctdl_module.h"
  16 #include "config.h"
  17 #include <sys/types.h>
  18 #include <sys/stat.h>
  19 #include <dirent.h>
  20
  21
  22 /*
  23  * DownLoad Room Image (see its icon or whatever)
  24  * If this command succeeds, it follows the same protocol as the DLAT command.
  25  */
  26 void cmd_dlri(char *cmdbuf) {
  27         if (CtdlAccessCheck(ac_logged_in_or_guest)) return;
  28         if (CC->room.msgnum_pic < 1) {
  29                 cprintf("%d No image found.\n", ERROR + FILE_NOT_FOUND);
  30                 return;
  31         }
  32
  33         struct CtdlMessage *msg = CtdlFetchMessage(CC->room.msgnum_pic, 1);
  34         if (msg != NULL) {
  35                 // The call to CtdlOutputPreLoadedMsg() with MT_SPEW_SECTION will cause the DLRI command
  36                 // to have the same output format as the DLAT command, because it calls the same code.
  37                 // For example: 600 402132|-1||image/gif|
  38                 safestrncpy(CC->download_desired_section, "1", sizeof CC->download_desired_section);
  39                 CtdlOutputPreLoadedMsg(msg, MT_SPEW_SECTION, HEADERS_NONE, 1, 0, 0);
  40                 CM_Free(msg);
  41         }
  42         else {
  43                 cprintf("%d No image found.\n", ERROR + MESSAGE_NOT_FOUND);
  44                 return;
  45         }
  46 }
  47
  48
  49 /*
  50  * UpLoad Room Image (avatar or photo or whatever)
  51  */
  52 void cmd_ulri(char *cmdbuf) {
  53         long data_length;
  54         char mimetype[SIZ];
  55
  56         if (CtdlAccessCheck(ac_room_aide)) return;
  57
  58         data_length = extract_long(cmdbuf, 0);
  59         extract_token(mimetype, cmdbuf, 1, '|', sizeof mimetype);
  60
  61         if (data_length < 20) {
  62                 cprintf("%d That's an awfully small file.  Try again.\n", ERROR + ILLEGAL_VALUE);
  63                 return;
  64         }
  65
  66         if (strncasecmp(mimetype, "image/", 6)) {
  67                 cprintf("%d Only image files are permitted.\n", ERROR + ILLEGAL_VALUE);
  68                 return;
  69         }
  70
  71         char *unencoded_data = malloc(data_length + 1);
  72         if (!unencoded_data) {
  73                 cprintf("%d Could not allocate %ld bytes of memory\n", ERROR + INTERNAL_ERROR , data_length);
  74                 return;
  75         }
  76
  77         cprintf("%d %ld\n", SEND_BINARY, data_length);
  78         client_read(unencoded_data, data_length);
  79
  80         // We've got the data read from the client, now save it.
  81         char *encoded_data = malloc((data_length * 2) + 100);
  82         if (encoded_data) {
  83                 sprintf(encoded_data, "Content-type: %s\nContent-transfer-encoding: base64\n\n", mimetype);
  84                 CtdlEncodeBase64(&encoded_data[strlen(encoded_data)], unencoded_data, data_length, 1);
  85                 long new_msgnum = quickie_message("Citadel", NULL, NULL, SYSCONFIGROOM, encoded_data, FMT_RFC822, "Image uploaded by admin user");
  86
  87                 if (CtdlGetRoomLock(&CC->room, CC->room.QRname) == 0) {
  88                         long old_msgnum = CC->room.msgnum_pic;
  89                         syslog(LOG_DEBUG, "Message %ld is now the photo for %s", new_msgnum, CC->room.QRname);
  90                         CC->room.msgnum_pic = new_msgnum;
  91                         CtdlPutRoomLock(&CC->room);
  92                         if (old_msgnum > 0) {
  93                                 syslog(LOG_DEBUG, "Deleting old message %ld from %s", old_msgnum, SYSCONFIGROOM);
  94                                 CtdlDeleteMessages(SYSCONFIGROOM, &old_msgnum, 1, "");
  95                         }
  96                 }
  97                 free(encoded_data);
  98         }
  99
 100         free(unencoded_data);
 101 }
 102
 103
 104 /*
 105  * DownLoad User Image (see their avatar or photo or whatever)
 106  * If this command succeeds, it follows the same protocol as the DLAT command.
 107  */
 108 void cmd_dlui(char *cmdbuf) {
 109         struct ctdluser ruser;
 110         char buf[SIZ];
 111
 112         if (CtdlAccessCheck(ac_logged_in_or_guest)) return;
 113         extract_token(buf, cmdbuf, 0, '|', sizeof buf);
 114         if (CtdlGetUser(&ruser, buf) != 0) {
 115                 cprintf("%d No such user.\n", ERROR + NO_SUCH_USER);
 116                 return;
 117         }
 118         if (ruser.msgnum_pic < 1) {
 119                 cprintf("%d No image found.\n", ERROR + FILE_NOT_FOUND);
 120                 return;
 121         }
 122
 123         struct CtdlMessage *msg = CtdlFetchMessage(ruser.msgnum_pic, 1);
 124         if (msg != NULL) {
 125                 // The call to CtdlOutputPreLoadedMsg() with MT_SPEW_SECTION will cause the DLUI command
 126                 // to have the same output format as the DLAT command, because it calls the same code.
 127                 // For example: 600 402132|-1||image/gif|
 128                 safestrncpy(CC->download_desired_section, "1", sizeof CC->download_desired_section);
 129                 CtdlOutputPreLoadedMsg(msg, MT_SPEW_SECTION, HEADERS_NONE, 1, 0, 0);
 130                 CM_Free(msg);
 131         }
 132         else {
 133                 cprintf("%d No image found.\n", ERROR + MESSAGE_NOT_FOUND);
 134                 return;
 135         }
 136 }
 137
 138
 139 /*
 140  * UpLoad User Image (avatar or photo or whatever)
 141  */
 142 void cmd_ului(char *cmdbuf) {
 143         long data_length;
 144         char mimetype[SIZ];
 145         char username[USERNAME_SIZE];
 146         char userconfigroomname[ROOMNAMELEN];
 147
 148         if (CtdlAccessCheck(ac_logged_in_or_guest)) return;
 149
 150         if (num_parms(cmdbuf) < 2) {
 151                 cprintf("%d Usage error\n", ERROR + ILLEGAL_VALUE);
 152                 return;
 153         }
 154
 155         data_length = extract_long(cmdbuf, 0);
 156         extract_token(mimetype, cmdbuf, 1, '|', sizeof mimetype);
 157         extract_token(username, cmdbuf, 2, '|', sizeof username);
 158
 159         if (data_length < 20) {
 160                 cprintf("%d That's an awfully small file.  Try again.\n", ERROR + ILLEGAL_VALUE);
 161                 return;
 162         }
 163
 164         if (strncasecmp(mimetype, "image/", 6)) {
 165                 cprintf("%d Only image files are permitted.\n", ERROR + ILLEGAL_VALUE);
 166                 return;
 167         }
 168
 169         if (IsEmptyStr(username)) {
 170                 safestrncpy(username, CC->curr_user, sizeof username);
 171         }
 172
 173         // Normal users can only change their own photo
 174         if ( (strcasecmp(username, CC->curr_user)) && (CC->user.axlevel < AxAideU) && (!CC->internal_pgm) ) {
 175                 cprintf("%d Higher access required to change another user's photo.\n", ERROR + HIGHER_ACCESS_REQUIRED);
 176         }
 177
 178         // Check to make sure the user exists
 179         struct ctdluser usbuf;
 180         if (CtdlGetUser(&usbuf, username) != 0) {               // check for existing user, don't lock it yet
 181                 cprintf("%d %s not found.\n", ERROR + NO_SUCH_USER , username);
 182                 return;
 183         }
 184         CtdlMailboxName(userconfigroomname, sizeof userconfigroomname, &usbuf, USERCONFIGROOM);
 185
 186         char *unencoded_data = malloc(data_length + 1);
 187         if (!unencoded_data) {
 188                 cprintf("%d Could not allocate %ld bytes of memory\n", ERROR + INTERNAL_ERROR , data_length);
 189                 return;
 190         }
 191
 192         cprintf("%d %ld\n", SEND_BINARY, data_length);
 193         client_read(unencoded_data, data_length);
 194
 195         // We've got the data read from the client, now save it.
 196         char *encoded_data = malloc((data_length * 2) + 100);
 197         if (encoded_data) {
 198                 sprintf(encoded_data, "Content-type: %s\nContent-transfer-encoding: base64\n\n", mimetype);
 199                 CtdlEncodeBase64(&encoded_data[strlen(encoded_data)], unencoded_data, data_length, 1);
 200                 long new_msgnum = quickie_message("Citadel", NULL, NULL, userconfigroomname, encoded_data, FMT_RFC822, "Photo uploaded by user");
 201
 202                 if (CtdlGetUserLock(&usbuf, username) == 0) {   // lock it this time
 203                         long old_msgnum = usbuf.msgnum_pic;
 204                         syslog(LOG_DEBUG, "Message %ld is now the photo for %s", new_msgnum, username);
 205                         usbuf.msgnum_pic = new_msgnum;
 206                         CtdlPutUserLock(&usbuf);
 207                         if (old_msgnum > 0) {
 208                                 syslog(LOG_DEBUG, "Deleting old message %ld from %s", old_msgnum, userconfigroomname);
 209                                 CtdlDeleteMessages(userconfigroomname, &old_msgnum, 1, "");
 210                         }
 211                 }
 212
 213                 free(encoded_data);
 214         }
 215
 216         free(unencoded_data);
 217 }
 218
 219
 220 /*
 221  * Import function called by import_old_userpic_files() for a single user
 222  */
 223 void import_one_userpic_file(char *username, long usernum, char *path) {
 224         syslog(LOG_DEBUG, "Import legacy userpic for %s, usernum=%ld, filename=%s", username, usernum, path);
 225
 226         FILE *fp = fopen(path, "r");
 227         if (!fp) return;
 228
 229         fseek(fp, 0, SEEK_END);
 230         long data_length = ftell(fp);
 231
 232         if (data_length >= 1) {
 233                 rewind(fp);
 234                 char *unencoded_data = malloc(data_length);
 235                 if (unencoded_data) {
 236                         fread(unencoded_data, data_length, 1, fp);
 237                         char *encoded_data = malloc((data_length * 2) + 100);
 238                         if (encoded_data) {
 239                                 sprintf(encoded_data, "Content-type: %s\nContent-transfer-encoding: base64\n\n", GuessMimeByFilename(path, strlen(path)));
 240                                 CtdlEncodeBase64(&encoded_data[strlen(encoded_data)], unencoded_data, data_length, 1);
 241
 242                                 char userconfigroomname[ROOMNAMELEN];
 243                                 struct ctdluser usbuf;
 244
 245                                 if (CtdlGetUser(&usbuf, username) == 0) {       // no need to lock it , we are still initializing
 246                                         long old_msgnum = usbuf.msgnum_pic;
 247                                         CtdlMailboxName(userconfigroomname, sizeof userconfigroomname, &usbuf, USERCONFIGROOM);
 248                                         long new_msgnum = quickie_message("Citadel", NULL, NULL, userconfigroomname, encoded_data, FMT_RFC822, "Photo imported from file");
 249                                         syslog(LOG_DEBUG, "Message %ld is now the photo for %s", new_msgnum, username);
 250                                         usbuf.msgnum_pic = new_msgnum;
 251                                         CtdlPutUser(&usbuf);
 252                                         unlink(path);                           // delete the old file , it's in the database now
 253                                         if (old_msgnum > 0) {
 254                                                 syslog(LOG_DEBUG, "Deleting old message %ld from %s", old_msgnum, userconfigroomname);
 255                                                 CtdlDeleteMessages(userconfigroomname, &old_msgnum, 1, "");
 256                                         }
 257                                 }
 258                                 free(encoded_data);
 259                         }
 260                         free(unencoded_data);
 261                 }
 262         }
 263         fclose(fp);
 264 }
 265
 266
 267 /*
 268  * Look for old-format "userpic" files and import them into the message base
 269  */
 270 void import_old_userpic_files(void) {
 271         DIR *filedir = NULL;
 272         struct dirent *filedir_entry;
 273         size_t d_namelen;
 274         struct ctdluser usbuf;
 275         long usernum = 0;
 276         int d_type = 0;
 277         struct stat s;
 278         char path[PATH_MAX];
 279
 280
 281         syslog(LOG_DEBUG, "Importing old style userpic files into the message base");
 282         filedir = opendir (ctdl_usrpic_dir);
 283         if (filedir == NULL) {
 284                 return;
 285         }
 286         while ( (filedir_entry = readdir(filedir)) , (filedir_entry != NULL))
 287         {
 288 #ifdef _DIRENT_HAVE_D_NAMLEN
 289                 d_namelen = filedir_entry->d_namlen;
 290
 291 #else
 292                 d_namelen = strlen(filedir_entry->d_name);
 293 #endif
 294
 295 #ifdef _DIRENT_HAVE_D_TYPE
 296                 d_type = filedir_entry->d_type;
 297 #else
 298
 299 #ifndef DT_UNKNOWN
 300 #define DT_UNKNOWN     0
 301 #define DT_DIR         4
 302 #define DT_REG         8
 303 #define DT_LNK         10
 304
 305 #define IFTODT(mode)   (((mode) & 0170000) >> 12)
 306 #define DTTOIF(dirtype)        ((dirtype) << 12)
 307 #endif
 308                 d_type = DT_UNKNOWN;
 309 #endif
 310                 if ((d_namelen == 1) &&
 311                     (filedir_entry->d_name[0] == '.'))
 312                         continue;
 313
 314                 if ((d_namelen == 2) &&
 315                     (filedir_entry->d_name[0] == '.') &&
 316                     (filedir_entry->d_name[1] == '.'))
 317                         continue;
 318
 319                 snprintf(path, PATH_MAX, "%s/%s", ctdl_usrpic_dir, filedir_entry->d_name);
 320                 if (d_type == DT_UNKNOWN) {
 321                         if (lstat(path, &s) == 0) {
 322                                 d_type = IFTODT(s.st_mode);
 323                         }
 324                 }
 325                 switch (d_type)
 326                 {
 327                 case DT_DIR:
 328                         break;
 329                 case DT_LNK:
 330                 case DT_REG:
 331                         usernum = atol(filedir_entry->d_name);
 332                         if (CtdlGetUserByNumber(&usbuf, usernum) == 0) {
 333                                 import_one_userpic_file(usbuf.fullname, usernum, path);
 334                         }
 335                 }
 336         }
 337         closedir(filedir);
 338         rmdir(ctdl_usrpic_dir);
 339 }
 340
 341
 342
 343 CTDL_MODULE_INIT(image)
 344 {
 345         if (!threading)
 346         {
 347                 import_old_userpic_files();
 348                 CtdlRegisterProtoHook(cmd_dlri, "DLRI", "DownLoad Room Image");
 349                 CtdlRegisterProtoHook(cmd_ulri, "ULRI", "UpLoad Room Image");
 350                 CtdlRegisterProtoHook(cmd_dlui, "DLUI", "DownLoad User Image");
 351                 CtdlRegisterProtoHook(cmd_ului, "ULUI", "UpLoad User Image");
 352         }
 353         /* return our module name for the log */
 354         return "image";
 355 }