-
Notifications
You must be signed in to change notification settings - Fork 52
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Implement Read for VISA and add loopback read/write unit test (#1004)
* Add read write support and a loop back system test for read/write. * Add tcp_echo_server * Fix for Linux * Create TcpEchoServer * Follow Dan's comments * Follow Dan's comments * Follow Dan's comment * Fix System tests on Linux * Follow Dan's comments * Fix indention. * Test added by mistake
- Loading branch information
1 parent
7d4dbb1
commit 296e4a3
Showing
3 changed files
with
282 additions
and
16 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,161 @@ | ||
| #include <cstring> | ||
| #include <memory> | ||
| #include <string> | ||
| #include <thread> | ||
| #ifdef _WIN32 | ||
| #include <winsock2.h> | ||
| #else | ||
| #include <arpa/inet.h> | ||
| #include <sys/socket.h> | ||
| #include <netinet/in.h> | ||
| #include <unistd.h> | ||
| #endif | ||
|
|
||
| #ifdef _WIN32 | ||
| typedef int socklen_t; | ||
| #else | ||
| typedef int SOCKET; | ||
| #endif | ||
|
|
||
| class TcpEchoSession : public std::enable_shared_from_this<TcpEchoSession> { | ||
| public: | ||
| TcpEchoSession(SOCKET socket_fd) | ||
| : socket_fd_(socket_fd) | ||
| , stop_session_(false) | ||
| { | ||
| } | ||
|
|
||
| ~TcpEchoSession() { | ||
| stop(); | ||
| #ifdef _WIN32 | ||
| closesocket(socket_fd_); | ||
| #else | ||
| close(socket_fd_); | ||
| #endif | ||
| } | ||
|
|
||
| void start() | ||
| { | ||
| while (!stop_session_) { | ||
| do_read(); | ||
| } | ||
| } | ||
|
|
||
| void stop() | ||
| { | ||
| stop_session_ = true; | ||
| } | ||
|
|
||
| private: | ||
| void do_read() | ||
| { | ||
| auto self(shared_from_this()); | ||
| char buffer[max_length] = {0}; | ||
| int n = recv(socket_fd_, buffer, max_length, 0); | ||
| if (n > 0) { | ||
| do_write(buffer, n); | ||
| } | ||
| } | ||
|
|
||
| void do_write(const char* data, int length) | ||
| { | ||
| auto self(shared_from_this()); | ||
| int n = send(socket_fd_, data, length, 0); | ||
| } | ||
|
|
||
| static const int max_length = 256; | ||
| SOCKET socket_fd_; | ||
| bool stop_session_; | ||
| }; | ||
|
|
||
| class TcpEchoServer | ||
| { | ||
| public: | ||
| TcpEchoServer() | ||
| : server_fd_(-1) | ||
| { | ||
| } | ||
|
|
||
| ~TcpEchoServer() | ||
| { | ||
| } | ||
|
|
||
| int start() | ||
| { | ||
| return start_server_session(); | ||
| } | ||
|
|
||
| void stop() | ||
| { | ||
| close_server_session(); | ||
| } | ||
|
|
||
| int get_server_port() | ||
| { | ||
| struct sockaddr_in assigned_address; | ||
| socklen_t len = sizeof(assigned_address); | ||
| if (getsockname(server_fd_, (struct sockaddr *)&assigned_address, &len) == -1) { | ||
| return -1; | ||
| } | ||
| else { | ||
| return ntohs(assigned_address.sin_port); | ||
| } | ||
| } | ||
|
|
||
| private: | ||
| int start_server_session() | ||
| { | ||
| #ifdef _WIN32 | ||
| WSADATA wsa_data; | ||
| WSAStartup(MAKEWORD(2, 2), &wsa_data); | ||
| #endif | ||
| server_fd_ = socket(AF_INET, SOCK_STREAM, 0); | ||
| if (server_fd_ == -1) { | ||
| return -1; | ||
| } | ||
| struct sockaddr_in server_address; | ||
| std::memset(&server_address, 0, sizeof(server_address)); | ||
| server_address.sin_family = AF_INET; | ||
| server_address.sin_addr.s_addr = inet_addr("127.0.0.1"); | ||
| server_address.sin_port = htons(0); | ||
|
|
||
| if (bind(server_fd_, (struct sockaddr*)&server_address, sizeof(server_address)) != 0) { | ||
| return -1; | ||
| } | ||
| server_thread_ = std::thread(&TcpEchoServer::run_server, this, server_fd_); | ||
| return 0; | ||
| } | ||
|
|
||
| void run_server(SOCKET server_fd) { | ||
| listen(server_fd, 1); | ||
| SOCKET client_fd = accept(server_fd, NULL, NULL); | ||
| if (client_fd == -1) { | ||
| return; | ||
| } | ||
| std::thread([this, client_fd]() { | ||
| handle_client(client_fd); | ||
| }).detach(); | ||
| } | ||
|
|
||
| void close_server_session() | ||
| { | ||
| session_->stop(); | ||
| server_thread_.join(); | ||
| #ifdef _WIN32 | ||
| closesocket(server_fd_); | ||
| WSACleanup(); | ||
| #else | ||
| close(server_fd_); | ||
| #endif | ||
| } | ||
|
|
||
| void handle_client(SOCKET client_fd) | ||
| { | ||
| session_ = std::make_shared<TcpEchoSession>(client_fd); | ||
| session_->start(); | ||
| } | ||
|
|
||
| SOCKET server_fd_; | ||
| std::thread server_thread_; | ||
| std::shared_ptr<TcpEchoSession> session_; | ||
| }; |