FILE* DaoStream_GetFile( DaoStream *self )
{
if( DaoType_ChildOf( self->ctype, dao_type_file_stream ) ){
DaoFileStream *stream = (DaoFileStream*) self;
return stream->file;
}else if( self->Write == DaoStdStream_WriteStdout || self->Write == DaoStdStream_WriteStderr ){
DaoStdStream *stream = (DaoStdStream*) self;
if( stream->redirect ){
return DaoStream_GetFile( stream->redirect );
}else if( self->Write == DaoStdStream_WriteStdout ){
return stdout;
}else if( self->Write == DaoStdStream_WriteStderr ){
return stderr;
}
}else if( self->Write == DaoStream_WriteStdout ){
return stdout;
}else if( self->Write == DaoStream_WriteStderr ){
return stderr;
}
return NULL;
}
static int SetCharColor( DaoStream *stream, int color, int RGB[3] )
{
int res = 0;
struct _CONSOLE_SCREEN_BUFFER_INFO info;
FILE *file = DaoStream_GetFile( stream );
HANDLE fd = INVALID_HANDLE_VALUE;
WORD attr;
if( file ) fd = (HANDLE)_get_osfhandle( _fileno( file ) );
if( fd == INVALID_HANDLE_VALUE ) fd = GetStdHandle( STD_OUTPUT_HANDLE );
if( !GetConsoleScreenBufferInfo( fd, &info ) ) return 255;
attr = info.wAttributes;
if( attr & RGB[2] ) res |= 1;
if( attr & RGB[1] ) res |= 2;
if( attr & RGB[0] ) res |= 4;
attr = attr & ~RGB[2] & ~RGB[1] & ~RGB[0];
if( color & 1 ) attr |= RGB[2];
if( color & 2 ) attr |= RGB[1];
if( color & 4 ) attr |= RGB[0];
if( !SetConsoleTextAttribute( fd, attr ) ) return 255;
return res;
}
static void PreparePostData( DaoProcess *proc, DaoMap *httpPOSTS, DaoMap *httpPOST, DaoMap *httpFILE )
{
DaoValue *vk = (DaoValue*) DaoProcess_NewMBString( proc, NULL, 0 );
DaoValue *vv = (DaoValue*) DaoProcess_NewMBString( proc, NULL, 0 );
DString *key = DaoString_Get( DaoValue_CastString( vk ) );
DString *value = DaoString_Get( DaoValue_CastString( vv ) );
DString *dynaBuffer = DString_New(1);
int i = 0;
int len = 0;
char buffer[ LOCAL_BUF_SIZE + 1 ];
char *last = buffer + (LOCAL_BUF_SIZE-1);
char *contentLength = getenv( "CONTENT_LENGTH" );
char *contentType = getenv( "CONTENT_TYPE" );
len = 0;
*last = 0;
if( contentLength != NULL ) len = strtol( contentLength, NULL, 10);
if( contentType != NULL ){
//printf( "CONTENT_TYPE = %s\n", contentType );
if( strstr( contentType, "multipart/form-data" ) == NULL ){
i = 0;
DString_Clear( dynaBuffer );
while( i < len ){
int n = 0;
int ch = getchar();
while( ch != EOF ){
buffer[n] = (char)ch;
n ++;
if( n == LOCAL_BUF_SIZE ) break;
ch = getchar();
}
buffer[ n ] = 0;
//printf( "%s|||||||||||||||||\n", buffer );
char *p = strchr( buffer, '&' );
if( p != NULL ){
*p = 0; // null-terminating
p++;
DString_AppendMBS( dynaBuffer, buffer );
ParseKeyValueString( proc, httpPOSTS, httpPOST, DString_GetMBS( dynaBuffer ) );
DString_Clear( dynaBuffer );
DString_AppendMBS( dynaBuffer, p );
}else{
DString_AppendMBS( dynaBuffer, buffer );
}
i += LOCAL_BUF_SIZE;
}
ParseKeyValueString( proc, httpPOSTS, httpPOST, DString_GetMBS( dynaBuffer ) );
}else{
char *boundary = strstr( contentType, "boundary" );
boundary = strstr( boundary, "=" ) + 1;
i = 0;
char *part = NULL;
while( ! feof( stdin ) ){
if( part == NULL ) part = fgets( buffer, LOCAL_BUF_SIZE, stdin );
if( part == NULL ) break;
if( strstr( part, boundary ) == NULL ) break;
// read content information
DString_Clear( dynaBuffer );
buffer[ LOCAL_BUF_SIZE ] = 0; // null-terminating
char *p = fgets( buffer, LOCAL_BUF_SIZE, stdin );
if( p == NULL ) break; // the last boundary scanned
p = strchr( p, '\n' );
*p = 0; // null-terminating
DString_AppendMBS( dynaBuffer, buffer );
char *info = (char*)DString_GetMBS( dynaBuffer );
info = strchr( info, '=' );
info += 2; // at char after name="
p = info;
while( *p != '\"' ) p ++;
*p = 0; // null-terminating
DString_SetMBS( key, info );
p ++;
if( (p = strstr(p,"filename") ) == NULL ){
p = fgets( buffer, LOCAL_BUF_SIZE, stdin );
p = fgets( buffer, LOCAL_BUF_SIZE, stdin );
// now real data:
DString_Clear( value );
while( p != NULL && strstr( p, boundary ) == NULL ){
char *t = strstr( p, "\r\n" );
if( t != NULL ) *t = 0;
DString_AppendMBS( value, buffer );
if( feof( stdin ) ) break;
p = fgets( buffer, LOCAL_BUF_SIZE, stdin );
t = strchr( p, '\n' );
if( t!= NULL ) *(t+1) = 0;
}
if( p != NULL ) part = p;
DaoMap_Insert( httpPOST, vk, vv );
}else{
DaoValue *vs = (DaoValue*) DaoProcess_NewStream( proc, tmpfile() );
FILE *file = DaoStream_GetFile( DaoValue_CastStream( vs ) );
char *t = NULL;
p = strchr( p, '\"' ) + 1;
info = p;
while( *p != '\"' ) p ++;
*p = 0; // null-terminating
//XXX stream->TYPER->SetName( stream, info );
DString_Clear( value );
// Content-Type ...\r\n
p = fgets( buffer, LOCAL_BUF_SIZE, stdin );
// \r\n
p = fgets( buffer, LOCAL_BUF_SIZE, stdin );
// data
#if 0
int count = fread( buffer, 1, LOCAL_BUF_SIZE, stdin );
while( count && strstr( buffer, boundary ) == NULL ){
fwrite( buffer, 1, count, file );
fprintf( file, "%s\n", "===========================" );
if( feof( stdin ) ) break;
count = fread( buffer, 1, LOCAL_BUF_SIZE, stdin );
}
#else
char tail[3] = { 0, 0, 0 };
int count, ntail = 0;
p = fgets( buffer, LOCAL_BUF_SIZE, stdin );
while( p != NULL && strstr( p, boundary ) == NULL ){
if( feof( stdin ) ){
// XXX
break;
}else{
t = p;
while( t != last && (*t) != '\n' ) t ++;
if( (*t) == '\n' ){
count = t-p+1;
if( count >= 2 ){
count -= 2;
if( ntail ) fwrite( tail, 1, ntail, file );
tail[0] = p[ count ];
tail[1] = p[ count+1 ];
ntail = 2;
fwrite( p, 1, count, file );
}else if( count == 1 ){
if( ntail == 2 ){
fwrite( tail, 1, 1, file );
tail[0] = tail[1];
tail[1] = p[0];
}else if( ntail ==1 ){
tail[1] = p[0];
ntail = 2;
}else{
tail[0] = p[0];
ntail = 1;
}
}
}else{
if( ntail ) fwrite( tail, 1, ntail, file );
count = LOCAL_BUF_SIZE-3;
tail[0] = p[ count ];
tail[1] = p[ count+1 ];
ntail = 2;
fwrite( p, 1, count, file );
}
}
p = fgets( buffer, LOCAL_BUF_SIZE, stdin );
}
#endif
//if( p != NULL ) part = p;
rewind( file );
DaoMap_Insert( httpFILE, vk, vs );
}
}
}
}
DString_Delete( dynaBuffer );
}
