X-Git-Url: https://code.citadel.org/?a=blobdiff_plain;f=libcitadel%2Flib%2Fstringbuf.c;h=4f381dd4813d0766bc4d9adeff105bc809568459;hb=1aa51c5b5260e04c08fd59b58815129db4f4bf32;hp=55daf6192c57b9127e1735b71a947ea9772804dd;hpb=f1e86bdcc55adbd5bc7af04ccf64af01674ea64c;p=citadel.git

diff --git a/libcitadel/lib/stringbuf.c b/libcitadel/lib/stringbuf.c
index 55daf6192..4f381dd48 100644
--- a/libcitadel/lib/stringbuf.c
+++ b/libcitadel/lib/stringbuf.c
@@ -1,3 +1,22 @@
+/*
+ * Copyright (c) 1987-2011 by the citadel.org team
+ *
+ * This program is open source software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+
+#define _GNU_SOURCE
 #include "sysdep.h"
 #include <ctype.h>
 #include <errno.h>
@@ -10,6 +29,7 @@
 #include <sys/types.h>
 #define SHOW_ME_VAPPEND_PRINTF
 #include <stdarg.h>
+
 #include "libcitadel.h"
 
 #ifdef HAVE_ICONV
@@ -20,6 +40,10 @@
 #include <execinfo.h>
 #endif
 
+#ifdef LINUX_SENDFILE
+#include <sys/sendfile.h>
+#endif
+
 #ifdef HAVE_ZLIB
 #include <zlib.h>
 int ZEXPORT compress_gzip(Bytef * dest, size_t * destLen,
@@ -1797,6 +1821,66 @@ void StrBufUrlescAppend(StrBuf *OutBuf, const StrBuf *In, const char *PlainIn)
 	*pt = '\0';
 }
 
+/** 
+ * @ingroup StrBuf_DeEnCoder
+ * @brief Escape a string for feeding out as a the username/password part of an URL while appending it to a Buffer
+ * @param OutBuf the output buffer
+ * @param In Buffer to encode
+ * @param PlainIn way in from plain old c strings
+ */
+void StrBufUrlescUPAppend(StrBuf *OutBuf, const StrBuf *In, const char *PlainIn)
+{
+	const char *pch, *pche;
+	char *pt, *pte;
+	int len;
+	
+	if (((In == NULL) && (PlainIn == NULL)) || (OutBuf == NULL) )
+		return;
+	if (PlainIn != NULL) {
+		len = strlen(PlainIn);
+		pch = PlainIn;
+		pche = pch + len;
+	}
+	else {
+		pch = In->buf;
+		pche = pch + In->BufUsed;
+		len = In->BufUsed;
+	}
+
+	if (len == 0) 
+		return;
+
+	pt = OutBuf->buf + OutBuf->BufUsed;
+	pte = OutBuf->buf + OutBuf->BufSize - 4; /**< we max append 3 chars at once plus the \0 */
+
+	while (pch < pche) {
+		if (pt >= pte) {
+			IncreaseBuf(OutBuf, 1, -1);
+			pte = OutBuf->buf + OutBuf->BufSize - 4; /**< we max append 3 chars at once plus the \0 */
+			pt = OutBuf->buf + OutBuf->BufUsed;
+		}
+
+		if((*pch >= 'a' && *pch <= 'z') ||
+		   (*pch >= 'A' && *pch <= 'Z') || /* A-Z */
+		   (*pch >= '0' && *pch <= ':') || /* 0-9 : */
+		   (*pch == '!') || (*pch == '_') || 
+		   (*pch == ',') || (*pch == '.'))
+		{
+			*(pt++) = *(pch++);
+			OutBuf->BufUsed++;
+		}			
+		else {
+			*pt = '%';
+			*(pt + 1) = HexList[(unsigned char)*pch][0];
+			*(pt + 2) = HexList[(unsigned char)*pch][1];
+			pt += 3;
+			OutBuf->BufUsed += 3;
+			pch ++;
+		}
+	}
+	*pt = '\0';
+}
+
 /** 
  * @ingroup StrBuf_DeEnCoder
  * @brief append a string in hex encoding to the buffer
@@ -2125,6 +2209,7 @@ long StrECMAEscAppend(StrBuf *Target, const StrBuf *Source, const char *PlainIn)
 	const char *aptr, *eiptr;
 	char *bptr, *eptr;
 	long len;
+	int IsUtf8Sequence;
 
 	if (((Source == NULL) && (PlainIn == NULL)) || (Target == NULL) )
 		return -1;
@@ -2663,6 +2748,8 @@ int StrBufRFC2047encode(StrBuf **target, const StrBuf *source)
 		    (ch  > 126) || 
 		    (ch ==  61) ||
 		    (ch == '=') ||
+                    (ch == '?') ||
+		    (ch == '_') ||
 		    (ch == '[') ||
 		    (ch == ']')   )
 		{
@@ -2670,7 +2757,10 @@ int StrBufRFC2047encode(StrBuf **target, const StrBuf *source)
 			(*target)->BufUsed += 3;
 		}
 		else {
-			(*target)->buf[(*target)->BufUsed] = ch;
+			if (ch == ' ')
+				(*target)->buf[(*target)->BufUsed] = '_';
+			else
+				(*target)->buf[(*target)->BufUsed] = ch;
 			(*target)->BufUsed++;
 		}
 	}
@@ -3315,7 +3405,7 @@ static inline int Ctdl_GetUtf8SequenceLength(const char *CharS, const char *Char
 	int n = 0;
         unsigned char test = (1<<7);
 
-	if ((*CharS & 0xC0) == 0) 
+	if ((*CharS & 0xC0) != 0xC0) 
 		return 1;
 
 	while ((n < 8) && 
@@ -3338,7 +3428,7 @@ static inline int Ctdl_GetUtf8SequenceLength(const char *CharS, const char *Char
 static inline int Ctdl_IsUtf8SequenceStart(const char Char)
 {
 /** 11??.???? indicates an UTF8 Sequence. */
