winamp/Src/replicant/jnetlib/headers.cpp
2024-09-24 14:54:57 +02:00

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;
}