120 lines
2.2 KiB
C++
120 lines
2.2 KiB
C++
#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;
|
|
}
|