X-Git-Url: https://code.citadel.org/?a=blobdiff_plain;f=webcit%2Fdownloads.c;h=6e6c9b981b5ebf0acf8bc46dc664fc9630956acd;hb=ed4e4918cc03352a056f097ea9e8d24918c49158;hp=60783c5685cad880b94c1f877dc821c473891218;hpb=fb6f6fa4ec4e3277e30d84326d48e6850822d318;p=citadel.git

diff --git a/webcit/downloads.c b/webcit/downloads.c
index 60783c568..6e6c9b981 100644
--- a/webcit/downloads.c
+++ b/webcit/downloads.c
@@ -13,6 +13,8 @@
 #include "webcit.h"
 #include "webserver.h"
 
+CtxType CTX_FILELIST = CTX_NONE;
+
 extern void output_static(const char* What);
 
 extern char* static_dirs[];
@@ -38,22 +40,22 @@ void FreeFiles(void *vFile)
 /* -------------------------------------------------------------------------------- */
 void tmplput_FILE_NAME(StrBuf *Target, WCTemplputParams *TP)
 {
-	FileListStruct *F = (FileListStruct*) CTX;
+	FileListStruct *F = (FileListStruct*) CTX(CTX_FILELIST);
 	StrBufAppendTemplate(Target, TP, F->Filename, 0);
 }
 void tmplput_FILE_SIZE(StrBuf *Target, WCTemplputParams *TP)
 {
-	FileListStruct *F = (FileListStruct*) CTX;
+	FileListStruct *F = (FileListStruct*) CTX(CTX_FILELIST);
 	StrBufAppendPrintf(Target, "%ld", F->FileSize);
 }
 void tmplput_FILEMIMETYPE(StrBuf *Target, WCTemplputParams *TP)
 {
-	FileListStruct *F = (FileListStruct*) CTX;
+	FileListStruct *F = (FileListStruct*) CTX(CTX_FILELIST);
 	StrBufAppendTemplate(Target, TP, F->MimeType, 0);
 }
 void tmplput_FILE_COMMENT(StrBuf *Target, WCTemplputParams *TP)
 {
-	FileListStruct *F = (FileListStruct*) CTX;
+	FileListStruct *F = (FileListStruct*) CTX(CTX_FILELIST);
 	StrBufAppendTemplate(Target, TP, F->Comment, 0);
 }
 
@@ -61,7 +63,7 @@ void tmplput_FILE_COMMENT(StrBuf *Target, WCTemplputParams *TP)
 
 int Conditional_FILE_ISPIC(StrBuf *Target, WCTemplputParams *TP)
 {
-	FileListStruct *F = (FileListStruct*) CTX;
+	FileListStruct *F = (FileListStruct*) CTX(CTX_FILELIST);
 	return F->IsPic;
 }
 
@@ -268,13 +270,15 @@ void download_file(void)
 	if (GetServerStatus(Buf, NULL) == 2) {
 		StrBufCutLeft(Buf, 4);
 		bytes = StrBufExtract_long(Buf, 0, '|');
-		if (!force_download) {
-			StrBufExtract_token(ContentType, Buf, 3, '|');
-		}
-		serv_read_binary(WCC->WBuf, bytes, Buf);
+		StrBufExtract_token(ContentType, Buf, 3, '|');
+
+		CheckGZipCompressionAllowed (SKEY(ContentType));
+		if (force_download)
+			FlushStrBuf(ContentType);
+
+		serv_read_binary_to_http(ContentType, bytes, 0, 0);
 		serv_puts("CLOS");
 		StrBuf_ServGetln(Buf);
-		http_transmit_thing(ChrPtr(ContentType), 0);
 	} else {
 		StrBufCutLeft(Buf, 4);
 		hprintf("HTTP/1.1 404 %s\n", ChrPtr(Buf));
@@ -395,6 +399,7 @@ void output_image(void)
 			/** Write it to the browser */
 			if (!IsEmptyStr(MimeType))
 			{
+				CheckGZipCompressionAllowed (MimeType, strlen(MimeType));
 				http_transmit_thing(MimeType, 0);
 				FreeStrBuf(&Buf);
 				return;
@@ -420,6 +425,7 @@ void
 InitModule_DOWNLOAD
 (void)
 {
+	RegisterCTX(CTX_FILELIST);
 
 	RegisterIterator("ROOM:FILES", 0, NULL, LoadFileList,
 			 NULL, DeleteHash, CTX_FILELIST, CTX_NONE, 
@@ -461,7 +467,7 @@ InitModule_DOWNLOAD
 	RegisterNamespace("FILE:MIMETYPE", 0, 2, tmplput_FILEMIMETYPE, NULL, CTX_FILELIST);
 	RegisterNamespace("FILE:COMMENT", 0, 2, tmplput_FILE_COMMENT, NULL, CTX_FILELIST);
 
-	RegisterConditional(HKEY("COND:FILE:ISPIC"), 0, Conditional_FILE_ISPIC, CTX_FILELIST);
+	RegisterConditional("COND:FILE:ISPIC", 0, Conditional_FILE_ISPIC, CTX_FILELIST);
 
 	WebcitAddUrlHandler(HKEY("image"), "", 0, output_image, ANONYMOUS);
 	WebcitAddUrlHandler(HKEY("display_mime_icon"), "", 0, display_mime_icon , ANONYMOUS);