ultimatepp/examples/cmdsrv/cmdsrv.cpp
rylek 985ac41440 Added DelayedWrite to server loop
git-svn-id: svn://ultimatepp.org/upp/trunk@1424 f0d560ea-af0d-0410-9eb7-867de7ffcac7
2009-07-17 12:03:14 +00:00

120 lines
3.2 KiB
C++

#include <Web/Web.h>
#ifdef PLATFORM_WIN32
#include <wincon.h>
#endif
using namespace Upp;
#define APP_VERSION "1.0.r2"
#define APP_DATE Date(2008, 9, 22)
bool service_stopped = false;
#ifdef PLATFORM_WIN32
static BOOL WINAPI EventHandler(DWORD CtrlType)
{
RLOG("EventHandler (CtrlType = " << (int)CtrlType << ")");
if(CtrlType == CTRL_LOGOFF_EVENT)
RLOG("LOGOFF_EVENT (ignored)");
else {
RLOG("exiting");
service_stopped = true;
}
return true;
}
void InstallSignalHandler()
{
SetConsoleCtrlHandler(&EventHandler, true);
}
#endif
#ifdef PLATFORM_POSIX
static void QuitSignal(int signal)
{
RLOG("Received signal " << signal << ", quitting...");
service_stopped = true;
}
void InstallSignalHandler()
{
signal(SIGPIPE, SIG_IGN);
signal(SIGINT, &QuitSignal);
signal(SIGTERM, &QuitSignal);
signal(SIGKILL, &QuitSignal);
}
#endif
CONSOLE_APP_MAIN
{
InstallSignalHandler();
HttpServer server;
int next_id = 0;
server.Logging();
Vector<String> cmd;
cmd <<= CommandLine();
int port = 1536;
if(cmd.GetCount() >= 1)
port = Nvl(ScanInt(cmd[0]), port);
if(!server.Open(port)) {
puts(Socket::GetErrorText());
return;
}
while(!service_stopped) {
server.DelayedWrite();
server.Wait(1000);
if(server.IsError()) {
puts("Server error: " + Socket::GetErrorText());
puts("Reopening socket...");
fflush(stdout);
if(!server.Reopen()) {
puts("Reopen on server socket failed: " + Socket::GetErrorText());
return;
}
}
fputs(NFormat("\rserver(%d) running at %`", port, GetSysTime()), stdout);
fflush(stdout);
One<HttpRequest> request = server.GetRequest();
if(!request)
continue;
HttpQuery query = request->GetQuery();
String header, body;
int items = query.GetInt("ITEMS", 0, 100000, 10);
String svr = query.GetString("SVR");
Htmls image_page;
int i;
image_page <<
"CMDSRV, version " << APP_VERSION << ", release date: " << Format(APP_DATE) << "<br>\n"
"Copyright &copy; 2005 Tomas Rylek<p>\n"
"CMDSRV parameters: &lt;port number, default = 1536&gt;<p>\n"
"Current parameters: <b>cmdsrv " << port << "</b><p>\n"
"<b>items</b> (<b>" << FormatInt(items) << "</b>) = number of output lines to generate (default = 10)<br>\n"
"<b>svr</b> (" << (IsNull(svr) ? "<i>empty</i>" : "<b>" + svr + "</b>")
<< ") = identifier for generated output lines (default = test)<br>\n"
"Example: <b>http://localhost";
String wisapi = GetFileName(query.GetString("$$WISAPI"));
String sample_query;
sample_query << "items=" << minmax(2 * items, 10, 100000) << "&amp;svr=A";
if(!IsNull(wisapi))
image_page << "/dir/<a href=\"" << wisapi << "?" << sample_query << "\">" << wisapi << "?";
else
image_page << ':' << port << "/<a href=\"any?" << sample_query << "\">" << "any?";
image_page << sample_query << "</a><p>\n";
for(i = 0; i < items; i++)
image_page << "Line #" << (i + 1) << ", svr=" << UrlEncode(svr) << "<br>\n";
image_page << "<p>\n"
"Current request:<br>\n";
for(i = 0; i < query.GetCount(); i++)
image_page << ToHtml(query.GetKey(i)) << "=<b>" << ToHtml(query.GetValue(i)) << "</b><br>\n";
header = HttpContentType(HttpTextHtml(CHARSET_WIN1252));
body = HtmlTitlePage("Simple text HTTP server", image_page);
request->Write(header, body);
request.Clear();
}
}