move to new cspot

components/spotify/cspot/bell/main/io/include/TCPSocket.h

@@ -0,0 +1,121 @@
+#ifndef BELL_BASIC_SOCKET_H
+#define BELL_BASIC_SOCKET_H
+
+#include <ctype.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <cstring>
+#include <iostream>
+#include <memory>
+#include <string>
+#include <vector>
+#include "BellSocket.h"
+#ifdef _WIN32
+#include <winsock2.h>
+#include <ws2tcpip.h>
+#include "win32shim.h"
+#else
+#include <netdb.h>
+#include <netinet/in.h>
+#include <netinet/tcp.h>
+#include <sys/ioctl.h>
+#include <sys/socket.h>
+#include <unistd.h>
+#ifdef __sun
+#include <sys/filio.h>
+#endif
+#endif
+#include <BellLogger.h>
+#include <fstream>
+#include <sstream>
+
+namespace bell {
+class TCPSocket : public bell::Socket {
+ private:
+  int sockFd;
+  bool isClosed = true;
+
+ public:
+  TCPSocket(){};
+  ~TCPSocket() { close(); };
+
+  void open(const std::string& host, uint16_t port) {
+    int err;
+    int domain = AF_INET;
+    int socketType = SOCK_STREAM;
+
+    struct addrinfo hints {
+    }, *addr;
+    //fine-tune hints according to which socket you want to open
+    hints.ai_family = domain;
+    hints.ai_socktype = socketType;
+    hints.ai_protocol =
+        IPPROTO_IP;  // no enum : possible value can be read in /etc/protocols
+    hints.ai_flags = AI_CANONNAME | AI_ALL | AI_ADDRCONFIG;
+
+    // BELL_LOG(info, "http", "%s %d", host.c_str(), port);
+
+    char portStr[6];
+    sprintf(portStr, "%u", port);
+    err = getaddrinfo(host.c_str(), portStr, &hints, &addr);
+    if (err != 0) {
+      throw std::runtime_error("Resolve failed");
+    }
+
+    sockFd = socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol);
+
+    err = connect(sockFd, addr->ai_addr, addr->ai_addrlen);
+    if (err < 0) {
+      close();
+      BELL_LOG(error, "http", "Could not connect to %s. Error %d", host.c_str(),
+               errno);
+      throw std::runtime_error("Resolve failed");
+    }
+
+    int flag = 1;
+    setsockopt(sockFd,       /* socket affected */
+               IPPROTO_TCP,  /* set option at TCP level */
+               TCP_NODELAY,  /* name of option */
+               (char*)&flag, /* the cast is historical cruft */
+               sizeof(int)); /* length of option value */
+
+    freeaddrinfo(addr);
+    isClosed = false;
+  }
+
+  size_t read(uint8_t* buf, size_t len) {
+    return recv(sockFd, (char*)buf, len, 0);
+  }
+
+  size_t write(uint8_t* buf, size_t len) {
+    return send(sockFd, (char*)buf, len, 0);
+  }
+
+  size_t poll() {
+#ifdef _WIN32
+    unsigned long value;
+    ioctlsocket(sockFd, FIONREAD, &value);
+#else
+    int value;
+    ioctl(sockFd, FIONREAD, &value);
+#endif
+    return value;
+  }
+  bool isOpen() { return !isClosed; }
+
+  void close() {
+    if (!isClosed) {
+#ifdef _WIN32
+      closesocket(sockFd);
+#else
+      ::close(sockFd);
+#endif
+      sockFd = -1;
+      isClosed = true;
+    }
+  }
+};
+
+}  // namespace bell
+
+#endif