netwerk/test/TestServ.cpp

   1 /* vim:set ts=4 sw=4 et cindent: */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #include "TestCommon.h"
   7 #include <stdlib.h>
   8 #include "nsIServiceManager.h"
   9 #include "nsIServerSocket.h"
  10 #include "nsISocketTransport.h"
  11 #include "nsNetUtil.h"
  12 #include "nsStringAPI.h"
  13 #include "nsCOMPtr.h"
  14 #include "prlog.h"
  15
  16 #if defined(PR_LOGGING)
  17 //
  18 // set NSPR_LOG_MODULES=Test:5
  19 //
  20 static PRLogModuleInfo *gTestLog = nullptr;
  21 #endif
  22 #define LOG(args) PR_LOG(gTestLog, PR_LOG_DEBUG, args)
  23
  24 class MySocketListener : public nsIServerSocketListener
  25 {
  26 protected:
  27     virtual ~MySocketListener() {}
  28
  29 public:
  30     NS_DECL_THREADSAFE_ISUPPORTS
  31     NS_DECL_NSISERVERSOCKETLISTENER
  32
  33     MySocketListener() {}
  34 };
  35
  36 NS_IMPL_ISUPPORTS(MySocketListener, nsIServerSocketListener)
  37
  38 NS_IMETHODIMP
  39 MySocketListener::OnSocketAccepted(nsIServerSocket *serv,
  40                                    nsISocketTransport *trans)
  41 {
  42     LOG(("MySocketListener::OnSocketAccepted [serv=%p trans=%p]\n", serv, trans));
  43
  44     nsAutoCString host;
  45     int32_t port;
  46
  47     trans->GetHost(host);
  48     trans->GetPort(&port);
  49
  50     LOG(("  -> %s:%d\n", host.get(), port));
  51
  52     nsCOMPtr<nsIInputStream> input;
  53     nsCOMPtr<nsIOutputStream> output;
  54     nsresult rv;
  55
  56     rv = trans->OpenInputStream(nsITransport::OPEN_BLOCKING, 0, 0, getter_AddRefs(input));
  57     if (NS_FAILED(rv))
  58         return rv;
  59
  60     rv = trans->OpenOutputStream(nsITransport::OPEN_BLOCKING, 0, 0, getter_AddRefs(output));
  61     if (NS_FAILED(rv))
  62         return rv;
  63
  64     char buf[256];
  65     uint32_t n;
  66
  67     rv = input->Read(buf, sizeof(buf), &n);
  68     if (NS_FAILED(rv))
  69         return rv;
  70
  71     const char response[] = "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\nFooooopy!!\r\n";
  72     rv = output->Write(response, sizeof(response) - 1, &n);
  73     if (NS_FAILED(rv))
  74         return rv;
  75
  76     input->Close();
  77     output->Close();
  78     return NS_OK;
  79 }
  80
  81 NS_IMETHODIMP
  82 MySocketListener::OnStopListening(nsIServerSocket *serv, nsresult status)
  83 {
  84     LOG(("MySocketListener::OnStopListening [serv=%p status=%x]\n", serv, status));
  85     QuitPumpingEvents();
  86     return NS_OK;
  87 }
  88
  89 static nsresult
  90 MakeServer(int32_t port)
  91 {
  92     nsresult rv;
  93     nsCOMPtr<nsIServerSocket> serv = do_CreateInstance(NS_SERVERSOCKET_CONTRACTID, &rv);
  94     if (NS_FAILED(rv))
  95         return rv;
  96
  97     rv = serv->Init(port, true, 5);
  98     if (NS_FAILED(rv))
  99         return rv;
 100
 101     rv = serv->GetPort(&port);
 102     if (NS_FAILED(rv))
 103         return rv;
 104     LOG(("  listening on port %d\n", port));
 105
 106     rv = serv->AsyncListen(new MySocketListener());
 107     return rv;
 108 }
 109
 110 int
 111 main(int argc, char* argv[])
 112 {
 113     if (test_common_init(&argc, &argv) != 0)
 114         return -1;
 115
 116     nsresult rv= (nsresult)-1;
 117     if (argc < 2) {
 118         printf("usage: %s <port>\n", argv[0]);
 119         return -1;
 120     }
 121
 122 #if defined(PR_LOGGING)
 123     gTestLog = PR_NewLogModule("Test");
 124 #endif
 125
 126     /*
 127      * The following code only deals with XPCOM registration stuff. and setting
 128      * up the event queues. Copied from TestSocketIO.cpp
 129      */
 130
 131     rv = NS_InitXPCOM2(nullptr, nullptr, nullptr);
 132     if (NS_FAILED(rv)) return -1;
 133
 134     {
 135         rv = MakeServer(atoi(argv[1]));
 136         if (NS_FAILED(rv)) {
 137             LOG(("MakeServer failed [rv=%x]\n", rv));
 138             return -1;
 139         }
 140
 141         // Enter the message pump to allow the URL load to proceed.
 142         PumpEvents();
 143     } // this scopes the nsCOMPtrs
 144     // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
 145     NS_ShutdownXPCOM(nullptr);
 146     return 0;
 147 }