private void OnClientConnect(IAsyncResult asyncResult) { try { HttpListener listener = (HttpListener)asyncResult.AsyncState; executorHasClient = true; // Here we complete/end the BeginGetContext() asynchronous call // by calling EndGetContext() - which returns the reference to // a new HttpListenerContext object. Then we can set up a new // thread to listen for the next connection. HttpListenerContext workerContext = listener.EndGetContext(asyncResult); IAsyncResult newResult = listener.BeginGetContext(new AsyncCallback(OnClientConnect), listener); ExtensionRequestPacket packet = new ExtensionRequestPacket(workerContext); // Console.WriteLine("ID {0} connected.", packet.ID); if (packet.PacketType == ChromeExtensionPacketType.Get) { // Console.WriteLine("Received GET request from from {0}", packet.ID); Send(packet, HostPageHtml, "text/html"); } else { // Console.WriteLine("Received from {0}:\n{1}", packet.ID, packet.Content); // pendingRequestQueue.Enqueue(packet); pendingRequestQueue.Add(packet); postRequestReceived.Set(); } // Console.WriteLine("ID {0} disconnected.", packet.ID); } catch (ObjectDisposedException) { System.Diagnostics.Debugger.Log(0, "1", "\n OnClientConnection: Socket has been closed\n"); } catch (SocketException se) { Console.WriteLine("ERROR:" + se.Message); } catch (HttpListenerException hle) { // When we shut the HttpListener down, there will always still be // a thread pending listening for a request. If there is no client // connected, we may have a real problem here. if (!executorHasClient) { Console.WriteLine(hle.Message); } } }
private void Send(ExtensionRequestPacket packet, string data, string sendAsContentType) { if (packet.PacketType == ChromeExtensionPacketType.Post) { // Console.WriteLine("Sending to {0}:\n{1}", packet.ID, data); // Reset the signal so that the processor will wait for another // POST message. postRequestReceived.Reset(); } byte[] byteData = Encoding.UTF8.GetBytes(data); HttpListenerResponse response = packet.Context.Response; response.KeepAlive = true; response.StatusCode = (int)HttpStatusCode.OK; response.StatusDescription = HttpStatusCode.OK.ToString(); response.ContentType = sendAsContentType; response.ContentLength64 = byteData.LongLength; response.Close(byteData, true); }
private void SendMessage(Command commandToExecute) { // Wait for a POST request to be pending from the Chrome extension. // When one is received, get the packet from the queue. bool signalReceived = postRequestReceived.WaitOne(TimeSpan.FromSeconds(ExtensionTimeoutInSeconds)); if (pendingRequestQueue.Count == 0) { throw new FatalChromeException("No pending requests from the extension in the queue"); } // ExtensionRequestPacket packet = pendingRequestQueue.Dequeue(); ExtensionRequestPacket packet = pendingRequestQueue[0]; pendingRequestQueue.RemoveAt(0); // Get the parameter names to correctly serialize the command to JSON. commandToExecute.Parameters.Add("request", commandNameMap[commandToExecute.Name]); string commandStringToSend = commandToExecute.ParametersAsJsonString; // Send the response to the Chrome extension. Send(packet, commandStringToSend, "application/json; charset=UTF-8"); }
private Response HandleResponse() { // Wait for a POST request to be pending from the Chrome extension. // Note that we need to leave the packet in the queue for the next // send message. postRequestReceived.WaitOne(TimeSpan.FromSeconds(ExtensionTimeoutInSeconds)); if (pendingRequestQueue.Count == 0) { throw new FatalChromeException("Expected a response from the extension, but none was found"); } // ExtensionRequestPacket packet = pendingRequestQueue.Peek(); ExtensionRequestPacket packet = pendingRequestQueue[0]; // Parse the packet content, and deserialize from a JSON object. string responseString = ParseResponse(packet.Content); Response response = new Response(); if (!string.IsNullOrEmpty(responseString)) { response = JsonConvert.DeserializeObject <Response>(responseString); if (response.Status == WebDriverResult.Success) { string valueAsString = response.Value as string; if (valueAsString != null) { // First, collapse all \r\n pairs to \n, then replace all \n with // System.Environment.NewLine. This ensures the consistency of // the values. response.Value = valueAsString.Replace("\r\n", "\n").Replace("\n", System.Environment.NewLine); } } } return(response); }
private void Send(ExtensionRequestPacket packet, string data, string sendAsContentType) { if (packet.PacketType == ChromeExtensionPacketType.Post) { // Console.WriteLine("Sending to {0}:\n{1}", packet.ID, data); // Reset the signal so that the processor will wait for another // POST message. postRequestReceived.Reset(); } byte[] byteData = Encoding.UTF8.GetBytes(data); HttpListenerResponse response = packet.Context.Response; response.KeepAlive = true; response.StatusCode = (int)HttpStatusCode.OK; response.StatusDescription = HttpStatusCode.OK.ToString(); response.ContentType = sendAsContentType; response.ContentLength64 = byteData.LongLength; response.Close(byteData, true); }
private void OnClientConnect(IAsyncResult asyncResult) { try { HttpListener listener = (HttpListener)asyncResult.AsyncState; executorHasClient = true; // Here we complete/end the BeginGetContext() asynchronous call // by calling EndGetContext() - which returns the reference to // a new HttpListenerContext object. Then we can set up a new // thread to listen for the next connection. HttpListenerContext workerContext = listener.EndGetContext(asyncResult); IAsyncResult newResult = listener.BeginGetContext(new AsyncCallback(OnClientConnect), listener); ExtensionRequestPacket packet = new ExtensionRequestPacket(workerContext); // Console.WriteLine("ID {0} connected.", packet.ID); if (packet.PacketType == ChromeExtensionPacketType.Get) { // Console.WriteLine("Received GET request from from {0}", packet.ID); Send(packet, HostPageHtml, "text/html"); } else { // Console.WriteLine("Received from {0}:\n{1}", packet.ID, packet.Content); // pendingRequestQueue.Enqueue(packet); pendingRequestQueue.Add(packet); postRequestReceived.Set(); } // Console.WriteLine("ID {0} disconnected.", packet.ID); } catch (ObjectDisposedException) { System.Diagnostics.Debugger.Log(0, "1", "\n OnClientConnection: Socket has been closed\n"); } catch (SocketException se) { Console.WriteLine("ERROR:" + se.Message); } catch (HttpListenerException hle) { // When we shut the HttpListener down, there will always still be // a thread pending listening for a request. If there is no client // connected, we may have a real problem here. if (!executorHasClient) { Console.WriteLine(hle.Message); } } }