|
|
@@ -0,0 +1,171 @@
|
|
|
+// Filename: fake_http_server.cxx
|
|
|
+// Created by: drose (10Dec02)
|
|
|
+//
|
|
|
+////////////////////////////////////////////////////////////////////
|
|
|
+//
|
|
|
+// PANDA 3D SOFTWARE
|
|
|
+// Copyright (c) 2001, Disney Enterprises, Inc. All rights reserved
|
|
|
+//
|
|
|
+// All use of this software is subject to the terms of the Panda 3d
|
|
|
+// Software license. You should have received a copy of this license
|
|
|
+// along with this source code; you will also find a current copy of
|
|
|
+// the license at http://www.panda3d.org/license.txt .
|
|
|
+//
|
|
|
+// To contact the maintainers of this program write to
|
|
|
+// [email protected] .
|
|
|
+//
|
|
|
+////////////////////////////////////////////////////////////////////
|
|
|
+
|
|
|
+#include <pandabase.h>
|
|
|
+
|
|
|
+#include "queuedConnectionManager.h"
|
|
|
+#include "queuedConnectionListener.h"
|
|
|
+#include "queuedConnectionReader.h"
|
|
|
+#include "connectionWriter.h"
|
|
|
+#include "netAddress.h"
|
|
|
+#include "connection.h"
|
|
|
+#include "netDatagram.h"
|
|
|
+#include "pmap.h"
|
|
|
+
|
|
|
+#include <ctype.h>
|
|
|
+
|
|
|
+QueuedConnectionManager cm;
|
|
|
+QueuedConnectionReader reader(&cm, 10);
|
|
|
+ConnectionWriter writer(&cm, 10);
|
|
|
+
|
|
|
+class ClientState {
|
|
|
+public:
|
|
|
+ ClientState(Connection *client);
|
|
|
+ void receive_data(const Datagram &data);
|
|
|
+ void receive_line(string line);
|
|
|
+
|
|
|
+ Connection *_client;
|
|
|
+ string _received;
|
|
|
+};
|
|
|
+
|
|
|
+ClientState::
|
|
|
+ClientState(Connection *client) {
|
|
|
+ _client = client;
|
|
|
+}
|
|
|
+
|
|
|
+void ClientState::
|
|
|
+receive_data(const Datagram &data) {
|
|
|
+ _received += data.get_message();
|
|
|
+ size_t next = 0;
|
|
|
+ size_t newline = _received.find('\n', next);
|
|
|
+ while (newline != string::npos) {
|
|
|
+ size_t last = next;
|
|
|
+ next = newline + 1;
|
|
|
+ if (newline > 0 && _received[newline - 1] == '\r') {
|
|
|
+ newline--;
|
|
|
+ }
|
|
|
+ receive_line(_received.substr(last, newline - last));
|
|
|
+ if (next < _received.size() && _received[next] == '\r') {
|
|
|
+ next++;
|
|
|
+ }
|
|
|
+ newline = _received.find('\n', next);
|
|
|
+ }
|
|
|
+ _received = _received.substr(next);
|
|
|
+}
|
|
|
+
|
|
|
+void ClientState::
|
|
|
+receive_line(string line) {
|
|
|
+ cerr << "received: " << line << "\n";
|
|
|
+ // trim trailing whitespace.
|
|
|
+ size_t size = line.size();
|
|
|
+ while (size > 0 && isspace(line[size - 1])) {
|
|
|
+ size--;
|
|
|
+ }
|
|
|
+ if (size != line.size()) {
|
|
|
+ line = line.substr(0, size);
|
|
|
+ }
|
|
|
+
|
|
|
+ /*
|
|
|
+ if (line.empty()) {
|
|
|
+ // Start to honor the request, as if we cared.
|
|
|
+ Datagram dg;
|
|
|
+ dg.append_data("HTTP/1.1 200 OK\r\n");
|
|
|
+ writer.send(dg, _client);
|
|
|
+ // Close the connection!
|
|
|
+ cm.close_connection(_client);
|
|
|
+ }
|
|
|
+ */
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+int
|
|
|
+main(int argc, char *argv[]) {
|
|
|
+ if (argc != 2) {
|
|
|
+ nout << "fake_http_server port\n";
|
|
|
+ exit(1);
|
|
|
+ }
|
|
|
+
|
|
|
+ int port = atoi(argv[1]);
|
|
|
+
|
|
|
+ PT(Connection) rendezvous = cm.open_TCP_server_rendezvous(port, 5);
|
|
|
+
|
|
|
+ if (rendezvous.is_null()) {
|
|
|
+ nout << "Cannot grab port " << port << ".\n";
|
|
|
+ exit(1);
|
|
|
+ }
|
|
|
+
|
|
|
+ nout << "Listening for connections on port " << port << "\n";
|
|
|
+
|
|
|
+ QueuedConnectionListener listener(&cm, 1);
|
|
|
+ listener.add_connection(rendezvous);
|
|
|
+
|
|
|
+ typedef pmap< PT(Connection), ClientState > Clients;
|
|
|
+ Clients clients;
|
|
|
+
|
|
|
+ reader.set_raw_mode(1);
|
|
|
+ writer.set_raw_mode(1);
|
|
|
+
|
|
|
+ bool shutdown = false;
|
|
|
+ while (!shutdown) {
|
|
|
+ // Check for new clients.
|
|
|
+ while (listener.new_connection_available()) {
|
|
|
+ PT(Connection) rv;
|
|
|
+ NetAddress address;
|
|
|
+ PT(Connection) new_connection;
|
|
|
+ if (listener.get_new_connection(rv, address, new_connection)) {
|
|
|
+ nout << "Got connection from " << address << "\n";
|
|
|
+ reader.add_connection(new_connection);
|
|
|
+ clients.insert(Clients::value_type(new_connection, ClientState(new_connection)));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // Check for reset clients.
|
|
|
+ while (cm.reset_connection_available()) {
|
|
|
+ PT(Connection) connection;
|
|
|
+ if (cm.get_reset_connection(connection)) {
|
|
|
+ nout << "Lost connection from "
|
|
|
+ << connection->get_address() << "\n";
|
|
|
+ clients.erase(connection);
|
|
|
+ cm.close_connection(connection);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // Process all available datagrams.
|
|
|
+ while (reader.data_available()) {
|
|
|
+ NetDatagram datagram;
|
|
|
+ if (reader.get_data(datagram)) {
|
|
|
+ PT(Connection) client = datagram.get_connection();
|
|
|
+ Clients::iterator ci = clients.find(client);
|
|
|
+ if (ci == clients.end()) {
|
|
|
+ nout << "Received data from unexpected client " << (void *)client
|
|
|
+ << "\n";
|
|
|
+ } else {
|
|
|
+ ClientState &state = (*ci).second;
|
|
|
+ state.receive_data(datagram);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return (0);
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|