moar code heere

[citadel.git] / citadel / modules / image / serv_image.c

diff --git a/citadel/modules/image/serv_image.c b/citadel/modules/image/serv_image.c
index 36f6838db2720f2c4dd65ee77b908c41beb22d0f..2c7647f3fdca996b7b7c0628b615aeaf7362e246 100644 (file)
--- a/citadel/modules/image/serv_image.c
+++ b/citadel/modules/image/serv_image.c
@@ -20,6 +20,107 @@
 #include <dirent.h>
 
 
+
+/*
+ * DownLoad User Image (see their avatar or photo or whatever)
+ * If this command succeeds, it follows the same protocol as the DLAT command.
+ */
+void cmd_dlui(char *cmdbuf)
+{
+       struct ctdluser ruser;
+       char buf[SIZ];
+
+       if (CtdlAccessCheck(ac_logged_in_or_guest)) return;
+       extract_token(buf, cmdbuf, 0, '|', sizeof buf);
+       if (CtdlGetUser(&ruser, buf) != 0) {
+               cprintf("%d No such user.\n", ERROR + NO_SUCH_USER);
+               return;
+       }
+       if (ruser.msgnum_pic < 1) {
+               cprintf("%d No image found.\n", ERROR + FILE_NOT_FOUND);
+               return;
+       }
+
+       struct CtdlMessage *msg = CtdlFetchMessage(ruser.msgnum_pic, 1, 1);
+       if (msg != NULL) {
+               // The call to CtdlOutputPreLoadedMsg() with MT_SPEW_SECTION will cause the DLUI command
+               // to have the same output format as the DLAT command, because it calls the same code.
+               // For example: 600 402132|-1||image/gif|
+               safestrncpy(CC->download_desired_section, "1", sizeof CC->download_desired_section);
+               CtdlOutputPreLoadedMsg(msg, MT_SPEW_SECTION, HEADERS_NONE, 1, 0, 0);
+               CM_Free(msg);
+       }
+       else {
+               cprintf("%d No image found.\n", ERROR + MESSAGE_NOT_FOUND);
+               return;
+       }
+}
+
+
+/*
+ * DownLoad User Image (avatar or photo or whatever)
+ */
+void cmd_ului(char *cmdbuf)
+{
+       long data_length;
+       char mimetype[SIZ];
+       char username[USERNAME_SIZE];
+
+       if (CtdlAccessCheck(ac_logged_in_or_guest)) return;
+
+       if (num_parms(cmdbuf) < 2)
+       {
+               cprintf("%d Usage error\n", ERROR + ILLEGAL_VALUE);
+               return;
+       }
+
+       data_length = extract_long(cmdbuf, 0);
+       extract_token(mimetype, cmdbuf, 1, '|', sizeof mimetype);
+       extract_token(username, cmdbuf, 2, '|', sizeof username);
+
+       if (data_length < 20) {
+               cprintf("%d That's an awfully small file.  Try again.\n", ERROR + ILLEGAL_VALUE);
+               return;
+       }
+
+       if (strncasecmp(mimetype, "image/", 6)) {
+               cprintf("%d Only image files are permitted.\n", ERROR + ILLEGAL_VALUE);
+               return;
+       }
+
+       if (IsEmptyStr(username)) {
+               safestrncpy(username, CC->curr_user, sizeof username);
+       }
+
+       // Normal users can only change their own photo
+       if ( (strcasecmp(username, CC->curr_user)) && (CC->user.axlevel < AxAideU) && (!CC->internal_pgm) ) {
+               cprintf("%d Higher access required to change another user's photo.\n", ERROR + HIGHER_ACCESS_REQUIRED);
+       }
+
+       // Check to make sure the user exists
+       // FIXME do this
+       struct ctdluser usbuf;
+       if (CtdlGetUser(&usbuf, username) != 0) {               // check for existing user, don't lock it yet
+               cprintf("%d %s not found.\n", ERROR + NO_SUCH_USER , username);
+               return;
+       }
+
+       char *unencoded_data = malloc(data_length + 1);
+       if (!unencoded_data) {
+               cprintf("%d Could not allocate %ld bytes of memory\n", ERROR + INTERNAL_ERROR , data_length);
+               return;
+       }
+
+       cprintf("%d %ld\n", SEND_BINARY, data_length);
+       client_read(unencoded_data, data_length);
+
+       // We've got the data read from the client, now save it.
+       // FIXME do this
+
+       free(unencoded_data);
+}
+
+
 /*
  * Import function called by import_old_userpic_files() for a single user
  */
@@ -40,6 +141,7 @@ void import_one_userpic_file(char *username, long usernum, char *path)
                        fread(unencoded_data, data_length, 1, fp);
                        char *encoded_data = malloc((data_length * 2) + 100);
                        if (encoded_data) {
+                               // FIXME try to guess the content-type based on the filename, don't assume GIF
                                sprintf(encoded_data, "Content-type: image/gif\nContent-transfer-encoding: base64\n\n");
                                CtdlEncodeBase64(&encoded_data[strlen(encoded_data)], unencoded_data, data_length, 1);
 
@@ -159,6 +261,8 @@ CTDL_MODULE_INIT(image)
        if (!threading)
        {
                import_old_userpic_files();
+               CtdlRegisterProtoHook(cmd_dlui, "DLUI", "DownLoad User Image");
+               CtdlRegisterProtoHook(cmd_ului, "ULUI", "UpLoad User Image");
        }
        /* return our module name for the log */
         return "image";