X-Git-Url: https://code.citadel.org/?a=blobdiff_plain;f=libcitadel%2Flib%2Fstringbuf.c;h=5e89ea2f19b48f8c38edd1b5835f3754109f3379;hb=e2c97f4b0f5420b21f491d6e61dca580f8772d39;hp=90c8cf3525b4c71c5deee20441eb26567a66ac90;hpb=12b9abbeeaaa4b866cd0212e85e8104c50dfd84e;p=citadel.git

diff --git a/libcitadel/lib/stringbuf.c b/libcitadel/lib/stringbuf.c
index 90c8cf352..5e89ea2f1 100644
--- a/libcitadel/lib/stringbuf.c
+++ b/libcitadel/lib/stringbuf.c
@@ -32,6 +32,9 @@
 
 #include "libcitadel.h"
 
+#include "b64/cencode.h"
+#include "b64/cdecode.h"
+
 #ifdef HAVE_ICONV
 #include <iconv.h>
 #endif
@@ -2823,6 +2826,76 @@ int StrBufDecodeBase64(StrBuf *Buf)
 	return siz;
 }
 
+/**
+ * @ingroup StrBuf_DeEnCoder
+ * @brief decode a buffer from base 64 encoding; expects targetbuffer
+ * @param BufIn Buffor to transform
+ * @param BufOut Buffer to put result into
+ */
+int StrBufDecodeBase64To(const StrBuf *BufIn, StrBuf *BufOut)
+{
+	if ((BufIn == NULL) || (BufOut == NULL))
+		return -1;
+
+	if (BufOut->BufSize < BufIn->BufUsed)
+		IncreaseBuf(BufOut, BufIn->BufUsed, 0);
+
+	BufOut->BufUsed = CtdlDecodeBase64(BufOut->buf,
+					   BufIn->buf,
+					   BufIn->BufUsed);
+	return BufOut->BufUsed;
+}
+
+void *StrBufNewStreamContext(eStreamType type)
+{
+	base64_decodestate *state;;
+
+	switch (type)
+	{
+	case eBase64Decode:
+		state = (base64_decodestate*) malloc(sizeof(base64_decodestate));
+		base64_init_decodestate(state);
+		return state;
+		break;
+	}
+	return NULL;
+}
+
+void StrBufDestroyStreamContext(eStreamType type, void **Stream)
+{
+	switch (type)
+	{
+	case eBase64Decode:
+		free(*Stream);
+		*Stream = NULL;
+		break;
+	}
+}
+void StrBufStreamDecodeTo(StrBuf *Target, const StrBuf *In, const char* pIn, long pInLen, void *Stream)
+{
+	base64_decodestate *state = Stream;
+	long ExpectLen;
+
+	if (In != NULL)
+	{
+		pIn = In->buf;
+		pInLen = In->BufUsed;
+	}
+	if ((pIn == NULL) || (Stream == NULL))
+		return;
+	
+	ExpectLen = (pInLen / 4) * 3;
+
+	if (Target->BufSize - Target->BufUsed < ExpectLen)
+	{
+		IncreaseBuf(Target, 1, Target->BufUsed + ExpectLen + 1);
+	}
+
+	ExpectLen = base64_decode_block(pIn, pInLen, Target->buf + Target->BufUsed, state);
+	Target->BufUsed += ExpectLen;
+	Target->buf[Target->BufUsed] = '\0';
+}
+
 /**
  * @ingroup StrBuf_DeEnCoder
  * @brief decode a buffer from base 64 encoding; destroys original
@@ -4186,15 +4259,23 @@ inline static void FDIOBufferFlush(FDIOBuffer *FDB)
 void FDIOBufferInit(FDIOBuffer *FDB, IOBuffer *IO, int FD, long TotalSendSize)
 {
 	FDIOBufferFlush(FDB);
-	FDB->ChunkSize = 
-		FDB->TotalSendSize = TotalSendSize;
+
+	FDB->TotalSendSize = TotalSendSize;
+	if (TotalSendSize > 0)
+		FDB->ChunkSize = TotalSendSize;
+	else
+	{
+		TotalSendSize = SIZ * 10;
+		FDB->ChunkSize = TotalSendSize;
+	}
 	FDB->IOB = IO;
+
 #ifdef LINUX_SPLICE
 	if (EnableSplice)
 		pipe(FDB->SplicePipe);
 	else
 #endif
-		FDB->ChunkBuffer = NewStrBufPlain(NULL, TotalSendSize + 1);
+		FDB->ChunkBuffer = NewStrBufPlain(NULL, TotalSendSize+ 1);
 
 	FDB->OtherFD = FD;
 }
@@ -4221,68 +4302,159 @@ void FDIOBufferDelete(FDIOBuffer *FDB)
 int FileSendChunked(FDIOBuffer *FDB, const char **Err)
 {
 	ssize_t sent, pipesize;
-#ifdef LINUX_SPLICE
-	if (EnableSplice)
+
+	if (FDB->TotalSendSize > 0)
 	{
-		if (FDB->PipeSize == 0)
+#ifdef LINUX_SPLICE
+		if (EnableSplice)
 		{
-			pipesize = splice(FDB->OtherFD,
-					  &FDB->TotalSentAlready, 
-					  FDB->SplicePipe[1],
-					  NULL, 
-					  FDB->ChunkSendRemain, 
-					  SPLICE_F_MOVE);
+			if (FDB->PipeSize == 0)
+			{
+				pipesize = splice(FDB->OtherFD,
+						  &FDB->TotalSentAlready, 
+						  FDB->SplicePipe[1],
+						  NULL, 
+						  FDB->ChunkSendRemain, 
+						  SPLICE_F_MOVE);
 	
-			if (pipesize == -1)
+				if (pipesize == -1)
+				{
+					*Err = strerror(errno);
+					return pipesize;
+				}
+				FDB->PipeSize = pipesize;
+			}
+			sent =  splice(FDB->SplicePipe[0],
+				       NULL, 
+				       FDB->IOB->fd,
+				       NULL, 
+				       FDB->PipeSize,
+				       SPLICE_F_MORE | SPLICE_F_MOVE | SPLICE_F_NONBLOCK);
+			if (sent == -1)
 			{
 				*Err = strerror(errno);
-				return pipesize;
+				return sent;
 			}
-			FDB->PipeSize = pipesize;
+			FDB->PipeSize -= sent;
+			FDB->ChunkSendRemain -= sent;
+			return sent;
 		}
-		sent =  splice(FDB->SplicePipe[0],
-			       NULL, 
-			       FDB->IOB->fd,
-			       NULL, 
-			       FDB->PipeSize,
-			       SPLICE_F_MORE | SPLICE_F_MOVE | SPLICE_F_NONBLOCK);
-		if (sent == -1)
+		else
+#endif
 		{
-			*Err = strerror(errno);
-			return sent;
+			char *pRead;
+			long nRead = 0;
+
+			pRead = FDB->ChunkBuffer->buf;
+			while ((FDB->ChunkBuffer->BufUsed < FDB->TotalSendSize) && (nRead >= 0))
+			{
+				nRead = read(FDB->OtherFD, pRead, FDB->TotalSendSize - FDB->ChunkBuffer->BufUsed);
+				if (nRead > 0) {
+					FDB->ChunkBuffer->BufUsed += nRead;
+					FDB->ChunkBuffer->buf[FDB->ChunkBuffer->BufUsed] = '\0';
+				}
+				else if (nRead == 0) {}
+				else return nRead;
+			}
+
+			nRead = write(FDB->IOB->fd,
+				      FDB->ChunkBuffer->buf     + FDB->TotalSentAlready,
+				      FDB->ChunkBuffer->BufUsed - FDB->TotalSentAlready);
+
+			if (nRead >= 0) {
+				FDB->TotalSentAlready += nRead;
+				FDB->ChunkSendRemain -= nRead;
+				return FDB->ChunkSendRemain;
+			}
+			else {
+				return nRead;
+			}
 		}
-		FDB->PipeSize -= sent;
-		FDB->ChunkSendRemain -= sent;
-		return sent;
 	}
 	else
-#endif
 	{
-		char *pRead;
-		long nRead = 0;
-
-		pRead = FDB->ChunkBuffer->buf;
-		while ((FDB->ChunkBuffer->BufUsed < FDB->TotalSendSize) && (nRead >= 0))
+#ifdef LINUX_SPLICE
+		if (EnableSplice)
 		{
-			nRead = read(FDB->OtherFD, pRead, FDB->TotalSendSize - FDB->ChunkBuffer->BufUsed);
-			if (nRead > 0) {
-				FDB->ChunkBuffer->BufUsed += nRead;
-				FDB->ChunkBuffer->buf[FDB->ChunkBuffer->BufUsed] = '\0';
+			if (FDB->PipeSize == 0)
+			{
+				pipesize = splice(FDB->OtherFD,
+						  &FDB->TotalSentAlready, 
+						  FDB->SplicePipe[1],
+						  NULL, 
+						  SIZ * 10, 
+						  SPLICE_F_MOVE);
+	
+				if (pipesize == -1)
+				{
+					*Err = strerror(errno);
+					return pipesize;
+				}
+				FDB->PipeSize = pipesize;
+				if (pipesize == 0)
+					return -1;
 			}
-			else if (nRead == 0) {}
-			else return nRead;
-		
+			sent =  splice(FDB->SplicePipe[0],
+				       NULL, 
+				       FDB->IOB->fd,
+				       NULL, 
+				       FDB->PipeSize,
+				       SPLICE_F_MORE | SPLICE_F_MOVE | SPLICE_F_NONBLOCK);
+			if (sent == -1)
+			{
+				*Err = strerror(errno);
+				return sent;
+			}
+			FDB->PipeSize -= sent;
+			FDB->ChunkSendRemain -= sent;
+			return sent;
 		}
+		else
+#endif
+		{
+			char *pRead;
+			long nRead = 0;
 
-		nRead = write(FDB->IOB->fd, FDB->ChunkBuffer->buf + FDB->TotalSentAlready, FDB->ChunkSendRemain);
+			pRead = FDB->ChunkBuffer->buf;
+			while ((FDB->ChunkSendRemain == 0) && 
+			       (FDB->ChunkBuffer->BufUsed < FDB->ChunkBuffer->BufSize) &&
+			       (nRead >= 0))
+			{
+				FDB->TotalSentAlready = 0;
+				nRead = read(FDB->OtherFD, pRead, FDB->ChunkBuffer->BufSize - FDB->ChunkBuffer->BufUsed);
+				if (nRead > 0) {
+					FDB->ChunkBuffer->BufUsed += nRead;
+					FDB->ChunkBuffer->buf[FDB->ChunkBuffer->BufUsed] = '\0';
+					FDB->ChunkSendRemain += nRead;
+				}
+				else if (nRead == 0)
+				{
+					return -1;
+				}
+				else
+				{
+					*Err = strerror(errno);
+					return nRead;
+				}
+			}
 
-		if (nRead >= 0) {
-			FDB->TotalSentAlready += nRead;
-			FDB->ChunkSendRemain -= nRead;
-			return FDB->ChunkSendRemain;
-		}
-		else {
-			return nRead;
+			nRead = write(FDB->IOB->fd,
+				      FDB->ChunkBuffer->buf     + FDB->TotalSentAlready,
+				      FDB->ChunkBuffer->BufUsed - FDB->TotalSentAlready);
+
+			if (nRead >= 0) {
+				FDB->TotalSentAlready += nRead;
+				FDB->ChunkSendRemain -= nRead;
+				if (FDB->ChunkSendRemain == 0)
+				{
+					FDB->ChunkBuffer->BufUsed = 0;
+					FDB->TotalSentAlready = 0;
+				}
+				return FDB->ChunkSendRemain;
+			}
+			else {
+				return nRead;
+			}
 		}
 	}
 }