/*----------------------------------------------------------------------
| Test2
+---------------------------------------------------------------------*/
static bool
Test2(PLT_TaskManager* task_manager, const char* url, NPT_Size& size)
{
NPT_LOG_INFO("########### TEST 2 ######################");
/* reset output param first */
size = 0;
PLT_RingBufferStreamReference ringbuffer_stream(new PLT_RingBufferStream());
NPT_OutputStreamReference output_stream(ringbuffer_stream);
NPT_InputStreamReference input_stream(ringbuffer_stream);
PLT_Downloader downloader(task_manager, url, output_stream);
downloader.Start();
while (1) {
switch(downloader.GetState()) {
case PLT_DOWNLOADER_SUCCESS:
ringbuffer_stream->SetEos();
/* fallthrough */
case PLT_DOWNLOADER_DOWNLOADING: {
NPT_Size bytes_read;
NPT_Result res = ReadBody(downloader, input_stream, bytes_read);
if (NPT_FAILED(res)) {
return (res==NPT_ERROR_EOS)?true:false;
}
size += bytes_read;
}
break;
case PLT_DOWNLOADER_ERROR:
return false;
default:
NPT_System::Sleep(NPT_TimeInterval(0, 10000));
break;
}
};
return false;
}
/*----------------------------------------------------------------------
| main
+---------------------------------------------------------------------*/
int
main(int argc, char** argv)
{
NPT_COMPILER_UNUSED(argc);
NPT_HttpRequestHandler *handler, *custom_handler;
NPT_Reference<NPT_DataBuffer> buffer;
NPT_Size size;
bool result;
PLT_RingBufferStreamReference ringbuffer_stream(new PLT_RingBufferStream());
/* parse command line */
ParseCommandLine(argv);
/* create http server */
PLT_HttpServer http_server(Options.port?Options.port:80);
NPT_String url = "http://127.0.0.1:" + NPT_String::FromInteger(http_server.GetPort());
NPT_String custom_url = url;
if (!Options.path.IsEmpty()) {
/* extract folder path */
int index1 = Options.path.ReverseFind('\\');
int index2 = Options.path.ReverseFind('/');
if (index1 <= 0 && index2 <=0) {
fprintf(stderr, "ERROR: invalid path\n");
exit(1);
}
NPT_DirectoryEntryInfo info;
NPT_CHECK_SEVERE(NPT_DirectoryEntry::GetInfo(Options.path, &info));
/* add file request handler */
handler = new NPT_HttpFileRequestHandler(
Options.path.Left(index1>index2?index1:index2),
"/");
http_server.AddRequestHandler(handler, "/", true);
/* build url*/
url += "/" + Options.path.SubString((index1>index2?index1:index2)+1);
} else {
/* create random data */
buffer = new NPT_DataBuffer(32768);
buffer->SetDataSize(32768);
/* add static handler */
handler = new NPT_HttpStaticRequestHandler(buffer->GetData(),
buffer->GetDataSize(),
"text/xml");
http_server.AddRequestHandler(handler, "/test");
/* build url*/
url += "/test";
}
/* add custom handler */
NPT_InputStreamReference stream(ringbuffer_stream);
custom_handler = new PLT_HttpCustomRequestHandler(stream,
"text/xml");
http_server.AddRequestHandler(custom_handler, "/custom");
custom_url += "/custom";
/* start server */
NPT_CHECK_SEVERE(http_server.Start());
/* a task manager for the tests downloader */
PLT_TaskManager task_manager;
/* small delay to let the server start */
NPT_System::Sleep(NPT_TimeInterval(1, 0));
/* execute tests */
result = Test1(&task_manager, url.GetChars(), size);
if (!result) return -1;
result = Test2(&task_manager, url.GetChars(), size);
if (!result) return -1;
result = Test3(&task_manager, custom_url.GetChars(), ringbuffer_stream, size);
if (!result) return -1;
NPT_System::Sleep(NPT_TimeInterval(1, 0));
http_server.Stop();
delete handler;
delete custom_handler;
return 0;
}
/*----------------------------------------------------------------------
| main
+---------------------------------------------------------------------*/
int
main(int argc, char** argv)
{
NPT_COMPILER_UNUSED(argc);
NPT_HttpRequestHandler* handler;
NPT_Reference<NPT_DataBuffer> buffer;
bool result;
/* parse command line */
ParseCommandLine(argv);
/* create http server */
PLT_HttpServer http_server(Options.port?Options.port:8089);
NPT_String url;
if (!Options.path.IsEmpty()) {
/* extract folder path */
int index1 = Options.path.ReverseFind('\\');
int index2 = Options.path.ReverseFind('/');
if (index1 <= 0 && index2 <=0) {
fprintf(stderr, "ERROR: invalid path\n");
exit(1);
}
NPT_FileInfo info;
NPT_CHECK_SEVERE(NPT_File::GetInfo(Options.path, &info));
/* add file request handler */
handler = new NPT_HttpFileRequestHandler(
Options.path.Left(index1>index2?index1:index2),
"/");
http_server.AddRequestHandler(handler, "/", true);
/* build url */
url = "/" + Options.path.SubString((index1>index2?index1:index2)+1);
} else {
/* create random garbage data */
buffer = new NPT_DataBuffer(32768);
buffer->SetDataSize(32768);
/* add static handler */
handler = new NPT_HttpStaticRequestHandler(buffer->GetData(),
buffer->GetDataSize(),
"application/octet-stream");
http_server.AddRequestHandler(handler, "/test");
/* build url */
url = "/test";
}
/* add custom handler */
PLT_RingBufferStreamReference ringbuffer_stream(new PLT_RingBufferStream());
NPT_InputStreamReference stream(ringbuffer_stream);
NPT_HttpRequestHandler* custom_handler = new PLT_HttpCustomRequestHandler(stream, "text/xml");
http_server.AddRequestHandler(custom_handler, "/custom");
/* start server */
NPT_CHECK_SEVERE(http_server.Start());
/* a task manager for the tests downloader */
PLT_TaskManager task_manager;
/* small delay to let the server start */
NPT_System::Sleep(NPT_TimeInterval(1.f));
/* execute tests */
NPT_Size size;
NPT_COMPILER_UNUSED(size);
#ifdef TEST1
result = Test1(&task_manager, NPT_HttpUrl("127.0.0.1", http_server.GetPort(), url), size);
if (!result) return -1;
#endif
#ifdef TEST2
result = Test2(&task_manager, NPT_HttpUrl("127.0.0.1", http_server.GetPort(), url), size);
if (!result) return -1;
#endif
#ifdef TEST3
result = Test3(&task_manager, NPT_HttpUrl("127.0.0.1", http_server.GetPort(), "/custom"), ringbuffer_stream, size);
if (!result) return -1;
#endif
#ifdef TEST4
result = Test4(&task_manager, NPT_HttpUrl("127.0.0.1", http_server.GetPort(), "/custom"), NPT_TimeInterval(.1f));
if (!result) return -1;
result = Test4(&task_manager, NPT_HttpUrl("127.0.0.1", http_server.GetPort(), "/custom"), NPT_TimeInterval(1.f));
if (!result) return -1;
result = Test4(&task_manager, NPT_HttpUrl("127.0.0.1", http_server.GetPort(), "/custom"), NPT_TimeInterval(2.f));
if (!result) return -1;
#endif
#ifdef TEST5
result = Test5(NPT_HttpUrl("127.0.0.1", http_server.GetPort(), "/test"));
if (!result) return -1;
#endif
NPT_System::Sleep(NPT_TimeInterval(1.f));
// abort server tasks that are waiting on ring buffer stream
ringbuffer_stream->Abort();
http_server.Stop();
return 0;
}