Compare commits
	
		
			34 Commits
		
	
	
		
			v7.8.4
			...
			feature/ht
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | 73f4ba7ee7 | ||
|  | 6f39592c7b | ||
|  | 38200fc5d7 | ||
|  | 285c12775a | ||
|  | 6d56f7223a | ||
|  | 1db3568375 | ||
|  | 0a752e7d18 | ||
|  | 7c2bc2cf7e | ||
|  | 79f601ac65 | ||
|  | 069eccf415 | ||
|  | b563541b14 | ||
|  | 3bcd6f97a6 | ||
|  | c04bc3cdfc | ||
|  | 846f0c680a | ||
|  | c552a03ef0 | ||
|  | 0f175143c9 | ||
|  | 85569cb401 | ||
|  | bd854553d4 | ||
|  | 38c57e1ed2 | ||
|  | 26cc5025fb | ||
|  | 806cf39efc | ||
|  | daaa7ec704 | ||
|  | 3cffc6f9a5 | ||
|  | f8b1a03ee6 | ||
|  | a7ff3c41a1 | ||
|  | 78dbba5521 | ||
|  | b211bdbe38 | ||
|  | a0a53ab986 | ||
|  | 8d819053ff | ||
|  | e20ddc2a08 | ||
|  | c415ba9427 | ||
|  | 0b7c3ec235 | ||
|  | 29c96f287f | ||
|  | 2a17cad1bf | 
| @@ -20,6 +20,7 @@ set( IXWEBSOCKET_SOURCES | ||||
|     ixwebsocket/IXSocket.cpp | ||||
|     ixwebsocket/IXSocketServer.cpp | ||||
|     ixwebsocket/IXSocketConnect.cpp | ||||
|     ixwebsocket/IXSocketFactory.cpp | ||||
|     ixwebsocket/IXDNSLookup.cpp | ||||
|     ixwebsocket/IXCancellationRequest.cpp | ||||
|     ixwebsocket/IXWebSocket.cpp | ||||
| @@ -29,6 +30,9 @@ set( IXWEBSOCKET_SOURCES | ||||
|     ixwebsocket/IXWebSocketPerMessageDeflate.cpp | ||||
|     ixwebsocket/IXWebSocketPerMessageDeflateCodec.cpp | ||||
|     ixwebsocket/IXWebSocketPerMessageDeflateOptions.cpp | ||||
|     ixwebsocket/IXWebSocketHttpHeaders.cpp | ||||
|     ixwebsocket/IXHttpClient.cpp | ||||
|     ixwebsocket/IXUrlParser.cpp | ||||
| ) | ||||
|  | ||||
| set( IXWEBSOCKET_HEADERS | ||||
| @@ -36,6 +40,7 @@ set( IXWEBSOCKET_HEADERS | ||||
|     ixwebsocket/IXSocket.h | ||||
|     ixwebsocket/IXSocketServer.h | ||||
|     ixwebsocket/IXSocketConnect.h | ||||
|     ixwebsocket/IXSocketFactory.h | ||||
|     ixwebsocket/IXSetThreadName.h | ||||
|     ixwebsocket/IXDNSLookup.h | ||||
|     ixwebsocket/IXCancellationRequest.h | ||||
| @@ -51,6 +56,8 @@ set( IXWEBSOCKET_HEADERS | ||||
|     ixwebsocket/IXWebSocketPerMessageDeflateOptions.h | ||||
|     ixwebsocket/IXWebSocketHttpHeaders.h | ||||
|     ixwebsocket/libwshandshake.hpp | ||||
|     ixwebsocket/IXHttpClient.h | ||||
|     ixwebsocket/IXUrlParser.h | ||||
| ) | ||||
|  | ||||
| # Platform specific code | ||||
|   | ||||
							
								
								
									
										76
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										76
									
								
								README.md
									
									
									
									
									
								
							| @@ -5,7 +5,7 @@ | ||||
| ## Introduction | ||||
|  | ||||
| [*WebSocket*](https://en.wikipedia.org/wiki/WebSocket) is a computer communications protocol, providing full-duplex | ||||
| communication channels over a single TCP connection. *IXWebSocket* is a C++ library for client and server Websocket communication. The code is derived from [easywsclient](https://github.com/dhbaird/easywsclient) and from the [Satori C SDK](https://github.com/satori-com/satori-rtm-sdk-c). It has been tested on the following platforms. | ||||
| communication channels over a single TCP connection. *IXWebSocket* is a C++ library for client and server Websocket communication, and for client HTTP communication. The code is derived from [easywsclient](https://github.com/dhbaird/easywsclient) and from the [Satori C SDK](https://github.com/satori-com/satori-rtm-sdk-c). It has been tested on the following platforms. | ||||
|  | ||||
| * macOS | ||||
| * iOS | ||||
| @@ -15,7 +15,7 @@ communication channels over a single TCP connection. *IXWebSocket* is a C++ libr | ||||
|  | ||||
| ## Examples | ||||
|  | ||||
| The ws folder countains many interactive programs for chat and file transfers demonstrating client and server usage. | ||||
| The [*ws*](https://github.com/machinezone/IXWebSocket/tree/master/ws) folder countains many interactive programs for chat, [file transfers](https://github.com/machinezone/IXWebSocket/blob/master/ws/ws_send.cpp), [curl like](https://github.com/machinezone/IXWebSocket/blob/master/ws/ws_http_client.cpp) http clients, demonstrating client and server usage. | ||||
|  | ||||
| Here is what the client API looks like. | ||||
|  | ||||
| @@ -25,7 +25,7 @@ ix::WebSocket webSocket; | ||||
| std::string url("ws://localhost:8080/"); | ||||
| webSocket.setUrl(url); | ||||
|  | ||||
| // Optional heart beat, sent every 45 seconds when there isn't any traffic | ||||
| // Optional heart beat, sent every 45 seconds when there is not any traffic | ||||
| // to make sure that load balancers do not kill an idle connection. | ||||
| webSocket.setHeartBeatPeriod(45); | ||||
|  | ||||
| @@ -110,6 +110,68 @@ server.wait(); | ||||
|  | ||||
| ``` | ||||
|  | ||||
| Here is what the HTTP client API looks like. Note that HTTP client support is very recent and subject to changes. | ||||
|  | ||||
| ``` | ||||
| // | ||||
| // Preparation | ||||
| // | ||||
| HttpClient httpClient; | ||||
| HttpRequestArgs args; | ||||
|  | ||||
| // Custom headers can be set | ||||
| WebSocketHttpHeaders headers; | ||||
| headers["Foo"] = "bar"; | ||||
| args.extraHeaders = parseHeaders(headersData); | ||||
|  | ||||
| // Timeout options | ||||
| args.connectTimeout = connectTimeout; | ||||
| args.transferTimeout = transferTimeout; | ||||
|  | ||||
| // Redirect options | ||||
| args.followRedirects = followRedirects; | ||||
| args.maxRedirects = maxRedirects; | ||||
|  | ||||
| // Misc | ||||
| args.compress = compress; // Enable gzip compression | ||||
| args.verbose = verbose; | ||||
| args.logger = [](const std::string& msg) | ||||
| { | ||||
|     std::cout << msg; | ||||
| }; | ||||
|  | ||||
| // | ||||
| // Request | ||||
| // | ||||
| HttpResponse out; | ||||
| std::string url = "https://www.google.com"; | ||||
|  | ||||
| // HEAD request | ||||
| out = httpClient.head(url, args); | ||||
|  | ||||
| // GET request | ||||
| out = httpClient.get(url, args); | ||||
|  | ||||
| // POST request with parameters | ||||
| HttpParameters httpParameters; | ||||
| httpParameters["foo"] = "bar"; | ||||
| out = httpClient.post(url, httpParameters, args); | ||||
|  | ||||
| // POST request with a body | ||||
| out = httpClient.post(url, std::string("foo=bar"), args); | ||||
|  | ||||
| // | ||||
| // Result | ||||
| // | ||||
| auto statusCode = std::get<0>(out); | ||||
| auto errorCode = std::get<1>(out); | ||||
| auto responseHeaders = std::get<2>(out); | ||||
| auto payload = std::get<3>(out); | ||||
| auto errorMsg = std::get<4>(out); | ||||
| auto uploadSize = std::get<5>(out); | ||||
| auto downloadSize = std::get<6>(out); | ||||
| ``` | ||||
|  | ||||
| ## Build | ||||
|  | ||||
| CMakefiles for the library and the examples are available. This library has few dependencies, so it is possible to just add the source files into your project. | ||||
| @@ -142,11 +204,11 @@ Large frames are broken up into smaller chunks or messages to avoid filling up t | ||||
|  | ||||
| * There is no text support for sending data, only the binary protocol is supported. Sending json or text over the binary protocol works well. | ||||
| * Automatic reconnection works at the TCP socket level, and will detect remote end disconnects. However, if the device/computer network become unreachable (by turning off wifi), it is quite hard to reliably and timely detect it at the socket level using `recv` and `send` error codes. [Here](https://stackoverflow.com/questions/14782143/linux-socket-how-to-detect-disconnected-network-in-a-client-program) is a good discussion on the subject. This behavior is consistent with other runtimes such as node.js. One way to detect a disconnected device with low level C code is to do a name resolution with DNS but this can be expensive. Mobile devices have good and reliable API to do that. | ||||
| * The server code is using select to detect incoming data, and creates one OS thread per connection. This isn't as scalable as strategies using epoll or kqueue. | ||||
| * The server code is using select to detect incoming data, and creates one OS thread per connection. This is not as scalable as strategies using epoll or kqueue. | ||||
|  | ||||
| ## C++ code organization | ||||
|  | ||||
| Here's a simplistic diagram which explains how the code is structured in term of class/modules. | ||||
| Here is a simplistic diagram which explains how the code is structured in term of class/modules. | ||||
|  | ||||
| ``` | ||||
| +-----------------------+ --- Public | ||||
| @@ -198,7 +260,7 @@ If the connection was closed and sending failed, the return value will be set to | ||||
| 1. WebSocket_ReadyState_Connecting - The connection is not yet open. | ||||
| 2. WebSocket_ReadyState_Open       - The connection is open and ready to communicate. | ||||
| 3. WebSocket_ReadyState_Closing    - The connection is in the process of closing. | ||||
| 4. WebSocket_MessageType_Close     - The connection is closed or couldn't be opened. | ||||
| 4. WebSocket_MessageType_Close     - The connection is closed or could not be opened. | ||||
|  | ||||
| ### Open and Close notifications | ||||
|  | ||||
| @@ -308,7 +370,7 @@ websocket.ping("ping data, optional (empty string is ok): limited to 125 bytes l | ||||
| ### Heartbeat. | ||||
|  | ||||
| You can configure an optional heart beat / keep-alive, sent every 45 seconds | ||||
| when there isn't any traffic to make sure that load balancers do not kill an | ||||
| when there is not any traffic to make sure that load balancers do not kill an | ||||
| idle connection. | ||||
|  | ||||
| ``` | ||||
|   | ||||
							
								
								
									
										22
									
								
								examples/cobra_publisher/ixcrypto/IXHash.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								examples/cobra_publisher/ixcrypto/IXHash.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,22 @@ | ||||
| /* | ||||
|  *  IXHash.h | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2018 Machine Zone. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #include <string> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     uint64_t djb2Hash(const std::string& data) | ||||
|     { | ||||
|         uint64_t hashAddress = 5381; | ||||
|  | ||||
|         for (auto& c : data) | ||||
|         { | ||||
|             hashAddress = ((hashAddress << 5) + hashAddress) + c; | ||||
|         } | ||||
|  | ||||
|         return hashAddress; | ||||
|     } | ||||
| } | ||||
							
								
								
									
										15
									
								
								examples/cobra_publisher/ixcrypto/IXHash.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								examples/cobra_publisher/ixcrypto/IXHash.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,15 @@ | ||||
| /* | ||||
|  *  IXHash.h | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2018 Machine Zone. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| #include <string> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     uint64_t djb2Hash(const std::string& data); | ||||
| } | ||||
|  | ||||
							
								
								
									
										75
									
								
								examples/cobra_publisher/ixcrypto/IXUuid.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										75
									
								
								examples/cobra_publisher/ixcrypto/IXUuid.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,75 @@ | ||||
| /* | ||||
|  *  IXUuid.cpp | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2018 Machine Zone. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| /** | ||||
|  * Generate a random uuid similar to the uuid python module | ||||
|  * | ||||
|  * >>> import uuid | ||||
|  * >>> uuid.uuid4().hex | ||||
|  * 'bec08155b37d4050a1f3c3fa0276bf12' | ||||
|  * | ||||
|  * Code adapted from https://github.com/r-lyeh-archived/sole | ||||
|  */ | ||||
|  | ||||
| #include "IXUuid.h" | ||||
|  | ||||
| #include <sstream> | ||||
| #include <string> | ||||
| #include <iomanip> | ||||
| #include <random> | ||||
|  | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     class Uuid | ||||
|     { | ||||
|         public: | ||||
|             Uuid(); | ||||
|             std::string toString() const; | ||||
|  | ||||
|         private: | ||||
|             uint64_t _ab; | ||||
|             uint64_t _cd; | ||||
|     }; | ||||
|  | ||||
|     Uuid::Uuid() | ||||
|     { | ||||
|         static std::random_device rd; | ||||
|         static std::uniform_int_distribution<uint64_t> dist(0, (uint64_t)(~0)); | ||||
|  | ||||
|         _ab = dist(rd); | ||||
|         _cd = dist(rd); | ||||
|  | ||||
|         _ab = (_ab & 0xFFFFFFFFFFFF0FFFULL) | 0x0000000000004000ULL; | ||||
|         _cd = (_cd & 0x3FFFFFFFFFFFFFFFULL) | 0x8000000000000000ULL; | ||||
|     } | ||||
|  | ||||
|     std::string Uuid::toString() const | ||||
|     { | ||||
|         std::stringstream ss; | ||||
|         ss << std::hex << std::nouppercase << std::setfill('0'); | ||||
|  | ||||
|         uint32_t a = (_ab >> 32); | ||||
|         uint32_t b = (_ab & 0xFFFFFFFF); | ||||
|         uint32_t c = (_cd >> 32); | ||||
|         uint32_t d = (_cd & 0xFFFFFFFF); | ||||
|  | ||||
|         ss << std::setw(8) << (a); | ||||
|         ss << std::setw(4) << (b >> 16); | ||||
|         ss << std::setw(4) << (b & 0xFFFF); | ||||
|         ss << std::setw(4) << (c >> 16 ); | ||||
|         ss << std::setw(4) << (c & 0xFFFF); | ||||
|         ss << std::setw(8) << d; | ||||
|  | ||||
|         return ss.str(); | ||||
|     } | ||||
|  | ||||
|     std::string uuid4() | ||||
|     { | ||||
|         Uuid id; | ||||
|         return id.toString(); | ||||
|     } | ||||
| } | ||||
							
								
								
									
										17
									
								
								examples/cobra_publisher/ixcrypto/IXUuid.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								examples/cobra_publisher/ixcrypto/IXUuid.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,17 @@ | ||||
| /* | ||||
|  *  IXUuid.h | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2017 Machine Zone. All rights reserved. | ||||
|  */ | ||||
| #pragma once | ||||
|  | ||||
| #include <string> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|    /** | ||||
|     * Generate a random uuid | ||||
|     */ | ||||
|    std::string uuid4(); | ||||
|  | ||||
| } | ||||
							
								
								
									
										1
									
								
								examples/ws_receive/.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								examples/ws_receive/.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | ||||
| build | ||||
							
								
								
									
										30
									
								
								examples/ws_receive/CMakeLists.txt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										30
									
								
								examples/ws_receive/CMakeLists.txt
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,30 @@ | ||||
| # | ||||
| # Author: Benjamin Sergeant | ||||
| # Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
| # | ||||
|  | ||||
| cmake_minimum_required (VERSION 3.4.1) | ||||
| project (ws_receive) | ||||
|  | ||||
| # There's -Weverything too for clang | ||||
| set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -Wextra -pedantic -Wshorten-64-to-32") | ||||
|  | ||||
| set (CMAKE_CXX_STANDARD 14) | ||||
|  | ||||
| option(USE_TLS "Add TLS support" ON) | ||||
|  | ||||
| add_subdirectory(${PROJECT_SOURCE_DIR}/../.. ixwebsocket) | ||||
|  | ||||
| include_directories(ws_receive .) | ||||
|  | ||||
| add_executable(ws_receive  | ||||
|   jsoncpp/jsoncpp.cpp | ||||
|   ixcrypto/IXBase64.cpp | ||||
|   ixcrypto/IXHash.cpp | ||||
|   ws_receive.cpp) | ||||
|  | ||||
| if (APPLE AND USE_TLS) | ||||
|     target_link_libraries(ws_receive "-framework foundation" "-framework security") | ||||
| endif() | ||||
|  | ||||
| target_link_libraries(ws_receive ixwebsocket) | ||||
							
								
								
									
										1
									
								
								examples/ws_receive/README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								examples/ws_receive/README.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | ||||
| ws_receive is a simple server upload program. It needs to be used in conjonction with ws_send. | ||||
							
								
								
									
										1
									
								
								examples/ws_receive/ixcrypto
									
									
									
									
									
										Symbolic link
									
								
							
							
						
						
									
										1
									
								
								examples/ws_receive/ixcrypto
									
									
									
									
									
										Symbolic link
									
								
							| @@ -0,0 +1 @@ | ||||
| ../cobra_publisher/ixcrypto | ||||
							
								
								
									
										333
									
								
								examples/ws_receive/jsoncpp/json/json-forwards.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										333
									
								
								examples/ws_receive/jsoncpp/json/json-forwards.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,333 @@ | ||||
| /// Json-cpp amalgated forward header (http://jsoncpp.sourceforge.net/). | ||||
| /// It is intended to be used with #include "json/json-forwards.h" | ||||
| /// This header provides forward declaration for all JsonCpp types. | ||||
|  | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
| // Beginning of content of file: LICENSE | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
|  | ||||
| /* | ||||
| The JsonCpp library's source code, including accompanying documentation, | ||||
| tests and demonstration applications, are licensed under the following | ||||
| conditions... | ||||
|  | ||||
| Baptiste Lepilleur and The JsonCpp Authors explicitly disclaim copyright in all | ||||
| jurisdictions which recognize such a disclaimer. In such jurisdictions, | ||||
| this software is released into the Public Domain. | ||||
|  | ||||
| In jurisdictions which do not recognize Public Domain property (e.g. Germany as of | ||||
| 2010), this software is Copyright (c) 2007-2010 by Baptiste Lepilleur and | ||||
| The JsonCpp Authors, and is released under the terms of the MIT License (see below). | ||||
|  | ||||
| In jurisdictions which recognize Public Domain property, the user of this | ||||
| software may choose to accept it either as 1) Public Domain, 2) under the | ||||
| conditions of the MIT License (see below), or 3) under the terms of dual | ||||
| Public Domain/MIT License conditions described here, as they choose. | ||||
|  | ||||
| The MIT License is about as close to Public Domain as a license can get, and is | ||||
| described in clear, concise terms at: | ||||
|  | ||||
|    http://en.wikipedia.org/wiki/MIT_License | ||||
|  | ||||
| The full text of the MIT License follows: | ||||
|  | ||||
| ======================================================================== | ||||
| Copyright (c) 2007-2010 Baptiste Lepilleur and The JsonCpp Authors | ||||
|  | ||||
| Permission is hereby granted, free of charge, to any person | ||||
| obtaining a copy of this software and associated documentation | ||||
| files (the "Software"), to deal in the Software without | ||||
| restriction, including without limitation the rights to use, copy, | ||||
| modify, merge, publish, distribute, sublicense, and/or sell copies | ||||
| of the Software, and to permit persons to whom the Software is | ||||
| furnished to do so, subject to the following conditions: | ||||
|  | ||||
| The above copyright notice and this permission notice shall be | ||||
| included in all copies or substantial portions of the Software. | ||||
|  | ||||
| THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | ||||
| EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | ||||
| MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND | ||||
| NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS | ||||
| BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN | ||||
| ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN | ||||
| CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | ||||
| SOFTWARE. | ||||
| ======================================================================== | ||||
| (END LICENSE TEXT) | ||||
|  | ||||
| The MIT license is compatible with both the GPL and commercial | ||||
| software, affording one all of the rights of Public Domain with the | ||||
| minor nuisance of being required to keep the above copyright notice | ||||
| and license text in the source code. Note also that by accepting the | ||||
| Public Domain "license" you can re-license your copy using whatever | ||||
| license you like. | ||||
|  | ||||
| */ | ||||
|  | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
| // End of content of file: LICENSE | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
|  | ||||
|  | ||||
|  | ||||
|  | ||||
|  | ||||
| #ifndef JSON_FORWARD_AMALGATED_H_INCLUDED | ||||
| # define JSON_FORWARD_AMALGATED_H_INCLUDED | ||||
| /// If defined, indicates that the source file is amalgated | ||||
| /// to prevent private header inclusion. | ||||
| #define JSON_IS_AMALGAMATION | ||||
|  | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
| // Beginning of content of file: include/json/config.h | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
|  | ||||
| // Copyright 2007-2010 Baptiste Lepilleur and The JsonCpp Authors | ||||
| // Distributed under MIT license, or public domain if desired and | ||||
| // recognized in your jurisdiction. | ||||
| // See file LICENSE for detail or copy at http://jsoncpp.sourceforge.net/LICENSE | ||||
|  | ||||
| #ifndef JSON_CONFIG_H_INCLUDED | ||||
| #define JSON_CONFIG_H_INCLUDED | ||||
| #include <stddef.h> | ||||
| #include <string> //typedef String | ||||
| #include <stdint.h> //typedef int64_t, uint64_t | ||||
|  | ||||
| /// If defined, indicates that json library is embedded in CppTL library. | ||||
| //# define JSON_IN_CPPTL 1 | ||||
|  | ||||
| /// If defined, indicates that json may leverage CppTL library | ||||
| //#  define JSON_USE_CPPTL 1 | ||||
| /// If defined, indicates that cpptl vector based map should be used instead of | ||||
| /// std::map | ||||
| /// as Value container. | ||||
| //#  define JSON_USE_CPPTL_SMALLMAP 1 | ||||
|  | ||||
| // If non-zero, the library uses exceptions to report bad input instead of C | ||||
| // assertion macros. The default is to use exceptions. | ||||
| #ifndef JSON_USE_EXCEPTION | ||||
| #define JSON_USE_EXCEPTION 1 | ||||
| #endif | ||||
|  | ||||
| /// If defined, indicates that the source file is amalgated | ||||
| /// to prevent private header inclusion. | ||||
| /// Remarks: it is automatically defined in the generated amalgated header. | ||||
| // #define JSON_IS_AMALGAMATION | ||||
|  | ||||
| #ifdef JSON_IN_CPPTL | ||||
| #include <cpptl/config.h> | ||||
| #ifndef JSON_USE_CPPTL | ||||
| #define JSON_USE_CPPTL 1 | ||||
| #endif | ||||
| #endif | ||||
|  | ||||
| #ifdef JSON_IN_CPPTL | ||||
| #define JSON_API CPPTL_API | ||||
| #elif defined(JSON_DLL_BUILD) | ||||
| #if defined(_MSC_VER) || defined(__MINGW32__) | ||||
| #define JSON_API __declspec(dllexport) | ||||
| #define JSONCPP_DISABLE_DLL_INTERFACE_WARNING | ||||
| #endif // if defined(_MSC_VER) | ||||
| #elif defined(JSON_DLL) | ||||
| #if defined(_MSC_VER) || defined(__MINGW32__) | ||||
| #define JSON_API __declspec(dllimport) | ||||
| #define JSONCPP_DISABLE_DLL_INTERFACE_WARNING | ||||
| #endif // if defined(_MSC_VER) | ||||
| #endif // ifdef JSON_IN_CPPTL | ||||
| #if !defined(JSON_API) | ||||
| #define JSON_API | ||||
| #endif | ||||
|  | ||||
| // If JSON_NO_INT64 is defined, then Json only support C++ "int" type for | ||||
| // integer | ||||
| // Storages, and 64 bits integer support is disabled. | ||||
| // #define JSON_NO_INT64 1 | ||||
|  | ||||
| #if defined(_MSC_VER) // MSVC | ||||
| #  if _MSC_VER <= 1200 // MSVC 6 | ||||
|     // Microsoft Visual Studio 6 only support conversion from __int64 to double | ||||
|     // (no conversion from unsigned __int64). | ||||
| #    define JSON_USE_INT64_DOUBLE_CONVERSION 1 | ||||
|     // Disable warning 4786 for VS6 caused by STL (identifier was truncated to '255' | ||||
|     // characters in the debug information) | ||||
|     // All projects I've ever seen with VS6 were using this globally (not bothering | ||||
|     // with pragma push/pop). | ||||
| #    pragma warning(disable : 4786) | ||||
| #  endif // MSVC 6 | ||||
|  | ||||
| #  if _MSC_VER >= 1500 // MSVC 2008 | ||||
|     /// Indicates that the following function is deprecated. | ||||
| #    define JSONCPP_DEPRECATED(message) __declspec(deprecated(message)) | ||||
| #  endif | ||||
|  | ||||
| #endif // defined(_MSC_VER) | ||||
|  | ||||
| // In c++11 the override keyword allows you to explicity define that a function | ||||
| // is intended to override the base-class version.  This makes the code more | ||||
| // managable and fixes a set of common hard-to-find bugs. | ||||
| #if __cplusplus >= 201103L | ||||
| # define JSONCPP_OVERRIDE override | ||||
| # define JSONCPP_NOEXCEPT noexcept | ||||
| #elif defined(_MSC_VER) && _MSC_VER > 1600 && _MSC_VER < 1900 | ||||
| # define JSONCPP_OVERRIDE override | ||||
| # define JSONCPP_NOEXCEPT throw() | ||||
| #elif defined(_MSC_VER) && _MSC_VER >= 1900 | ||||
| # define JSONCPP_OVERRIDE override | ||||
| # define JSONCPP_NOEXCEPT noexcept | ||||
| #else | ||||
| # define JSONCPP_OVERRIDE | ||||
| # define JSONCPP_NOEXCEPT throw() | ||||
| #endif | ||||
|  | ||||
| #ifndef JSON_HAS_RVALUE_REFERENCES | ||||
|  | ||||
| #if defined(_MSC_VER) && _MSC_VER >= 1600 // MSVC >= 2010 | ||||
| #define JSON_HAS_RVALUE_REFERENCES 1 | ||||
| #endif // MSVC >= 2010 | ||||
|  | ||||
| #ifdef __clang__ | ||||
| #if __has_feature(cxx_rvalue_references) | ||||
| #define JSON_HAS_RVALUE_REFERENCES 1 | ||||
| #endif  // has_feature | ||||
|  | ||||
| #elif defined __GNUC__ // not clang (gcc comes later since clang emulates gcc) | ||||
| #if defined(__GXX_EXPERIMENTAL_CXX0X__) || (__cplusplus >= 201103L) | ||||
| #define JSON_HAS_RVALUE_REFERENCES 1 | ||||
| #endif  // GXX_EXPERIMENTAL | ||||
|  | ||||
| #endif // __clang__ || __GNUC__ | ||||
|  | ||||
| #endif // not defined JSON_HAS_RVALUE_REFERENCES | ||||
|  | ||||
| #ifndef JSON_HAS_RVALUE_REFERENCES | ||||
| #define JSON_HAS_RVALUE_REFERENCES 0 | ||||
| #endif | ||||
|  | ||||
| #ifdef __clang__ | ||||
| #  if __has_extension(attribute_deprecated_with_message) | ||||
| #    define JSONCPP_DEPRECATED(message)  __attribute__ ((deprecated(message))) | ||||
| #  endif | ||||
| #elif defined __GNUC__ // not clang (gcc comes later since clang emulates gcc) | ||||
| #  if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 5)) | ||||
| #    define JSONCPP_DEPRECATED(message)  __attribute__ ((deprecated(message))) | ||||
| #  elif (__GNUC__ > 3 || (__GNUC__ == 3 && __GNUC_MINOR__ >= 1)) | ||||
| #    define JSONCPP_DEPRECATED(message)  __attribute__((__deprecated__)) | ||||
| #  endif  // GNUC version | ||||
| #endif // __clang__ || __GNUC__ | ||||
|  | ||||
| #if !defined(JSONCPP_DEPRECATED) | ||||
| #define JSONCPP_DEPRECATED(message) | ||||
| #endif // if !defined(JSONCPP_DEPRECATED) | ||||
|  | ||||
| #if __GNUC__ >= 6 | ||||
| #  define JSON_USE_INT64_DOUBLE_CONVERSION 1 | ||||
| #endif | ||||
|  | ||||
| #if !defined(JSON_IS_AMALGAMATION) | ||||
|  | ||||
| # include "version.h" | ||||
|  | ||||
| # if JSONCPP_USING_SECURE_MEMORY | ||||
| #  include "allocator.h" //typedef Allocator | ||||
| # endif | ||||
|  | ||||
| #endif // if !defined(JSON_IS_AMALGAMATION) | ||||
|  | ||||
| namespace Json { | ||||
| typedef int Int; | ||||
| typedef unsigned int UInt; | ||||
| #if defined(JSON_NO_INT64) | ||||
| typedef int LargestInt; | ||||
| typedef unsigned int LargestUInt; | ||||
| #undef JSON_HAS_INT64 | ||||
| #else                 // if defined(JSON_NO_INT64) | ||||
| // For Microsoft Visual use specific types as long long is not supported | ||||
| #if defined(_MSC_VER) // Microsoft Visual Studio | ||||
| typedef __int64 Int64; | ||||
| typedef unsigned __int64 UInt64; | ||||
| #else                 // if defined(_MSC_VER) // Other platforms, use long long | ||||
| typedef int64_t Int64; | ||||
| typedef uint64_t UInt64; | ||||
| #endif // if defined(_MSC_VER) | ||||
| typedef Int64 LargestInt; | ||||
| typedef UInt64 LargestUInt; | ||||
| #define JSON_HAS_INT64 | ||||
| #endif // if defined(JSON_NO_INT64) | ||||
| #if JSONCPP_USING_SECURE_MEMORY | ||||
| #define JSONCPP_STRING        std::basic_string<char, std::char_traits<char>, Json::SecureAllocator<char> > | ||||
| #define JSONCPP_OSTRINGSTREAM std::basic_ostringstream<char, std::char_traits<char>, Json::SecureAllocator<char> > | ||||
| #define JSONCPP_OSTREAM       std::basic_ostream<char, std::char_traits<char>> | ||||
| #define JSONCPP_ISTRINGSTREAM std::basic_istringstream<char, std::char_traits<char>, Json::SecureAllocator<char> > | ||||
| #define JSONCPP_ISTREAM       std::istream | ||||
| #else | ||||
| #define JSONCPP_STRING        std::string | ||||
| #define JSONCPP_OSTRINGSTREAM std::ostringstream | ||||
| #define JSONCPP_OSTREAM       std::ostream | ||||
| #define JSONCPP_ISTRINGSTREAM std::istringstream | ||||
| #define JSONCPP_ISTREAM       std::istream | ||||
| #endif // if JSONCPP_USING_SECURE_MEMORY | ||||
| } // end namespace Json | ||||
|  | ||||
| #endif // JSON_CONFIG_H_INCLUDED | ||||
|  | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
| // End of content of file: include/json/config.h | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
|  | ||||
|  | ||||
|  | ||||
|  | ||||
|  | ||||
|  | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
| // Beginning of content of file: include/json/forwards.h | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
|  | ||||
| // Copyright 2007-2010 Baptiste Lepilleur and The JsonCpp Authors | ||||
| // Distributed under MIT license, or public domain if desired and | ||||
| // recognized in your jurisdiction. | ||||
| // See file LICENSE for detail or copy at http://jsoncpp.sourceforge.net/LICENSE | ||||
|  | ||||
| #ifndef JSON_FORWARDS_H_INCLUDED | ||||
| #define JSON_FORWARDS_H_INCLUDED | ||||
|  | ||||
| #if !defined(JSON_IS_AMALGAMATION) | ||||
| #include "config.h" | ||||
| #endif // if !defined(JSON_IS_AMALGAMATION) | ||||
|  | ||||
| namespace Json { | ||||
|  | ||||
| // writer.h | ||||
| class FastWriter; | ||||
| class StyledWriter; | ||||
|  | ||||
| // reader.h | ||||
| class Reader; | ||||
|  | ||||
| // features.h | ||||
| class Features; | ||||
|  | ||||
| // value.h | ||||
| typedef unsigned int ArrayIndex; | ||||
| class StaticString; | ||||
| class Path; | ||||
| class PathArgument; | ||||
| class Value; | ||||
| class ValueIteratorBase; | ||||
| class ValueIterator; | ||||
| class ValueConstIterator; | ||||
|  | ||||
| } // namespace Json | ||||
|  | ||||
| #endif // JSON_FORWARDS_H_INCLUDED | ||||
|  | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
| // End of content of file: include/json/forwards.h | ||||
| // ////////////////////////////////////////////////////////////////////// | ||||
|  | ||||
|  | ||||
|  | ||||
|  | ||||
|  | ||||
| #endif //ifndef JSON_FORWARD_AMALGATED_H_INCLUDED | ||||
							
								
								
									
										2186
									
								
								examples/ws_receive/jsoncpp/json/json.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										2186
									
								
								examples/ws_receive/jsoncpp/json/json.h
									
									
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										5386
									
								
								examples/ws_receive/jsoncpp/jsoncpp.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										5386
									
								
								examples/ws_receive/jsoncpp/jsoncpp.cpp
									
									
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										29
									
								
								examples/ws_receive/package-lock.json
									
									
									
										generated
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								examples/ws_receive/package-lock.json
									
									
									
										generated
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,29 @@ | ||||
| { | ||||
|   "requires": true, | ||||
|   "lockfileVersion": 1, | ||||
|   "dependencies": { | ||||
|     "async-limiter": { | ||||
|       "version": "1.0.0", | ||||
|       "resolved": "https://registry.npmjs.org/async-limiter/-/async-limiter-1.0.0.tgz", | ||||
|       "integrity": "sha512-jp/uFnooOiO+L211eZOoSyzpOITMXx1rBITauYykG3BRYPu8h0UcxsPNB04RR5vo4Tyz3+ay17tR6JVf9qzYWg==" | ||||
|     }, | ||||
|     "base-64": { | ||||
|       "version": "0.1.0", | ||||
|       "resolved": "https://registry.npmjs.org/base-64/-/base-64-0.1.0.tgz", | ||||
|       "integrity": "sha1-eAqZyE59YAJgNhURxId2E78k9rs=" | ||||
|     }, | ||||
|     "djb2": { | ||||
|       "version": "0.0.2", | ||||
|       "resolved": "https://registry.npmjs.org/djb2/-/djb2-0.0.2.tgz", | ||||
|       "integrity": "sha1-RAs4kao6uBQrVNRpsXe66p6W5O8=" | ||||
|     }, | ||||
|     "ws": { | ||||
|       "version": "6.1.4", | ||||
|       "resolved": "https://registry.npmjs.org/ws/-/ws-6.1.4.tgz", | ||||
|       "integrity": "sha512-eqZfL+NE/YQc1/ZynhojeV8q+H050oR8AZ2uIev7RU10svA9ZnJUddHcOUZTJLinZ9yEfdA2kSATS2qZK5fhJA==", | ||||
|       "requires": { | ||||
|         "async-limiter": "1.0.0" | ||||
|       } | ||||
|     } | ||||
|   } | ||||
| } | ||||
							
								
								
									
										153
									
								
								examples/ws_receive/ws_receive.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										153
									
								
								examples/ws_receive/ws_receive.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,153 @@ | ||||
| /* | ||||
|  *  ws_receive.cpp | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2018 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #include <iostream> | ||||
| #include <sstream> | ||||
| #include <fstream> | ||||
| #include <ixwebsocket/IXWebSocketServer.h> | ||||
| #include <jsoncpp/json/json.h> | ||||
| #include <ixcrypto/IXBase64.h> | ||||
| #include <ixcrypto/IXHash.h> | ||||
|  | ||||
|  | ||||
| namespace | ||||
| { | ||||
|     // We should cleanup the file name and full path further to remove .. as well | ||||
|     std::string extractFilename(const std::string& path) | ||||
|     { | ||||
|         std::string filename("filename.conf"); | ||||
|         std::string::size_type idx; | ||||
|  | ||||
|         idx = path.rfind('/'); | ||||
|         if (idx != std::string::npos) | ||||
|         { | ||||
|             std::string filename = path.substr(idx+1); | ||||
|             return filename; | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|             return std::string(); | ||||
|         } | ||||
|     } | ||||
| } | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     void errorHandler(const std::string& errMsg, | ||||
|                       const std::string& id, | ||||
|                       std::shared_ptr<ix::WebSocket> webSocket) | ||||
|     { | ||||
|         Json::Value pdu; | ||||
|         pdu["kind"] = "error"; | ||||
|         pdu["id"] = id; | ||||
|         pdu["message"] = errMsg; | ||||
|         webSocket->send(pdu.toStyledString()); | ||||
|     } | ||||
|  | ||||
|     void messageHandler(const std::string& str, | ||||
|                         std::shared_ptr<ix::WebSocket> webSocket) | ||||
|     { | ||||
|         std::cerr << "Received message: " << str.size() << std::endl; | ||||
|  | ||||
|         Json::Value data; | ||||
|         Json::Reader reader; | ||||
|         if (!reader.parse(str, data)) | ||||
|         { | ||||
|             errorHandler("Invalid JSON", std::string(), webSocket); | ||||
|             return; | ||||
|         } | ||||
|  | ||||
|         std::cout << "id: " << data["id"].asString() << std::endl; | ||||
|  | ||||
|         std::string content = ix::base64_decode(data["content"].asString()); | ||||
|         std::cout << "Content size: " << content.size() << std::endl; | ||||
|  | ||||
|         // Validate checksum | ||||
|         uint64_t cksum = ix::djb2Hash(data["content"].asString()); | ||||
|         uint64_t cksumRef = data["djb2_hash"].asUInt64(); | ||||
|  | ||||
|         std::cout << "Computed hash: " << cksum << std::endl; | ||||
|         std::cout << "Reference hash: " << cksumRef << std::endl; | ||||
|  | ||||
|         if (cksum != cksumRef) | ||||
|         { | ||||
|             errorHandler("Hash mismatch.", std::string(), webSocket); | ||||
|             return; | ||||
|         } | ||||
|  | ||||
|         std::string filename = data["filename"].asString(); | ||||
|         filename = extractFilename(filename); | ||||
|  | ||||
|         std::ofstream out(filename); | ||||
|         out << content; | ||||
|         out.close(); | ||||
|  | ||||
|         Json::Value pdu; | ||||
|         pdu["ack"] = true; | ||||
|         pdu["id"] = data["id"]; | ||||
|         pdu["filename"] = data["filename"]; | ||||
|         webSocket->send(pdu.toStyledString()); | ||||
|     } | ||||
| } | ||||
|  | ||||
| int main(int argc, char** argv) | ||||
| { | ||||
|     int port = 8080; | ||||
|     if (argc == 2) | ||||
|     { | ||||
|         std::stringstream ss; | ||||
|         ss << argv[1]; | ||||
|         ss >> port; | ||||
|     } | ||||
|  | ||||
|     ix::WebSocketServer server(port); | ||||
|  | ||||
|     server.setOnConnectionCallback( | ||||
|         [&server](std::shared_ptr<ix::WebSocket> webSocket) | ||||
|         { | ||||
|             webSocket->setOnMessageCallback( | ||||
|                 [webSocket, &server](ix::WebSocketMessageType messageType, | ||||
|                                      const std::string& str, | ||||
|                                      size_t wireSize, | ||||
|                                      const ix::WebSocketErrorInfo& error, | ||||
|                                      const ix::WebSocketOpenInfo& openInfo, | ||||
|                                      const ix::WebSocketCloseInfo& closeInfo) | ||||
|                 { | ||||
|                     if (messageType == ix::WebSocket_MessageType_Open) | ||||
|                     { | ||||
|                         std::cerr << "New connection" << std::endl; | ||||
|                         std::cerr << "Uri: " << openInfo.uri << std::endl; | ||||
|                         std::cerr << "Headers:" << std::endl; | ||||
|                         for (auto it : openInfo.headers) | ||||
|                         { | ||||
|                             std::cerr << it.first << ": " << it.second << std::endl; | ||||
|                         } | ||||
|                     } | ||||
|                     else if (messageType == ix::WebSocket_MessageType_Close) | ||||
|                     { | ||||
|                         std::cerr << "Closed connection" << std::endl; | ||||
|                     } | ||||
|                     else if (messageType == ix::WebSocket_MessageType_Message) | ||||
|                     { | ||||
|                         messageHandler(str, webSocket); | ||||
|                     } | ||||
|                 } | ||||
|             ); | ||||
|         } | ||||
|     ); | ||||
|  | ||||
|     auto res = server.listen(); | ||||
|     if (!res.first) | ||||
|     { | ||||
|         std::cerr << res.second << std::endl; | ||||
|         return 1; | ||||
|     } | ||||
|  | ||||
|     server.start(); | ||||
|     server.wait(); | ||||
|  | ||||
|     return 0; | ||||
| } | ||||
							
								
								
									
										43
									
								
								examples/ws_receive/ws_receive.js
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										43
									
								
								examples/ws_receive/ws_receive.js
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,43 @@ | ||||
| /* | ||||
|  *  ws_receive.js | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
| const WebSocket = require('ws') | ||||
| const djb2 = require('djb2') | ||||
| const fs = require('fs') | ||||
|  | ||||
| const wss = new WebSocket.Server({ port: 8080, | ||||
|                                    perMessageDeflate: false, | ||||
|                                    maxPayload: 1024 * 1024 * 1024 * 1024}); | ||||
|  | ||||
| wss.on('connection', function connection(ws) { | ||||
|   ws.on('message', function incoming(data) { | ||||
|     console.log('Received message') | ||||
|  | ||||
|     let str = data.toString() | ||||
|     let obj = JSON.parse(str) | ||||
|  | ||||
|     console.log(obj.id) | ||||
|     console.log(obj.djb2_hash) | ||||
|     console.log(djb2(obj.content)) | ||||
|  | ||||
|     var content = Buffer.from(obj.content, 'base64') | ||||
|     // let bytes = base64.decode(obj.content) | ||||
|  | ||||
|     let path = obj.filename | ||||
|     fs.writeFile(path, content, function(err) { | ||||
|       if (err) { | ||||
|         throw err | ||||
|       } else { | ||||
|         console.log('wrote data to disk') | ||||
|       } | ||||
|     }); | ||||
|  | ||||
|     let response = { | ||||
|       id: obj.id | ||||
|     } | ||||
|  | ||||
|     ws.send(JSON.stringify(response)) | ||||
|   }); | ||||
| }); | ||||
							
								
								
									
										1
									
								
								examples/ws_send/.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								examples/ws_send/.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | ||||
| build | ||||
							
								
								
									
										31
									
								
								examples/ws_send/CMakeLists.txt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								examples/ws_send/CMakeLists.txt
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | ||||
| # | ||||
| # Author: Benjamin Sergeant | ||||
| # Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
| # | ||||
|  | ||||
| cmake_minimum_required (VERSION 3.4.1) | ||||
| project (ws_send) | ||||
|  | ||||
| # There's -Weverything too for clang | ||||
| set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -Wextra -pedantic -Wshorten-64-to-32") | ||||
|  | ||||
| set (CMAKE_CXX_STANDARD 14) | ||||
|  | ||||
| option(USE_TLS "Add TLS support" ON) | ||||
|  | ||||
| add_subdirectory(${PROJECT_SOURCE_DIR}/../.. ixwebsocket) | ||||
|  | ||||
| include_directories(ws_send .) | ||||
|  | ||||
| add_executable(ws_send  | ||||
|   jsoncpp/jsoncpp.cpp | ||||
|   ixcrypto/IXBase64.cpp | ||||
|   ixcrypto/IXUuid.cpp | ||||
|   ixcrypto/IXHash.cpp | ||||
|   ws_send.cpp) | ||||
|  | ||||
| if (APPLE AND USE_TLS) | ||||
|     target_link_libraries(ws_send "-framework foundation" "-framework security") | ||||
| endif() | ||||
|  | ||||
| target_link_libraries(ws_send ixwebsocket) | ||||
							
								
								
									
										1
									
								
								examples/ws_send/README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								examples/ws_send/README.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | ||||
| ws_send is a simple upload program. It needs to be used in conjonction with ws_receive. | ||||
							
								
								
									
										1
									
								
								examples/ws_send/ixcrypto
									
									
									
									
									
										Symbolic link
									
								
							
							
						
						
									
										1
									
								
								examples/ws_send/ixcrypto
									
									
									
									
									
										Symbolic link
									
								
							| @@ -0,0 +1 @@ | ||||
| ../cobra_publisher/ixcrypto | ||||
							
								
								
									
										1
									
								
								examples/ws_send/jsoncpp
									
									
									
									
									
										Symbolic link
									
								
							
							
						
						
									
										1
									
								
								examples/ws_send/jsoncpp
									
									
									
									
									
										Symbolic link
									
								
							| @@ -0,0 +1 @@ | ||||
| ../cobra_publisher/jsoncpp | ||||
							
								
								
									
										306
									
								
								examples/ws_send/ws_send.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										306
									
								
								examples/ws_send/ws_send.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,306 @@ | ||||
| /* | ||||
|  *  ws_send.cpp | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2017-2018 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #include <iostream> | ||||
| #include <fstream> | ||||
| #include <sstream> | ||||
| #include <vector> | ||||
| #include <condition_variable> | ||||
| #include <mutex> | ||||
| #include <chrono> | ||||
| #include <ixwebsocket/IXWebSocket.h> | ||||
| #include <ixwebsocket/IXSocket.h> | ||||
| #include <ixcrypto/IXUuid.h> | ||||
| #include <ixcrypto/IXBase64.h> | ||||
| #include <ixcrypto/IXHash.h> | ||||
| #include <jsoncpp/json/json.h> | ||||
|  | ||||
| using namespace ix; | ||||
|  | ||||
| namespace | ||||
| { | ||||
|     void log(const std::string& msg) | ||||
|     { | ||||
|         std::cout << msg << std::endl; | ||||
|     } | ||||
|  | ||||
|     class WebSocketSender | ||||
|     { | ||||
|         public: | ||||
|             WebSocketSender(const std::string& _url, | ||||
|                             bool enablePerMessageDeflate); | ||||
|  | ||||
|             void subscribe(const std::string& channel); | ||||
|             void start(); | ||||
|             void stop(); | ||||
|  | ||||
|             void waitForConnection(); | ||||
|             void waitForAck(); | ||||
|  | ||||
|             void sendMessage(const std::string& filename, bool throttle); | ||||
|  | ||||
|         private: | ||||
|             std::string _url; | ||||
|             std::string _id; | ||||
|             ix::WebSocket _webSocket; | ||||
|             bool _enablePerMessageDeflate; | ||||
|  | ||||
|             std::mutex _conditionVariableMutex; | ||||
|             std::condition_variable _condition; | ||||
|     }; | ||||
|  | ||||
|     WebSocketSender::WebSocketSender(const std::string& url, | ||||
|                                      bool enablePerMessageDeflate) : | ||||
|         _url(url), | ||||
|         _enablePerMessageDeflate(enablePerMessageDeflate) | ||||
|     { | ||||
|         ; | ||||
|     } | ||||
|  | ||||
|     void WebSocketSender::stop() | ||||
|     { | ||||
|         _webSocket.stop(); | ||||
|     } | ||||
|  | ||||
|     void WebSocketSender::waitForConnection() | ||||
|     { | ||||
|         std::cout << "Connecting..." << std::endl; | ||||
|  | ||||
|         std::unique_lock<std::mutex> lock(_conditionVariableMutex); | ||||
|         _condition.wait(lock); | ||||
|     } | ||||
|  | ||||
|     void WebSocketSender::waitForAck() | ||||
|     { | ||||
|         std::cout << "Waiting for ack..." << std::endl; | ||||
|  | ||||
|         std::unique_lock<std::mutex> lock(_conditionVariableMutex); | ||||
|         _condition.wait(lock); | ||||
|     } | ||||
|  | ||||
|     std::string load(const std::string& path) | ||||
|     { | ||||
|         // std::vector<uint8_t> memblock; | ||||
|         std::string str; | ||||
|  | ||||
|         std::ifstream file(path); | ||||
|         if (!file.is_open()) return std::string(); | ||||
|  | ||||
|         file.seekg(0, file.end); | ||||
|         std::streamoff size = file.tellg(); | ||||
|         file.seekg(0, file.beg); | ||||
|  | ||||
|         str.resize(size); | ||||
|         file.read((char*)&str.front(), static_cast<std::streamsize>(size)); | ||||
|  | ||||
|         return str; | ||||
|     } | ||||
|  | ||||
|     void WebSocketSender::start() | ||||
|     { | ||||
|         _webSocket.setUrl(_url); | ||||
|  | ||||
|         ix::WebSocketPerMessageDeflateOptions webSocketPerMessageDeflateOptions( | ||||
|             _enablePerMessageDeflate, false, false, 15, 15); | ||||
|         _webSocket.setPerMessageDeflateOptions(webSocketPerMessageDeflateOptions); | ||||
|  | ||||
|         std::stringstream ss; | ||||
|         log(std::string("Connecting to url: ") + _url); | ||||
|  | ||||
|         _webSocket.setOnMessageCallback( | ||||
|             [this](ix::WebSocketMessageType messageType, | ||||
|                const std::string& str, | ||||
|                size_t wireSize, | ||||
|                const ix::WebSocketErrorInfo& error, | ||||
|                const ix::WebSocketOpenInfo& openInfo, | ||||
|                const ix::WebSocketCloseInfo& closeInfo) | ||||
|             { | ||||
|                 std::stringstream ss; | ||||
|                 if (messageType == ix::WebSocket_MessageType_Open) | ||||
|                 { | ||||
|                     _condition.notify_one(); | ||||
|  | ||||
|                     log("ws_send: connected"); | ||||
|                     std::cout << "Uri: " << openInfo.uri << std::endl; | ||||
|                     std::cout << "Handshake Headers:" << std::endl; | ||||
|                     for (auto it : openInfo.headers) | ||||
|                     { | ||||
|                         std::cout << it.first << ": " << it.second << std::endl; | ||||
|                     } | ||||
|                 } | ||||
|                 else if (messageType == ix::WebSocket_MessageType_Close) | ||||
|                 { | ||||
|                     ss << "ws_send: connection closed:"; | ||||
|                     ss << " code " << closeInfo.code; | ||||
|                     ss << " reason " << closeInfo.reason << std::endl; | ||||
|                     log(ss.str()); | ||||
|                 } | ||||
|                 else if (messageType == ix::WebSocket_MessageType_Message) | ||||
|                 { | ||||
|                     _condition.notify_one(); | ||||
|  | ||||
|                     ss << "ws_send: received message: " | ||||
|                        << str; | ||||
|                     log(ss.str()); | ||||
|  | ||||
|                     Json::Value data; | ||||
|                     Json::Reader reader; | ||||
|                     if (!reader.parse(str, data)) | ||||
|                     { | ||||
|                         std::cerr << "Invalid JSON response" << std::endl; | ||||
|                         return; | ||||
|                     } | ||||
|  | ||||
|                     std::string id = data["id"].asString(); | ||||
|                     if (_id != id) | ||||
|                     { | ||||
|                         std::cerr << "Invalid id" << std::endl; | ||||
|                     } | ||||
|                 } | ||||
|                 else if (messageType == ix::WebSocket_MessageType_Error) | ||||
|                 { | ||||
|                     ss << "Connection error: " << error.reason      << std::endl; | ||||
|                     ss << "#retries: "         << error.retries     << std::endl; | ||||
|                     ss << "Wait time(ms): "    << error.wait_time   << std::endl; | ||||
|                     ss << "HTTP Status: "      << error.http_status << std::endl; | ||||
|                     log(ss.str()); | ||||
|                 } | ||||
|                 else | ||||
|                 { | ||||
|                     ss << "Invalid ix::WebSocketMessageType"; | ||||
|                     log(ss.str()); | ||||
|                 } | ||||
|             }); | ||||
|  | ||||
|         _webSocket.start(); | ||||
|     } | ||||
|  | ||||
|     class Bench | ||||
|     { | ||||
|         public: | ||||
|             Bench(const std::string& description) : | ||||
|                 _description(description), | ||||
|                 _start(std::chrono::system_clock::now()), | ||||
|                 _reported(false) | ||||
|             { | ||||
|                 ; | ||||
|             } | ||||
|  | ||||
|             ~Bench() | ||||
|             { | ||||
|                 if (!_reported) | ||||
|                 { | ||||
|                     report(); | ||||
|                 } | ||||
|             } | ||||
|  | ||||
|             void report() | ||||
|             { | ||||
|                 auto now = std::chrono::system_clock::now(); | ||||
|                 auto milliseconds = std::chrono::duration_cast<std::chrono::milliseconds>(now - _start); | ||||
|  | ||||
|                 _ms = milliseconds.count(); | ||||
|                 std::cout << _description << " completed in " | ||||
|                           << _ms << "ms" << std::endl; | ||||
|  | ||||
|                 _reported = true; | ||||
|             } | ||||
|  | ||||
|             uint64_t getDuration() const | ||||
|             { | ||||
|                 return _ms; | ||||
|             } | ||||
|  | ||||
|         private: | ||||
|             std::string _description; | ||||
|             std::chrono::time_point<std::chrono::system_clock> _start; | ||||
|             uint64_t _ms; | ||||
|             bool _reported; | ||||
|     }; | ||||
|  | ||||
|     void WebSocketSender::sendMessage(const std::string& filename, | ||||
|                                       bool throttle) | ||||
|     { | ||||
|         std::string content; | ||||
|         { | ||||
|             Bench bench("load file from disk"); | ||||
|             content = load(filename); | ||||
|         } | ||||
|  | ||||
|         _id = uuid4(); | ||||
|  | ||||
|         std::string b64Content; | ||||
|         { | ||||
|             Bench bench("base 64 encode file"); | ||||
|             b64Content = base64_encode(content, content.size()); | ||||
|         } | ||||
|  | ||||
|         Json::Value pdu; | ||||
|         pdu["kind"] = "send"; | ||||
|         pdu["id"] = _id; | ||||
|         pdu["content"] = b64Content; | ||||
|         pdu["djb2_hash"] = djb2Hash(b64Content); | ||||
|         pdu["filename"] = filename; | ||||
|  | ||||
|         Bench bench("Sending file through websocket"); | ||||
|         _webSocket.send(pdu.toStyledString(), | ||||
|                         [throttle](int current, int total) -> bool | ||||
|         { | ||||
|             std::cout << "Step " << current << " out of " << total << std::endl; | ||||
|  | ||||
|             if (throttle) | ||||
|             { | ||||
|                 std::chrono::duration<double, std::milli> duration(10); | ||||
|                 std::this_thread::sleep_for(duration); | ||||
|             } | ||||
|  | ||||
|             return true; | ||||
|         }); | ||||
|  | ||||
|         bench.report(); | ||||
|         auto duration = bench.getDuration(); | ||||
|         auto transferRate = 1000 * b64Content.size() / duration; | ||||
|         transferRate /= (1024 * 1024); | ||||
|         std::cout << "Send transfer rate: " << transferRate << "MB/s" << std::endl; | ||||
|     } | ||||
|  | ||||
|     void wsSend(const std::string& url, | ||||
|                 const std::string& path, | ||||
|                 bool enablePerMessageDeflate, | ||||
|                 bool throttle) | ||||
|     { | ||||
|         WebSocketSender webSocketSender(url, enablePerMessageDeflate); | ||||
|         webSocketSender.start(); | ||||
|  | ||||
|         webSocketSender.waitForConnection(); | ||||
|  | ||||
|         std::cout << "Sending..." << std::endl; | ||||
|         webSocketSender.sendMessage(path, throttle); | ||||
|  | ||||
|         webSocketSender.waitForAck(); | ||||
|  | ||||
|         std::cout << "Done !" << std::endl; | ||||
|         webSocketSender.stop(); | ||||
|     } | ||||
| } | ||||
|  | ||||
| int main(int argc, char** argv) | ||||
| { | ||||
|     if (argc != 3) | ||||
|     { | ||||
|         std::cerr << "Usage: ws_send <url> <path>" << std::endl; | ||||
|         return 1; | ||||
|     } | ||||
|     std::string url = argv[1]; | ||||
|     std::string path = argv[2]; | ||||
|  | ||||
|     bool throttle = false; | ||||
|     bool enablePerMessageDeflate = false; | ||||
|  | ||||
|     Socket::init(); | ||||
|     wsSend(url, path, enablePerMessageDeflate, throttle); | ||||
|     return 0; | ||||
| } | ||||
							
								
								
									
										471
									
								
								ixwebsocket/IXHttpClient.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										471
									
								
								ixwebsocket/IXHttpClient.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,471 @@ | ||||
| /* | ||||
|  *  IXHttpClient.cpp | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #include "IXHttpClient.h" | ||||
| #include "IXUrlParser.h" | ||||
| #include "IXWebSocketHttpHeaders.h" | ||||
| #include "IXSocketFactory.h" | ||||
|  | ||||
| #include <sstream> | ||||
| #include <iomanip> | ||||
| #include <vector> | ||||
| #include <cstring> | ||||
|  | ||||
| #include <zlib.h> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     const std::string HttpClient::kPost = "POST"; | ||||
|     const std::string HttpClient::kGet = "GET"; | ||||
|     const std::string HttpClient::kHead = "HEAD"; | ||||
|  | ||||
|     HttpClient::HttpClient() | ||||
|     { | ||||
|  | ||||
|     } | ||||
|  | ||||
|     HttpClient::~HttpClient() | ||||
|     { | ||||
|  | ||||
|     } | ||||
|  | ||||
|     HttpResponse HttpClient::request( | ||||
|         const std::string& url, | ||||
|         const std::string& verb, | ||||
|         const std::string& body, | ||||
|         const HttpRequestArgs& args, | ||||
|         int redirects) | ||||
|     { | ||||
|         uint64_t uploadSize = 0; | ||||
|         uint64_t downloadSize = 0; | ||||
|         int code = 0; | ||||
|         WebSocketHttpHeaders headers; | ||||
|         std::string payload; | ||||
|  | ||||
|         std::string protocol, host, path, query; | ||||
|         int port; | ||||
|         bool websocket = false; | ||||
|  | ||||
|         if (!UrlParser::parse(url, protocol, host, path, query, port, websocket)) | ||||
|         { | ||||
|             std::stringstream ss; | ||||
|             ss << "Cannot parse url: " << url; | ||||
|             return std::make_tuple(code, HttpErrorCode_UrlMalformed, | ||||
|                                    headers, payload, ss.str(), | ||||
|                                    uploadSize, downloadSize); | ||||
|         } | ||||
|  | ||||
|         bool tls = protocol == "https"; | ||||
|         std::string errorMsg; | ||||
|         _socket = createSocket(tls, errorMsg); | ||||
|  | ||||
|         if (!_socket) | ||||
|         { | ||||
|             return std::make_tuple(code, HttpErrorCode_CannotCreateSocket, | ||||
|                                    headers, payload, errorMsg, | ||||
|                                    uploadSize, downloadSize); | ||||
|         } | ||||
|  | ||||
|         // Build request string | ||||
|         std::stringstream ss; | ||||
|         ss << verb << " " << path << " HTTP/1.1\r\n"; | ||||
|         ss << "Host: " << host << "\r\n"; | ||||
|         ss << "User-Agent: ixwebsocket/1.0.0" << "\r\n"; | ||||
|         ss << "Accept: */*" << "\r\n"; | ||||
|  | ||||
|         if (args.compress) | ||||
|         { | ||||
|             ss << "Accept-Encoding: gzip" << "\r\n"; | ||||
|         } | ||||
|  | ||||
|         // Append extra headers | ||||
|         for (auto&& it : args.extraHeaders) | ||||
|         { | ||||
|             ss << it.first << ": " << it.second << "\r\n"; | ||||
|         } | ||||
|  | ||||
|         if (verb == kPost) | ||||
|         { | ||||
|             ss << "Content-Length: " << body.size() << "\r\n"; | ||||
|  | ||||
|             // Set default Content-Type if unspecified | ||||
|             if (args.extraHeaders.find("Content-Type") == args.extraHeaders.end()) | ||||
|             { | ||||
|                 ss << "Content-Type: application/x-www-form-urlencoded" << "\r\n"; | ||||
|             } | ||||
|             ss << "\r\n"; | ||||
|             ss << body; | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|             ss << "\r\n"; | ||||
|         } | ||||
|  | ||||
|         std::string req(ss.str()); | ||||
|         std::string errMsg; | ||||
|         std::atomic<bool> requestInitCancellation(false); | ||||
|  | ||||
|         // Make a cancellation object dealing with connection timeout | ||||
|         auto isCancellationRequested = | ||||
|             makeCancellationRequestWithTimeout(args.connectTimeout, requestInitCancellation); | ||||
|  | ||||
|         bool success = _socket->connect(host, port, errMsg, isCancellationRequested); | ||||
|         if (!success) | ||||
|         { | ||||
|             std::stringstream ss; | ||||
|             ss << "Cannot connect to url: " << url; | ||||
|             return std::make_tuple(code, HttpErrorCode_CannotConnect, | ||||
|                                    headers, payload, ss.str(), | ||||
|                                    uploadSize, downloadSize); | ||||
|         } | ||||
|  | ||||
|         // Make a new cancellation object dealing with transfer timeout | ||||
|         isCancellationRequested = | ||||
|             makeCancellationRequestWithTimeout(args.transferTimeout, requestInitCancellation); | ||||
|  | ||||
|         if (args.verbose) | ||||
|         { | ||||
|             std::stringstream ss; | ||||
|             ss << "Sending " << verb << " request " | ||||
|                << "to " << host << ":" << port << std::endl | ||||
|                << "request size: " << req.size() << " bytes" << std::endl | ||||
|                << "=============" << std::endl | ||||
|                << req | ||||
|                << "=============" << std::endl | ||||
|                << std::endl; | ||||
|  | ||||
|             log(ss.str(), args); | ||||
|         } | ||||
|  | ||||
|         if (!_socket->writeBytes(req, isCancellationRequested)) | ||||
|         { | ||||
|             std::string errorMsg("Cannot send request"); | ||||
|             return std::make_tuple(code, HttpErrorCode_SendError, | ||||
|                                    headers, payload, errorMsg, | ||||
|                                    uploadSize, downloadSize); | ||||
|         } | ||||
|  | ||||
|         uploadSize = req.size(); | ||||
|  | ||||
|         auto lineResult = _socket->readLine(isCancellationRequested); | ||||
|         auto lineValid = lineResult.first; | ||||
|         auto line = lineResult.second; | ||||
|  | ||||
|         if (!lineValid) | ||||
|         { | ||||
|             std::string errorMsg("Cannot retrieve status line"); | ||||
|             return std::make_tuple(code, HttpErrorCode_CannotReadStatusLine, | ||||
|                                    headers, payload, errorMsg, | ||||
|                                    uploadSize, downloadSize); | ||||
|         } | ||||
|  | ||||
|         if (args.verbose) | ||||
|         { | ||||
|             std::stringstream ss; | ||||
|             ss << "Status line " << line; | ||||
|             log(ss.str(), args); | ||||
|         } | ||||
|  | ||||
|         if (sscanf(line.c_str(), "HTTP/1.1 %d", &code) != 1) | ||||
|         { | ||||
|             std::string errorMsg("Cannot parse response code from status line"); | ||||
|             return std::make_tuple(code, HttpErrorCode_MissingStatus, | ||||
|                                    headers, payload, errorMsg, | ||||
|                                    uploadSize, downloadSize); | ||||
|         } | ||||
|  | ||||
|         auto result = parseHttpHeaders(_socket, isCancellationRequested); | ||||
|         auto headersValid = result.first; | ||||
|         headers = result.second; | ||||
|  | ||||
|         if (!headersValid) | ||||
|         { | ||||
|             std::string errorMsg("Cannot parse http headers"); | ||||
|             return std::make_tuple(code, HttpErrorCode_HeaderParsingError, | ||||
|                                    headers, payload, errorMsg, | ||||
|                                    uploadSize, downloadSize); | ||||
|         } | ||||
|  | ||||
|         // Redirect ? | ||||
|         if ((code >= 301 && code <= 308) && args.followRedirects) | ||||
|         { | ||||
|             if (headers.find("Location") == headers.end()) | ||||
|             { | ||||
|                 std::string errorMsg("Missing location header for redirect"); | ||||
|                 return std::make_tuple(code, HttpErrorCode_MissingLocation, | ||||
|                                        headers, payload, errorMsg, | ||||
|                                        uploadSize, downloadSize); | ||||
|             } | ||||
|  | ||||
|             if (redirects >= args.maxRedirects) | ||||
|             { | ||||
|                 std::stringstream ss; | ||||
|                 ss << "Too many redirects: " << redirects; | ||||
|                 return std::make_tuple(code, HttpErrorCode_TooManyRedirects, | ||||
|                                        headers, payload, ss.str(), | ||||
|                                        uploadSize, downloadSize); | ||||
|             } | ||||
|  | ||||
|             // Recurse | ||||
|             std::string location = headers["Location"]; | ||||
|             return request(location, verb, body, args, redirects+1); | ||||
|         } | ||||
|  | ||||
|         if (verb == "HEAD") | ||||
|         { | ||||
|             return std::make_tuple(code, HttpErrorCode_Ok, | ||||
|                                    headers, payload, std::string(), | ||||
|                                    uploadSize, downloadSize); | ||||
|         } | ||||
|  | ||||
|         // Parse response: | ||||
|         if (headers.find("Content-Length") != headers.end()) | ||||
|         { | ||||
|             ssize_t contentLength = -1; | ||||
|             ss.str(""); | ||||
|             ss << headers["Content-Length"]; | ||||
|             ss >> contentLength; | ||||
|  | ||||
|             payload.reserve(contentLength); | ||||
|  | ||||
|             // FIXME: very inefficient way to read bytes, but it works... | ||||
|             for (int i = 0; i < contentLength; ++i) | ||||
|             { | ||||
|                 char c; | ||||
|                 if (!_socket->readByte(&c, isCancellationRequested)) | ||||
|                 { | ||||
|                     return std::make_tuple(code, HttpErrorCode_ReadError, | ||||
|                                            headers, payload, "Cannot read byte", | ||||
|                                            uploadSize, downloadSize); | ||||
|                 } | ||||
|  | ||||
|                 payload += c; | ||||
|             } | ||||
|         } | ||||
|         else if (headers.find("Transfer-Encoding") != headers.end() && | ||||
|                  headers["Transfer-Encoding"] == "chunked") | ||||
|         { | ||||
|             std::stringstream ss; | ||||
|  | ||||
|             while (true) | ||||
|             { | ||||
|                 lineResult = _socket->readLine(isCancellationRequested); | ||||
|                 line = lineResult.second; | ||||
|  | ||||
|                 if (!lineResult.first) | ||||
|                 { | ||||
|                     return std::make_tuple(code, HttpErrorCode_ChunkReadError, | ||||
|                                            headers, payload, errorMsg, | ||||
|                                            uploadSize, downloadSize); | ||||
|                 } | ||||
|  | ||||
|                 uint64_t chunkSize; | ||||
|                 ss.str(""); | ||||
|                 ss << std::hex << line; | ||||
|                 ss >> chunkSize; | ||||
|  | ||||
|                 if (args.verbose) | ||||
|                 { | ||||
|                     std::stringstream oss; | ||||
|                     oss << "Reading " << chunkSize << " bytes" | ||||
|                         << std::endl; | ||||
|                     log(oss.str(), args); | ||||
|                 } | ||||
|  | ||||
|                 payload.reserve(payload.size() + chunkSize); | ||||
|  | ||||
|                 // Read another line | ||||
|  | ||||
|                 for (uint64_t i = 0; i < chunkSize; ++i) | ||||
|                 { | ||||
|                     char c; | ||||
|                     if (!_socket->readByte(&c, isCancellationRequested)) | ||||
|                     { | ||||
|                         errorMsg = "Cannot read byte"; | ||||
|                         return std::make_tuple(code, HttpErrorCode_ChunkReadError, | ||||
|                                                headers, payload, errorMsg, | ||||
|                                                uploadSize, downloadSize); | ||||
|                     } | ||||
|  | ||||
|                     payload += c; | ||||
|                 } | ||||
|  | ||||
|                 lineResult = _socket->readLine(isCancellationRequested); | ||||
|  | ||||
|                 if (!lineResult.first) | ||||
|                 { | ||||
|                     return std::make_tuple(code, HttpErrorCode_ChunkReadError, | ||||
|                                            headers, payload, errorMsg, | ||||
|                                            uploadSize, downloadSize); | ||||
|                 } | ||||
|  | ||||
|                 if (chunkSize == 0) break; | ||||
|             } | ||||
|         } | ||||
|         else if (code == 204) | ||||
|         { | ||||
|             ; // 204 is NoContent response code | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|             std::string errorMsg("Cannot read http body"); | ||||
|             return std::make_tuple(code, HttpErrorCode_CannotReadBody, | ||||
|                                    headers, payload, errorMsg, | ||||
|                                    uploadSize, downloadSize); | ||||
|         } | ||||
|  | ||||
|         downloadSize = payload.size(); | ||||
|  | ||||
|         // If the content was compressed with gzip, decode it | ||||
|         if (headers["Content-Encoding"] == "gzip") | ||||
|         { | ||||
|             std::string decompressedPayload; | ||||
|             if (!gzipInflate(payload, decompressedPayload)) | ||||
|             { | ||||
|                 std::string errorMsg("Error decompressing payload"); | ||||
|                 return std::make_tuple(code, HttpErrorCode_Gzip, | ||||
|                                        headers, payload, errorMsg, | ||||
|                                        uploadSize, downloadSize); | ||||
|             } | ||||
|             payload = decompressedPayload; | ||||
|         } | ||||
|  | ||||
|         return std::make_tuple(code, HttpErrorCode_Ok, | ||||
|                                headers, payload, std::string(), | ||||
|                                uploadSize, downloadSize); | ||||
|     } | ||||
|  | ||||
|     HttpResponse HttpClient::get(const std::string& url, | ||||
|                                  const HttpRequestArgs& args) | ||||
|     { | ||||
|         return request(url, kGet, std::string(), args); | ||||
|     } | ||||
|  | ||||
|     HttpResponse HttpClient::head(const std::string& url, | ||||
|                                   const HttpRequestArgs& args) | ||||
|     { | ||||
|         return request(url, kHead, std::string(), args); | ||||
|     } | ||||
|  | ||||
|     HttpResponse HttpClient::post(const std::string& url, | ||||
|                                   const HttpParameters& httpParameters, | ||||
|                                   const HttpRequestArgs& args) | ||||
|     { | ||||
|         return request(url, kPost, serializeHttpParameters(httpParameters), args); | ||||
|     } | ||||
|  | ||||
|     HttpResponse HttpClient::post(const std::string& url, | ||||
|                                   const std::string& body, | ||||
|                                   const HttpRequestArgs& args) | ||||
|     { | ||||
|         return request(url, kPost, body, args); | ||||
|     } | ||||
|  | ||||
|     std::string HttpClient::urlEncode(const std::string& value) | ||||
|     { | ||||
|         std::ostringstream escaped; | ||||
|         escaped.fill('0'); | ||||
|         escaped << std::hex; | ||||
|  | ||||
|         for (std::string::const_iterator i = value.begin(), n = value.end(); | ||||
|              i != n; ++i) | ||||
|         { | ||||
|             std::string::value_type c = (*i); | ||||
|  | ||||
|             // Keep alphanumeric and other accepted characters intact | ||||
|             if (isalnum(c) || c == '-' || c == '_' || c == '.' || c == '~') | ||||
|             { | ||||
|                 escaped << c; | ||||
|                 continue; | ||||
|             } | ||||
|  | ||||
|             // Any other characters are percent-encoded | ||||
|             escaped << std::uppercase; | ||||
|             escaped << '%' << std::setw(2) << int((unsigned char) c); | ||||
|             escaped << std::nouppercase; | ||||
|         } | ||||
|  | ||||
|         return escaped.str(); | ||||
|     } | ||||
|  | ||||
|     std::string HttpClient::serializeHttpParameters(const HttpParameters& httpParameters) | ||||
|     { | ||||
|         std::stringstream ss; | ||||
|         size_t count = httpParameters.size(); | ||||
|         size_t i = 0; | ||||
|  | ||||
|         for (auto&& it : httpParameters) | ||||
|         { | ||||
|             ss << urlEncode(it.first) | ||||
|                << "=" | ||||
|                << urlEncode(it.second); | ||||
|  | ||||
|             if (i++ < (count-1)) | ||||
|             { | ||||
|                ss << "&"; | ||||
|             } | ||||
|         } | ||||
|         return ss.str(); | ||||
|     } | ||||
|  | ||||
|     bool HttpClient::gzipInflate( | ||||
|         const std::string& in, | ||||
|         std::string& out) | ||||
|     { | ||||
|         z_stream inflateState; | ||||
|         std::memset(&inflateState, 0, sizeof(inflateState)); | ||||
|  | ||||
|         inflateState.zalloc = Z_NULL; | ||||
|         inflateState.zfree = Z_NULL; | ||||
|         inflateState.opaque = Z_NULL; | ||||
|         inflateState.avail_in = 0; | ||||
|         inflateState.next_in = Z_NULL; | ||||
|  | ||||
|         if (inflateInit2(&inflateState, 16+MAX_WBITS) != Z_OK) | ||||
|         { | ||||
|             return false; | ||||
|         } | ||||
|  | ||||
|         inflateState.avail_in = (uInt) in.size(); | ||||
|         inflateState.next_in = (unsigned char *)(const_cast<char *>(in.data())); | ||||
|  | ||||
|         const int kBufferSize = 1 << 14; | ||||
|  | ||||
|         std::unique_ptr<unsigned char[]> compressBuffer = | ||||
|             std::make_unique<unsigned char[]>(kBufferSize); | ||||
|  | ||||
|         do | ||||
|         { | ||||
|             inflateState.avail_out = (uInt) kBufferSize; | ||||
|             inflateState.next_out = compressBuffer.get(); | ||||
|  | ||||
|             int ret = inflate(&inflateState, Z_SYNC_FLUSH); | ||||
|  | ||||
|             if (ret == Z_NEED_DICT || ret == Z_DATA_ERROR || ret == Z_MEM_ERROR) | ||||
|             { | ||||
|                 inflateEnd(&inflateState); | ||||
|                 return false; | ||||
|             } | ||||
|  | ||||
|             out.append( | ||||
|                 reinterpret_cast<char *>(compressBuffer.get()), | ||||
|                 kBufferSize - inflateState.avail_out | ||||
|             ); | ||||
|         } while (inflateState.avail_out == 0); | ||||
|  | ||||
|         inflateEnd(&inflateState); | ||||
|         return true; | ||||
|     } | ||||
|  | ||||
|     void HttpClient::log(const std::string& msg, | ||||
|                          const HttpRequestArgs& args) | ||||
|     { | ||||
|         if (args.logger) | ||||
|         { | ||||
|             args.logger(msg); | ||||
|         } | ||||
|     } | ||||
| } | ||||
							
								
								
									
										106
									
								
								ixwebsocket/IXHttpClient.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										106
									
								
								ixwebsocket/IXHttpClient.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,106 @@ | ||||
| /* | ||||
|  *  IXHttpClient.h | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| #include <algorithm> | ||||
| #include <functional> | ||||
| #include <mutex> | ||||
| #include <atomic> | ||||
| #include <tuple> | ||||
| #include <memory> | ||||
| #include <map> | ||||
|  | ||||
| #include "IXSocket.h" | ||||
| #include "IXWebSocketHttpHeaders.h" | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     enum HttpErrorCode | ||||
|     { | ||||
|         HttpErrorCode_Ok = 0, | ||||
|         HttpErrorCode_CannotConnect = 1, | ||||
|         HttpErrorCode_Timeout = 2, | ||||
|         HttpErrorCode_Gzip = 3, | ||||
|         HttpErrorCode_UrlMalformed = 4, | ||||
|         HttpErrorCode_CannotCreateSocket = 5, | ||||
|         HttpErrorCode_SendError = 6, | ||||
|         HttpErrorCode_ReadError = 7, | ||||
|         HttpErrorCode_CannotReadStatusLine = 8, | ||||
|         HttpErrorCode_MissingStatus = 9, | ||||
|         HttpErrorCode_HeaderParsingError = 10, | ||||
|         HttpErrorCode_MissingLocation = 11, | ||||
|         HttpErrorCode_TooManyRedirects = 12, | ||||
|         HttpErrorCode_ChunkReadError = 13, | ||||
|         HttpErrorCode_CannotReadBody = 14 | ||||
|     }; | ||||
|  | ||||
|     using HttpResponse = std::tuple<int, // status | ||||
|                                     HttpErrorCode, // error code | ||||
|                                     WebSocketHttpHeaders, | ||||
|                                     std::string, // payload | ||||
|                                     std::string, // error msg | ||||
|                                     uint64_t,    // upload size | ||||
|                                     uint64_t>;   // download size | ||||
|  | ||||
|     using HttpParameters = std::map<std::string, std::string>; | ||||
|     using Logger = std::function<void(const std::string&)>; | ||||
|  | ||||
|     struct HttpRequestArgs | ||||
|     { | ||||
|         std::string url; | ||||
|         WebSocketHttpHeaders extraHeaders; | ||||
|         std::string body; | ||||
|         int connectTimeout; | ||||
|         int transferTimeout; | ||||
|         bool followRedirects; | ||||
|         int maxRedirects; | ||||
|         bool verbose; | ||||
|         bool compress; | ||||
|         Logger logger; | ||||
|     }; | ||||
|  | ||||
|     class HttpClient { | ||||
|     public: | ||||
|         HttpClient(); | ||||
|         ~HttpClient(); | ||||
|  | ||||
|         HttpResponse get(const std::string& url, | ||||
|                          const HttpRequestArgs& args); | ||||
|         HttpResponse head(const std::string& url, | ||||
|                           const HttpRequestArgs& args); | ||||
|  | ||||
|         HttpResponse post(const std::string& url, | ||||
|                           const HttpParameters& httpParameters, | ||||
|                           const HttpRequestArgs& args); | ||||
|         HttpResponse post(const std::string& url, | ||||
|                           const std::string& body, | ||||
|                           const HttpRequestArgs& args); | ||||
|  | ||||
|     private: | ||||
|         HttpResponse request(const std::string& url, | ||||
|                              const std::string& verb, | ||||
|                              const std::string& body, | ||||
|                              const HttpRequestArgs& args, | ||||
|                              int redirects = 0); | ||||
|  | ||||
|         std::string serializeHttpParameters(const HttpParameters& httpParameters); | ||||
|  | ||||
|         std::string urlEncode(const std::string& value); | ||||
|  | ||||
|         void log(const std::string& msg, const HttpRequestArgs& args); | ||||
|  | ||||
|         bool gzipInflate( | ||||
|             const std::string& in, | ||||
|             std::string& out); | ||||
|  | ||||
|         std::shared_ptr<Socket> _socket; | ||||
|  | ||||
|         const static std::string kPost; | ||||
|         const static std::string kGet; | ||||
|         const static std::string kHead; | ||||
|     }; | ||||
| } | ||||
							
								
								
									
										42
									
								
								ixwebsocket/IXSocketFactory.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										42
									
								
								ixwebsocket/IXSocketFactory.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,42 @@ | ||||
| /* | ||||
|  *  IXSocketFactory.cpp | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #include "IXSocketFactory.h" | ||||
|  | ||||
| #if defined(__APPLE__) or defined(__linux__) | ||||
| # ifdef __APPLE__ | ||||
| #  include <ixwebsocket/IXSocketAppleSSL.h> | ||||
| # else | ||||
| #  include <ixwebsocket/IXSocketOpenSSL.h> | ||||
| # endif | ||||
| #endif | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     std::shared_ptr<Socket> createSocket(bool tls, | ||||
|                                          std::string& errorMsg) | ||||
|     { | ||||
|         errorMsg.clear(); | ||||
|  | ||||
|         if (!tls) | ||||
|         { | ||||
|             return std::make_shared<Socket>(); | ||||
|         } | ||||
|         else | ||||
|         { | ||||
| #ifdef IXWEBSOCKET_USE_TLS | ||||
| # ifdef __APPLE__ | ||||
|             return std::make_shared<SocketAppleSSL>(); | ||||
| # else | ||||
|             return std::make_shared<SocketOpenSSL>(); | ||||
| # endif | ||||
| #else | ||||
|             errorMsg = "TLS support is not enabled on this platform."; | ||||
|             return nullptr; | ||||
| #endif | ||||
|         } | ||||
|     } | ||||
| } | ||||
							
								
								
									
										17
									
								
								ixwebsocket/IXSocketFactory.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								ixwebsocket/IXSocketFactory.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,17 @@ | ||||
|  | ||||
| /* | ||||
|  *  IXSocketFactory.h | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2018 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| #include <memory> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     class Socket; | ||||
|     std::shared_ptr<Socket> createSocket(bool tls, | ||||
|                                          std::string& errorMsg); | ||||
| } | ||||
							
								
								
									
										104
									
								
								ixwebsocket/IXUrlParser.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										104
									
								
								ixwebsocket/IXUrlParser.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,104 @@ | ||||
| /* | ||||
|  *  IXUrlParser.cpp | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #include "IXUrlParser.h" | ||||
|  | ||||
| #include <iostream> | ||||
| #include <sstream> | ||||
|  | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     // | ||||
|     // The only difference between those 2 regex is the protocol | ||||
|     // | ||||
|     std::regex UrlParser::_httpRegex("(http|https)://([^/ :]+):?([^/ ]*)(/?[^ #?]*)\\x3f?([^ #]*)#?([^ ]*)"); | ||||
|     std::regex UrlParser::_webSocketRegex("(ws|wss)://([^/ :]+):?([^/ ]*)(/?[^ #?]*)\\x3f?([^ #]*)#?([^ ]*)"); | ||||
|  | ||||
|     bool UrlParser::parse(const std::string& url, | ||||
|                           std::string& protocol, | ||||
|                           std::string& host, | ||||
|                           std::string& path, | ||||
|                           std::string& query, | ||||
|                           int& port, | ||||
|                           bool websocket) | ||||
|     { | ||||
|         std::cmatch what; | ||||
|         if (!regex_match(url.c_str(), what, | ||||
|                          websocket ? _webSocketRegex : _httpRegex)) | ||||
|         { | ||||
|             return false; | ||||
|         } | ||||
|  | ||||
|         std::string portStr; | ||||
|  | ||||
|         protocol = std::string(what[1].first, what[1].second); | ||||
|         host     = std::string(what[2].first, what[2].second); | ||||
|         portStr  = std::string(what[3].first, what[3].second); | ||||
|         path     = std::string(what[4].first, what[4].second); | ||||
|         query    = std::string(what[5].first, what[5].second); | ||||
|  | ||||
|         if (portStr.empty()) | ||||
|         { | ||||
|             if (protocol == "ws" || protocol == "http") | ||||
|             { | ||||
|                 port = 80; | ||||
|             } | ||||
|             else if (protocol == "wss" || protocol == "https") | ||||
|             { | ||||
|                 port = 443; | ||||
|             } | ||||
|             else | ||||
|             { | ||||
|                 // Invalid protocol. Should be caught by regex check | ||||
|                 // but this missing branch trigger cpplint linter. | ||||
|                 return false; | ||||
|             } | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|             std::stringstream ss; | ||||
|             ss << portStr; | ||||
|             ss >> port; | ||||
|         } | ||||
|  | ||||
|         if (path.empty()) | ||||
|         { | ||||
|             path = "/"; | ||||
|         } | ||||
|         else if (path[0] != '/') | ||||
|         { | ||||
|             path = '/' + path; | ||||
|         } | ||||
|  | ||||
|         if (!query.empty()) | ||||
|         { | ||||
|             path += "?"; | ||||
|             path += query; | ||||
|         } | ||||
|  | ||||
|         return true; | ||||
|     } | ||||
|  | ||||
|     void UrlParser::printUrl(const std::string& url, bool websocket) | ||||
|     { | ||||
|         std::string protocol, host, path, query; | ||||
|         int port {0}; | ||||
|  | ||||
|         if (!parse(url, protocol, host, path, query, port, websocket)) | ||||
|         { | ||||
|             return; | ||||
|         } | ||||
|  | ||||
|         std::cout << "[" << url << "]" << std::endl; | ||||
|         std::cout << protocol << std::endl; | ||||
|         std::cout << host << std::endl; | ||||
|         std::cout << port << std::endl; | ||||
|         std::cout << path << std::endl; | ||||
|         std::cout << query << std::endl; | ||||
|         std::cout << "-------------------------------" << std::endl; | ||||
|     } | ||||
| } | ||||
							
								
								
									
										31
									
								
								ixwebsocket/IXUrlParser.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								ixwebsocket/IXUrlParser.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | ||||
| /* | ||||
|  *  IXUrlParser.h | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| #include <string> | ||||
| #include <regex> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     class UrlParser | ||||
|     { | ||||
|     public: | ||||
|         static bool parse(const std::string& url, | ||||
|                           std::string& protocol, | ||||
|                           std::string& host, | ||||
|                           std::string& path, | ||||
|                           std::string& query, | ||||
|                           int& port, | ||||
|                           bool websocket); | ||||
|  | ||||
|         static void printUrl(const std::string& url, bool websocket); | ||||
|  | ||||
|     private: | ||||
|         static std::regex _httpRegex; | ||||
|         static std::regex _webSocketRegex; | ||||
|     }; | ||||
| } | ||||
| @@ -6,6 +6,7 @@ | ||||
|  | ||||
| #include "IXWebSocketHandshake.h" | ||||
| #include "IXSocketConnect.h" | ||||
| #include "IXUrlParser.h" | ||||
|  | ||||
| #include "libwshandshake.hpp" | ||||
|  | ||||
| @@ -32,90 +33,6 @@ namespace ix | ||||
|  | ||||
|     } | ||||
|  | ||||
|     bool WebSocketHandshake::parseUrl(const std::string& url, | ||||
|                                       std::string& protocol, | ||||
|                                       std::string& host, | ||||
|                                       std::string& path, | ||||
|                                       std::string& query, | ||||
|                                       int& port) | ||||
|     { | ||||
|         std::regex ex("(ws|wss)://([^/ :]+):?([^/ ]*)(/?[^ #?]*)\\x3f?([^ #]*)#?([^ ]*)"); | ||||
|         std::cmatch what; | ||||
|         if (!regex_match(url.c_str(), what, ex)) | ||||
|         { | ||||
|             return false; | ||||
|         } | ||||
|  | ||||
|         std::string portStr; | ||||
|  | ||||
|         protocol = std::string(what[1].first, what[1].second); | ||||
|         host     = std::string(what[2].first, what[2].second); | ||||
|         portStr  = std::string(what[3].first, what[3].second); | ||||
|         path     = std::string(what[4].first, what[4].second); | ||||
|         query    = std::string(what[5].first, what[5].second); | ||||
|  | ||||
|         if (portStr.empty()) | ||||
|         { | ||||
|             if (protocol == "ws") | ||||
|             { | ||||
|                 port = 80; | ||||
|             } | ||||
|             else if (protocol == "wss") | ||||
|             { | ||||
|                 port = 443; | ||||
|             } | ||||
|             else | ||||
|             { | ||||
|                 // Invalid protocol. Should be caught by regex check | ||||
|                 // but this missing branch trigger cpplint linter. | ||||
|                 return false; | ||||
|             } | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|             std::stringstream ss; | ||||
|             ss << portStr; | ||||
|             ss >> port; | ||||
|         } | ||||
|  | ||||
|         if (path.empty()) | ||||
|         { | ||||
|             path = "/"; | ||||
|         } | ||||
|         else if (path[0] != '/') | ||||
|         { | ||||
|             path = '/' + path; | ||||
|         } | ||||
|  | ||||
|         if (!query.empty()) | ||||
|         { | ||||
|             path += "?"; | ||||
|             path += query; | ||||
|         } | ||||
|  | ||||
|         return true; | ||||
|     } | ||||
|  | ||||
|     void WebSocketHandshake::printUrl(const std::string& url) | ||||
|     { | ||||
|         std::string protocol, host, path, query; | ||||
|         int port {0}; | ||||
|  | ||||
|         if (!WebSocketHandshake::parseUrl(url, protocol, host, | ||||
|                                           path, query, port)) | ||||
|         { | ||||
|             return; | ||||
|         } | ||||
|  | ||||
|         std::cout << "[" << url << "]" << std::endl; | ||||
|         std::cout << protocol << std::endl; | ||||
|         std::cout << host << std::endl; | ||||
|         std::cout << port << std::endl; | ||||
|         std::cout << path << std::endl; | ||||
|         std::cout << query << std::endl; | ||||
|         std::cout << "-------------------------------" << std::endl; | ||||
|     } | ||||
|  | ||||
|     std::string WebSocketHandshake::trim(const std::string& str) | ||||
|     { | ||||
|         std::string out(str); | ||||
| @@ -192,61 +109,6 @@ namespace ix | ||||
|         return s; | ||||
|     } | ||||
|  | ||||
|  | ||||
|     std::pair<bool, WebSocketHttpHeaders> WebSocketHandshake::parseHttpHeaders( | ||||
|         const CancellationRequest& isCancellationRequested) | ||||
|     { | ||||
|         WebSocketHttpHeaders headers; | ||||
|  | ||||
|         char line[256]; | ||||
|         int i; | ||||
|  | ||||
|         while (true) | ||||
|         { | ||||
|             int colon = 0; | ||||
|  | ||||
|             for (i = 0; | ||||
|                  i < 2 || (i < 255 && line[i-2] != '\r' && line[i-1] != '\n'); | ||||
|                  ++i) | ||||
|             { | ||||
|                 if (!_socket->readByte(line+i, isCancellationRequested)) | ||||
|                 { | ||||
|                     return std::make_pair(false, headers); | ||||
|                 } | ||||
|  | ||||
|                 if (line[i] == ':' && colon == 0) | ||||
|                 { | ||||
|                     colon = i; | ||||
|                 } | ||||
|             } | ||||
|             if (line[0] == '\r' && line[1] == '\n') | ||||
|             { | ||||
|                 break; | ||||
|             } | ||||
|  | ||||
|             // line is a single header entry. split by ':', and add it to our | ||||
|             // header map. ignore lines with no colon. | ||||
|             if (colon > 0) | ||||
|             { | ||||
|                 line[i] = '\0'; | ||||
|                 std::string lineStr(line); | ||||
|                 // colon is ':', colon+1 is ' ', colon+2 is the start of the value. | ||||
|                 // i is end of string (\0), i-colon is length of string minus key; | ||||
|                 // subtract 1 for '\0', 1 for '\n', 1 for '\r', | ||||
|                 // 1 for the ' ' after the ':', and total is -4 | ||||
|                 std::string name(lineStr.substr(0, colon)); | ||||
|                 std::string value(lineStr.substr(colon + 2, i - colon - 4)); | ||||
|  | ||||
|                 // Make the name lower case. | ||||
|                 std::transform(name.begin(), name.end(), name.begin(), ::tolower); | ||||
|  | ||||
|                 headers[name] = value; | ||||
|             } | ||||
|         } | ||||
|  | ||||
|         return std::make_pair(true, headers); | ||||
|     } | ||||
|  | ||||
|     WebSocketInitResult WebSocketHandshake::sendErrorResponse(int code, const std::string& reason) | ||||
|     { | ||||
|         std::stringstream ss; | ||||
| @@ -355,7 +217,7 @@ namespace ix | ||||
|             return WebSocketInitResult(false, status, ss.str()); | ||||
|         } | ||||
|  | ||||
|         auto result = parseHttpHeaders(isCancellationRequested); | ||||
|         auto result = parseHttpHeaders(_socket, isCancellationRequested); | ||||
|         auto headersValid = result.first; | ||||
|         auto headers = result.second; | ||||
|  | ||||
| @@ -450,7 +312,7 @@ namespace ix | ||||
|         } | ||||
|  | ||||
|         // Retrieve and validate HTTP headers | ||||
|         auto result = parseHttpHeaders(isCancellationRequested); | ||||
|         auto result = parseHttpHeaders(_socket, isCancellationRequested); | ||||
|         auto headersValid = result.first; | ||||
|         auto headers = result.second; | ||||
|  | ||||
|   | ||||
| @@ -59,19 +59,10 @@ namespace ix | ||||
|         WebSocketInitResult serverHandshake(int fd, | ||||
|                                             int timeoutSecs); | ||||
|  | ||||
|         static bool parseUrl(const std::string& url, | ||||
|                              std::string& protocol, | ||||
|                              std::string& host, | ||||
|                              std::string& path, | ||||
|                              std::string& query, | ||||
|                              int& port); | ||||
|  | ||||
|     private: | ||||
|         static void printUrl(const std::string& url); | ||||
|         std::string genRandomString(const int len); | ||||
|  | ||||
|         // Parse HTTP headers | ||||
|         std::pair<bool, WebSocketHttpHeaders> parseHttpHeaders(const CancellationRequest& isCancellationRequested); | ||||
|         WebSocketInitResult sendErrorResponse(int code, const std::string& reason); | ||||
|  | ||||
|         std::tuple<std::string, std::string, std::string> parseRequestLine(const std::string& line); | ||||
|   | ||||
							
								
								
									
										66
									
								
								ixwebsocket/IXWebSocketHttpHeaders.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										66
									
								
								ixwebsocket/IXWebSocketHttpHeaders.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,66 @@ | ||||
| /* | ||||
|  *  IXWebSocketHttpHeaders.h | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2018 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #include "IXWebSocketHttpHeaders.h" | ||||
| #include "IXSocket.h" | ||||
|  | ||||
| #include <string> | ||||
| #include <unordered_map> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     std::pair<bool, WebSocketHttpHeaders> parseHttpHeaders( | ||||
|         std::shared_ptr<Socket> socket, | ||||
|         const CancellationRequest& isCancellationRequested) | ||||
|     { | ||||
|         WebSocketHttpHeaders headers; | ||||
|  | ||||
|         char line[1024]; | ||||
|         int i; | ||||
|  | ||||
|         while (true) | ||||
|         { | ||||
|             int colon = 0; | ||||
|  | ||||
|             for (i = 0; | ||||
|                  i < 2 || (i < 1023 && line[i-2] != '\r' && line[i-1] != '\n'); | ||||
|                  ++i) | ||||
|             { | ||||
|                 if (!socket->readByte(line+i, isCancellationRequested)) | ||||
|                 { | ||||
|                     return std::make_pair(false, headers); | ||||
|                 } | ||||
|  | ||||
|                 if (line[i] == ':' && colon == 0) | ||||
|                 { | ||||
|                     colon = i; | ||||
|                 } | ||||
|             } | ||||
|             if (line[0] == '\r' && line[1] == '\n') | ||||
|             { | ||||
|                 break; | ||||
|             } | ||||
|  | ||||
|             // line is a single header entry. split by ':', and add it to our | ||||
|             // header map. ignore lines with no colon. | ||||
|             if (colon > 0) | ||||
|             { | ||||
|                 line[i] = '\0'; | ||||
|                 std::string lineStr(line); | ||||
|                 // colon is ':', colon+1 is ' ', colon+2 is the start of the value. | ||||
|                 // i is end of string (\0), i-colon is length of string minus key; | ||||
|                 // subtract 1 for '\0', 1 for '\n', 1 for '\r', | ||||
|                 // 1 for the ' ' after the ':', and total is -4 | ||||
|                 std::string name(lineStr.substr(0, colon)); | ||||
|                 std::string value(lineStr.substr(colon + 2, i - colon - 4)); | ||||
|  | ||||
|                 headers[name] = value; | ||||
|             } | ||||
|         } | ||||
|  | ||||
|         return std::make_pair(true, headers); | ||||
|     } | ||||
| } | ||||
| @@ -6,10 +6,40 @@ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| #include "IXCancellationRequest.h" | ||||
|  | ||||
| #include <string> | ||||
| #include <unordered_map> | ||||
| #include <map> | ||||
| #include <memory> | ||||
| #include <algorithm> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     using WebSocketHttpHeaders = std::unordered_map<std::string, std::string>; | ||||
|     class Socket; | ||||
|  | ||||
|     struct CaseInsensitiveLess | ||||
|     { | ||||
|         // Case Insensitive compare_less binary function | ||||
|         struct NocaseCompare | ||||
|         { | ||||
|             bool operator() (const unsigned char& c1, const unsigned char& c2) const | ||||
|             { | ||||
|                 return std::tolower(c1) < std::tolower(c2); | ||||
|             } | ||||
|         }; | ||||
|  | ||||
|         bool operator() (const std::string & s1, const std::string & s2) const | ||||
|         { | ||||
|             return std::lexicographical_compare | ||||
|                 (s1.begin(), s1.end(),   // source range | ||||
|                  s2.begin(), s2.end(),   // dest range | ||||
|                  NocaseCompare());  // comparison | ||||
|         } | ||||
|     }; | ||||
|  | ||||
|     using WebSocketHttpHeaders = std::map<std::string, std::string, CaseInsensitiveLess>; | ||||
|  | ||||
|     std::pair<bool, WebSocketHttpHeaders> parseHttpHeaders( | ||||
|         std::shared_ptr<Socket> socket, | ||||
|         const CancellationRequest& isCancellationRequested); | ||||
| } | ||||
|   | ||||
| @@ -11,6 +11,8 @@ | ||||
| #include "IXWebSocketTransport.h" | ||||
| #include "IXWebSocketHandshake.h" | ||||
| #include "IXWebSocketHttpHeaders.h" | ||||
| #include "IXUrlParser.h" | ||||
| #include "IXSocketFactory.h" | ||||
|  | ||||
| #ifdef IXWEBSOCKET_USE_TLS | ||||
| # ifdef __APPLE__ | ||||
| @@ -70,31 +72,31 @@ namespace ix | ||||
|     { | ||||
|         std::string protocol, host, path, query; | ||||
|         int port; | ||||
|         bool websocket = true; | ||||
|  | ||||
|         if (!WebSocketHandshake::parseUrl(url, protocol, host, | ||||
|                                           path, query, port)) | ||||
|         if (!UrlParser::parse(url, protocol, host, path, query, port, websocket)) | ||||
|         { | ||||
|             return WebSocketInitResult(false, 0, | ||||
|                                        std::string("Could not parse URL ") + url); | ||||
|         } | ||||
|  | ||||
|         if (protocol == "wss") | ||||
|         if (protocol != "ws" && protocol != "wss") | ||||
|         { | ||||
|             _socket.reset(); | ||||
| #ifdef IXWEBSOCKET_USE_TLS | ||||
| # ifdef __APPLE__ | ||||
|              _socket = std::make_shared<SocketAppleSSL>(); | ||||
| # else | ||||
|              _socket = std::make_shared<SocketOpenSSL>(); | ||||
| # endif | ||||
| #else | ||||
|             return WebSocketInitResult(false, 0, "TLS is not supported."); | ||||
| #endif | ||||
|             std::stringstream ss; | ||||
|             ss << "Invalid protocol: " << protocol | ||||
|                << " for url " << url | ||||
|                << " . Supported protocols are ws and wss"; | ||||
|  | ||||
|             return WebSocketInitResult(false, 0, ss.str()); | ||||
|         } | ||||
|         else | ||||
|  | ||||
|         bool tls = protocol == "wss"; | ||||
|         std::string errorMsg; | ||||
|         _socket = createSocket(tls, errorMsg); | ||||
|  | ||||
|         if (!_socket) | ||||
|         { | ||||
|             _socket.reset(); | ||||
|             _socket = std::make_shared<Socket>(); | ||||
|             return WebSocketInitResult(false, 0, errorMsg); | ||||
|         } | ||||
|  | ||||
|         WebSocketHandshake webSocketHandshake(_requestInitCancellation, | ||||
|   | ||||
							
								
								
									
										2
									
								
								makefile
									
									
									
									
									
								
							
							
						
						
									
										2
									
								
								makefile
									
									
									
									
									
								
							| @@ -15,7 +15,7 @@ run: | ||||
|  | ||||
| # this is helpful to remove trailing whitespaces | ||||
| trail: | ||||
| 	sh third_party/remove_trailing_whitespaces.sh | ||||
| 	sh third_party/remote_trailing_whitespaces.sh | ||||
|  | ||||
| build: | ||||
| 	(cd examples/satori_publisher ; mkdir -p build ; cd build ; cmake .. ; make) | ||||
|   | ||||
| @@ -23,6 +23,7 @@ add_executable(ws | ||||
|   ixcrypto/IXHash.cpp | ||||
|   ixcrypto/IXUuid.cpp | ||||
|  | ||||
|   ws_http_client.cpp | ||||
|   ws_ping_pong.cpp | ||||
|   ws_broadcast_server.cpp | ||||
|   ws_echo_server.cpp | ||||
|   | ||||
| @@ -6,6 +6,7 @@ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| #include <cstdint> | ||||
| #include <vector> | ||||
|  | ||||
| namespace ix | ||||
|   | ||||
							
								
								
									
										66
									
								
								ws/ws.cpp
									
									
									
									
									
								
							
							
						
						
									
										66
									
								
								ws/ws.cpp
									
									
									
									
									
								
							| @@ -1,9 +1,14 @@ | ||||
| /* | ||||
|  *  ws.cpp | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2017-2018 Machine Zone, Inc. All rights reserved. | ||||
|  *  Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| // | ||||
| // Main driver for websocket utilities | ||||
| // | ||||
| #include "ws.h" | ||||
|  | ||||
| // | ||||
| // Main drive for websocket utilities | ||||
| // | ||||
| @@ -15,28 +20,6 @@ | ||||
| #include <cli11/CLI11.hpp> | ||||
| #include <ixwebsocket/IXSocket.h> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     int ws_ping_pong_main(const std::string& url); | ||||
|  | ||||
|     int ws_echo_server_main(int port); | ||||
|  | ||||
|     int ws_broadcast_server_main(int port); | ||||
|  | ||||
|     int ws_chat_main(const std::string& url, | ||||
|                      const std::string& user); | ||||
|  | ||||
|     int ws_connect_main(const std::string& url); | ||||
|  | ||||
|     int ws_receive_main(const std::string& url, | ||||
|                         bool enablePerMessageDeflate); | ||||
|  | ||||
|     int ws_transfer_main(int port); | ||||
|  | ||||
|     int ws_send_main(const std::string& url, | ||||
|                      const std::string& path); | ||||
| } | ||||
|  | ||||
| int main(int argc, char** argv) | ||||
| { | ||||
|     CLI::App app{"ws is a websocket tool"}; | ||||
| @@ -45,12 +28,23 @@ int main(int argc, char** argv) | ||||
|     std::string url("ws://127.0.0.1:8080"); | ||||
|     std::string path; | ||||
|     std::string user; | ||||
|     std::string data; | ||||
|     std::string headers; | ||||
|     std::string output; | ||||
|     bool headersOnly = false; | ||||
|     bool followRedirects = false; | ||||
|     bool verbose = false; | ||||
|     bool save = false; | ||||
|     bool compress = false; | ||||
|     int port = 8080; | ||||
|     int connectTimeOut = 60; | ||||
|     int transferTimeout = 1800; | ||||
|     int maxRedirects = 5; | ||||
|  | ||||
|     CLI::App* sendApp = app.add_subcommand("send", "Send a file"); | ||||
|     sendApp->add_option("url", url, "Connection url")->required(); | ||||
|     sendApp->add_option("path", path, "Path to the file to send")->required(); | ||||
|  | ||||
|     sendApp->add_option("path", path, "Path to the file to send") | ||||
|         ->required()->check(CLI::ExistingPath); | ||||
|     CLI::App* receiveApp = app.add_subcommand("receive", "Receive a file"); | ||||
|     receiveApp->add_option("url", url, "Connection url")->required(); | ||||
|  | ||||
| @@ -73,6 +67,21 @@ int main(int argc, char** argv) | ||||
|     CLI::App* pingPongApp = app.add_subcommand("ping", "Ping pong"); | ||||
|     pingPongApp->add_option("url", url, "Connection url")->required(); | ||||
|  | ||||
|     CLI::App* httpClientApp = app.add_subcommand("curl", "HTTP Client"); | ||||
|     httpClientApp->add_option("url", url, "Connection url")->required(); | ||||
|     httpClientApp->add_option("-d", data, "Form data")->join(); | ||||
|     httpClientApp->add_option("-F", data, "Form data")->join(); | ||||
|     httpClientApp->add_option("-H", headers, "Header")->join(); | ||||
|     httpClientApp->add_option("--output", output, "Output file"); | ||||
|     httpClientApp->add_flag("-I", headersOnly, "Send a HEAD request"); | ||||
|     httpClientApp->add_flag("-L", followRedirects, "Follow redirects"); | ||||
|     httpClientApp->add_option("--max-redirects", maxRedirects, "Max Redirects"); | ||||
|     httpClientApp->add_flag("-v", verbose, "Verbose"); | ||||
|     httpClientApp->add_flag("-O", save, "Save output to disk"); | ||||
|     httpClientApp->add_flag("--compress", compress, "Enable gzip compression"); | ||||
|     httpClientApp->add_option("--connect-timeout", connectTimeOut, "Connection timeout"); | ||||
|     httpClientApp->add_option("--transfer-timeout", transferTimeout, "Transfer timeout"); | ||||
|  | ||||
|     CLI11_PARSE(app, argc, argv); | ||||
|  | ||||
|     ix::Socket::init(); | ||||
| @@ -110,9 +119,12 @@ int main(int argc, char** argv) | ||||
|     { | ||||
|         return ix::ws_ping_pong_main(url); | ||||
|     } | ||||
|     else | ||||
|     else if (app.got_subcommand("curl")) | ||||
|     { | ||||
|         assert(false); | ||||
|         return ix::ws_http_client_main(url, headers, data, headersOnly, | ||||
|                                        connectTimeOut, transferTimeout, | ||||
|                                        followRedirects, maxRedirects, verbose, | ||||
|                                        save, output, compress); | ||||
|     } | ||||
|  | ||||
|     return 1; | ||||
|   | ||||
							
								
								
									
										43
									
								
								ws/ws.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										43
									
								
								ws/ws.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,43 @@ | ||||
| /* | ||||
|  *  ws.h | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
| #pragma once | ||||
|  | ||||
| #include <string> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     int ws_http_client_main(const std::string& url, | ||||
|                             const std::string& headers, | ||||
|                             const std::string& data, | ||||
|                             bool headersOnly, | ||||
|                             int connectTimeout, | ||||
|                             int transferTimeout, | ||||
|                             bool followRedirects, | ||||
|                             int maxRedirects, | ||||
|                             bool verbose, | ||||
|                             bool save, | ||||
|                             const std::string& output, | ||||
|                             bool compress); | ||||
|  | ||||
|     int ws_ping_pong_main(const std::string& url); | ||||
|  | ||||
|     int ws_echo_server_main(int port); | ||||
|  | ||||
|     int ws_broadcast_server_main(int port); | ||||
|  | ||||
|     int ws_chat_main(const std::string& url, | ||||
|                      const std::string& user); | ||||
|  | ||||
|     int ws_connect_main(const std::string& url); | ||||
|  | ||||
|     int ws_receive_main(const std::string& url, | ||||
|                         bool enablePerMessageDeflate); | ||||
|  | ||||
|     int ws_transfer_main(int port); | ||||
|  | ||||
|     int ws_send_main(const std::string& url, | ||||
|                      const std::string& path); | ||||
| } | ||||
| @@ -5,9 +5,10 @@ | ||||
|  */ | ||||
|  | ||||
| // | ||||
| // Simple chat program that talks to the node.js server at | ||||
| // websocket_chat_server/broacast-server.js | ||||
| //  | ||||
| // Simple chat program that talks to a broadcast server | ||||
| // Broadcast server can be ran with `ws broadcast_server` | ||||
| // | ||||
|  | ||||
| #include <iostream> | ||||
| #include <sstream> | ||||
| #include <queue> | ||||
|   | ||||
							
								
								
									
										183
									
								
								ws/ws_http_client.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										183
									
								
								ws/ws_http_client.cpp
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,183 @@ | ||||
| /* | ||||
|  *  http_client.cpp | ||||
|  *  Author: Benjamin Sergeant | ||||
|  *  Copyright (c) 2019 Machine Zone, Inc. All rights reserved. | ||||
|  */ | ||||
|  | ||||
| #include <iostream> | ||||
| #include <sstream> | ||||
| #include <fstream> | ||||
| #include <ixwebsocket/IXHttpClient.h> | ||||
| #include <ixwebsocket/IXWebSocketHttpHeaders.h> | ||||
|  | ||||
| namespace ix | ||||
| { | ||||
|     std::string extractFilename(const std::string& path) | ||||
|     { | ||||
|         std::string::size_type idx; | ||||
|  | ||||
|         idx = path.rfind('/'); | ||||
|         if (idx != std::string::npos) | ||||
|         { | ||||
|             std::string filename = path.substr(idx+1); | ||||
|             return filename; | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|             return path; | ||||
|         } | ||||
|     } | ||||
|  | ||||
|     WebSocketHttpHeaders parseHeaders(const std::string& data) | ||||
|     { | ||||
|         WebSocketHttpHeaders headers; | ||||
|  | ||||
|         // Split by \n | ||||
|         std::string token; | ||||
|         std::stringstream tokenStream(data); | ||||
|  | ||||
|         while (std::getline(tokenStream, token)) | ||||
|         { | ||||
|             std::size_t pos = token.rfind(':'); | ||||
|  | ||||
|             // Bail out if last '.' is found | ||||
|             if (pos == std::string::npos) continue; | ||||
|  | ||||
|             auto key = token.substr(0, pos); | ||||
|             auto val = token.substr(pos+2); | ||||
|  | ||||
|             std::cerr << key << ": " << val << std::endl; | ||||
|             headers[key] = val; | ||||
|         } | ||||
|  | ||||
|         return headers; | ||||
|     } | ||||
|  | ||||
|     // | ||||
|     // Useful endpoint to test HTTP post | ||||
|     // https://postman-echo.com/post | ||||
|     // | ||||
|     HttpParameters parsePostParameters(const std::string& data) | ||||
|     { | ||||
|         HttpParameters httpParameters; | ||||
|  | ||||
|         // Split by \n | ||||
|         std::string token; | ||||
|         std::stringstream tokenStream(data); | ||||
|  | ||||
|         while (std::getline(tokenStream, token)) | ||||
|         { | ||||
|             std::size_t pos = token.rfind('='); | ||||
|  | ||||
|             // Bail out if last '.' is found | ||||
|             if (pos == std::string::npos) continue; | ||||
|  | ||||
|             auto key = token.substr(0, pos); | ||||
|             auto val = token.substr(pos+1); | ||||
|  | ||||
|             std::cerr << key << ": " << val << std::endl; | ||||
|             httpParameters[key] = val; | ||||
|         } | ||||
|  | ||||
|         return httpParameters; | ||||
|     } | ||||
|  | ||||
|     int ws_http_client_main(const std::string& url, | ||||
|                             const std::string& headersData, | ||||
|                             const std::string& data, | ||||
|                             bool headersOnly, | ||||
|                             int connectTimeout, | ||||
|                             int transferTimeout, | ||||
|                             bool followRedirects, | ||||
|                             int maxRedirects, | ||||
|                             bool verbose, | ||||
|                             bool save, | ||||
|                             const std::string& output, | ||||
|                             bool compress) | ||||
|     { | ||||
|         HttpRequestArgs args; | ||||
|         args.extraHeaders = parseHeaders(headersData); | ||||
|         args.connectTimeout = connectTimeout; | ||||
|         args.transferTimeout = transferTimeout; | ||||
|         args.followRedirects = followRedirects; | ||||
|         args.maxRedirects = maxRedirects; | ||||
|         args.verbose = verbose; | ||||
|         args.compress = compress; | ||||
|         args.logger = [](const std::string& msg) | ||||
|         { | ||||
|             std::cout << msg; | ||||
|         }; | ||||
|  | ||||
|         HttpParameters httpParameters = parsePostParameters(data); | ||||
|  | ||||
|         HttpClient httpClient; | ||||
|         HttpResponse out; | ||||
|         if (headersOnly) | ||||
|         { | ||||
|             out = httpClient.head(url, args); | ||||
|         } | ||||
|         else if (data.empty()) | ||||
|         { | ||||
|             out = httpClient.get(url, args); | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|             out = httpClient.post(url, httpParameters, args); | ||||
|         } | ||||
|  | ||||
|         auto statusCode = std::get<0>(out); | ||||
|         auto errorCode = std::get<1>(out); | ||||
|         auto responseHeaders = std::get<2>(out); | ||||
|         auto payload = std::get<3>(out); | ||||
|         auto errorMsg = std::get<4>(out); | ||||
|         auto uploadSize = std::get<5>(out); | ||||
|         auto downloadSize = std::get<6>(out); | ||||
|  | ||||
|         for (auto it : responseHeaders) | ||||
|         { | ||||
|             std::cerr << it.first << ": " << it.second << std::endl; | ||||
|         } | ||||
|  | ||||
|         std::cerr << "Upload size: " << uploadSize << std::endl; | ||||
|         std::cerr << "Download size: " << downloadSize << std::endl; | ||||
|  | ||||
|         std::cerr << "Status: " << statusCode << std::endl; | ||||
|         if (errorCode != HttpErrorCode_Ok) | ||||
|         { | ||||
|             std::cerr << "error message: " << errorMsg << std::endl; | ||||
|         } | ||||
|  | ||||
|         if (!headersOnly && errorCode == HttpErrorCode_Ok) | ||||
|         { | ||||
|             if (save || !output.empty()) | ||||
|             { | ||||
|                 // FIMXE we should decode the url first | ||||
|                 std::string filename = extractFilename(url); | ||||
|                 if (!output.empty()) | ||||
|                 { | ||||
|                     filename = output; | ||||
|                 } | ||||
|  | ||||
|                 std::cout << "Writing to disk: " << filename << std::endl; | ||||
|                 std::ofstream out(filename); | ||||
|                 out.write((char*)&payload.front(), payload.size()); | ||||
|                 out.close(); | ||||
|             } | ||||
|             else | ||||
|             { | ||||
|                 if (responseHeaders["Content-Type"] != "application/octet-stream") | ||||
|                 { | ||||
|                     std::cout << "payload: " << payload << std::endl; | ||||
|                 } | ||||
|                 else | ||||
|                 { | ||||
|                     std::cerr << "Binary output can mess up your terminal." << std::endl; | ||||
|                     std::cerr << "Use the -O flag to save the file to disk." << std::endl; | ||||
|                     std::cerr << "You can also use the --output option to specify a filename." << std::endl; | ||||
|                 } | ||||
|             } | ||||
|         } | ||||
|  | ||||
|         return 0; | ||||
|     } | ||||
| } | ||||
		Reference in New Issue
	
	Block a user