Initial community commit

Src/replicant/jnetlib/headers.cpp

@@ -0,0 +1,119 @@
+#include "foundation/error.h"
+#include "foundation/types.h"
+
+#include "headers.h"
+#include "netinc.h"
+
+#include <stdlib.h>
+#include <string.h>
+
+JNL_Headers::JNL_Headers()
+{
+	m_recvheaders      = NULL;
+	m_recvheaders_size = 0;
+}
+
+JNL_Headers::~JNL_Headers()
+{
+	if ( m_recvheaders )
+		free( m_recvheaders );
+}
+
+void JNL_Headers::Reset()
+{
+	if ( m_recvheaders )
+		free( m_recvheaders );
+
+	m_recvheaders      = NULL;
+	m_recvheaders_size = 0;
+}
+
+const char *JNL_Headers::GetAllHeaders()
+{
+	// double null terminated, null delimited list
+	if ( m_recvheaders )
+		return m_recvheaders;
+	else
+		return "\0\0";
+}
+
+const char *JNL_Headers::GetHeader( const char *headername )
+{
+	char *ret = NULL;
+
+	if ( headername[ 0 ] == 0 || !m_recvheaders )
+		return NULL;
+
+	size_t headername_size = strlen( headername );
+	char *buf = (char *)malloc( headername_size + 2 );
+	strcpy( buf, headername );
+
+	if ( buf[ headername_size - 1 ] != ':' )
+	{
+		buf[ headername_size++ ] = ':';
+		buf[ headername_size ]   = 0;
+	}
+
+	char *p = m_recvheaders;
+	while ( p && *p )
+	{
+		if ( !strncasecmp( buf, p, headername_size ) )
+		{
+			ret = p + headername_size;
+			while ( ret && *ret && *ret == ' ' )
+				ret++;
+
+			break;
+		}
+
+		p += strlen( p ) + 1;
+	}
+
+	free( buf );
+
+	return ret;
+}
+
+int JNL_Headers::Add( const char *buf )
+{
+	if ( !m_recvheaders )
+	{
+		m_recvheaders_size = strlen( buf ) + 1;
+		if ( m_recvheaders_size == 0 || m_recvheaders_size == (size_t)-1 ) // check for overflow
+		{
+			return NErr_OutOfMemory;
+		}
+
+		m_recvheaders = (char *)malloc( m_recvheaders_size + 1 );
+		if ( m_recvheaders )
+		{
+			strcpy( m_recvheaders, buf ); // safe because we malloc'd specifically above
+			m_recvheaders[ m_recvheaders_size ] = 0;
+		}
+		else
+		{
+			return NErr_OutOfMemory;
+		}
+	}
+	else
+	{
+		size_t oldsize = m_recvheaders_size;
+		m_recvheaders_size += strlen( buf ) + 1;
+		if ( m_recvheaders_size + 1 < oldsize ) // check for overflow
+		{
+			return NErr_OutOfMemory;
+		}
+
+		char *n = (char *)realloc( m_recvheaders, m_recvheaders_size + 1 );
+		if ( !n )
+		{
+			return NErr_OutOfMemory;
+		}
+
+		strcpy( n + oldsize, buf ); // safe because we malloc specifially for the size
+		n[ m_recvheaders_size ] = 0; // double null terminate
+		m_recvheaders = n;
+	}
+
+	return NErr_Success;
+}