-	return ((Char & 0xC0) != 0);
+	return ((Char & 0xC0) == 0xC0);
 }
 
 /**
@@ -3699,7 +3789,18 @@ eReadState StrBufChunkSipLine(StrBuf *LineBuf, IOBuffer *FB)
 			optr --;
 		if ((*(ptr - 1) != '\r') && (*(ptr - 1) != '\n')) {
 			LineBuf->BufUsed = optr - LineBuf->buf;
-			*optr = '\0';       
+			*optr = '\0';
+			if ((FB->ReadWritePointer != NULL) && 
+			    (FB->ReadWritePointer != FB->Buf->buf))
+			{
+				/* Ok, the client application read all the data 
+				   it was interested in so far. Since there is more to read, 
+				   we now shrink the buffer, and move the rest over.
+				*/
+				StrBufCutLeft(FB->Buf, 
+					      FB->ReadWritePointer - FB->Buf->buf);
+				FB->ReadWritePointer = FB->Buf->buf;
+			}
 			return eMustReadMore;
 		}
 	}
@@ -3739,6 +3840,203 @@ eReadState StrBufCheckBuffer(IOBuffer *FB)
 	return eReadSuccess;
 }
 
+long IOBufferStrLength(IOBuffer *FB)
+{
+	if (FB->ReadWritePointer == NULL)
+		return StrLength(FB->Buf);
+	
+	return StrLength(FB->Buf) - (FB->ReadWritePointer - FB->Buf->buf);
+}
+
+void FDIOBufferInit(FDIOBuffer *FDB, IOBuffer *IO, int FD, long TotalSendSize)
+{
+	memset(FDB, 0, sizeof(FDIOBuffer));
+	FDB->ChunkSize = 
+		FDB->TotalSendSize = TotalSendSize;
+	FDB->IOB = IO;
+#ifndef LINUX_SENDFILE
+	FDB->ChunkBuffer = NewStrBufPlain(NULL, TotalSendSize + 1);
+#else
+	pipe(FDB->SplicePipe);
+#endif
+	FDB->OtherFD = FD;
+}
+
+void FDIOBufferDelete(FDIOBuffer *FDB)
+{
+#ifndef LINUX_SENDFILE
+	FreeStrBuf(&FDB->ChunkBuffer);
+#else
+	close(FDB->SplicePipe[0]);
+	close(FDB->SplicePipe[1]);
+	
+#endif
+	close(FDB->OtherFD);
+	memset(FDB, 0, sizeof(FDIOBuffer));	
+}
+
+int FileSendChunked(FDIOBuffer *FDB, const char **Err)
+{
+	char *pRead;
+	long nRead = 0;
+	
+#ifdef LINUX_SENDFILE
+	ssize_t sent;
+	sent = sendfile(FDB->IOB->fd, FDB->OtherFD, &FDB->TotalSentAlready, FDB->ChunkSendRemain);
+	if (sent == -1)
+	{
+		*Err = strerror(errno);
+		return sent;
+	}
+	FDB->ChunkSendRemain -= sent;
+	FDB->TotalSentAlready += sent;
+	return FDB->ChunkSendRemain;
+#else
+	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->ChunkSendRemain);
+
+	if (nRead >= 0) {
+		FDB->TotalSentAlready += nRead;
+		FDB->ChunkSendRemain -= nRead;
+		return FDB->ChunkSendRemain;
+	}
+	else {
+		return nRead;
+	}
+#endif
+}
+
+int FileRecvChunked(FDIOBuffer *FDB, const char **Err)
+{
+	ssize_t sent, pipesize;
+
+#ifdef LINUX_SENDFILE
+
+	pipesize = splice(FDB->IOB->fd, NULL, 
+			  FDB->SplicePipe[1], NULL, 
+			  FDB->ChunkSendRemain, 
+			  SPLICE_F_MORE | SPLICE_F_MOVE|SPLICE_F_NONBLOCK);
+	if (pipesize == -1)
+	{
+		*Err = strerror(errno);
+		return pipesize;
+	}
+	
+	sent = splice(FDB->SplicePipe[0], NULL, 
+		      FDB->OtherFD, &FDB->TotalSentAlready, 
+		      pipesize, SPLICE_F_MORE | SPLICE_F_MOVE);
+	if (sent == -1)
+	{
+		*Err = strerror(errno);
+		return sent;
+	}
+	FDB->ChunkSendRemain -= sent;
+	return sent;
+#else
+	
+	sent = read(FDB->IOB->fd, FDB->ChunkBuffer->buf, FDB->ChunkSendRemain);
+	if (sent > 0) {
+		int nWritten = 0;
+		int rc; 
+		
+		FDB->ChunkBuffer->BufUsed = sent;
+
+		while (nWritten < FDB->ChunkBuffer->BufUsed) {
+			rc =  write(FDB->OtherFD, FDB->ChunkBuffer->buf + nWritten, FDB->ChunkBuffer->BufUsed - nWritten);
+			if (rc < 0) {
+				*Err = strerror(errno);
+				return rc;
+			}
+			nWritten += rc;
+
+		}
+		FDB->ChunkBuffer->BufUsed = 0;
+		FDB->TotalSentAlready += sent;
+		FDB->ChunkSendRemain -= sent;
+		return FDB->ChunkSendRemain;
+	}
+	else if (sent < 0) {
+		*Err = strerror(errno);
+		return sent;
+	}
+
+#endif
+	return 0;
+}
+
+eReadState WriteIOBAlreadyRead(FDIOBuffer *FDB, const char **Error)
+{
+	int IsNonBlock;
+	int fdflags;
+	long rlen;
+	long should_write;
+	int nSuccessLess = 0;
+	struct timeval tv;
+	fd_set rfds;
+
+	fdflags = fcntl(FDB->OtherFD, F_GETFL);
+	IsNonBlock = (fdflags & O_NONBLOCK) == O_NONBLOCK;
+
+	while ((FDB->IOB->ReadWritePointer - FDB->IOB->Buf->buf < FDB->IOB->Buf->BufUsed) &&
+	       (FDB->ChunkSendRemain > 0))
+	{
+		if (IsNonBlock){
+			tv.tv_sec = 1; /* selectresolution; */
+			tv.tv_usec = 0;
+			
+			FD_ZERO(&rfds);
+			FD_SET(FDB->OtherFD, &rfds);
+			if (select(FDB->OtherFD + 1, NULL, &rfds, NULL, &tv) == -1) {
+				*Error = strerror(errno);
+				return eReadFail;
+			}
+		}
+		if (IsNonBlock && !  FD_ISSET(FDB->OtherFD, &rfds)) {
+			nSuccessLess ++;
+			continue;
+		}
+
+		should_write = FDB->IOB->Buf->BufUsed - 
+			(FDB->IOB->ReadWritePointer - FDB->IOB->Buf->buf);
+		if (should_write > FDB->ChunkSendRemain)
+			should_write = FDB->ChunkSendRemain;
+
+		rlen = write(FDB->OtherFD, 
+			     FDB->IOB->ReadWritePointer, 
+			     should_write);
+		if (rlen < 1) {
+			*Error = strerror(errno);
+						
+			return eReadFail;
+		}
+		FDB->TotalSentAlready += rlen;
+		FDB->IOB->ReadWritePointer += rlen;
+		FDB->ChunkSendRemain -= rlen;
+	}
+	if (FDB->IOB->ReadWritePointer >= FDB->IOB->Buf->buf + FDB->IOB->Buf->BufUsed)
+	{
+		FlushStrBuf(FDB->IOB->Buf);
+		FDB->IOB->ReadWritePointer = NULL;
+	}
+
+	if (FDB->ChunkSendRemain == 0)
+		return eReadSuccess;
+	else 
+		return eMustReadMore;
+}
+
 /*******************************************************************************
  *           File I/O; Prefer buffered read since its faster!                  *
  *******************************************************************************/
@@ -4415,9 +4713,10 @@ void StrBufStripSlashes(StrBuf *Dir, int RemoveTrailingSlash)
 			b++; a++;
 		}
 	}
-	if ((RemoveTrailingSlash) && (*(b - 1) != '/')){
-		*b = '/';
-		b++;
+	if ((RemoveTrailingSlash) &&
+	    (b > Dir->buf) && 
+	    (*(b - 1) == '/')){
+		b--;
 	}
 	*b = '\0';
 	Dir->BufUsed = b - Dir->buf;