Initial commit.

2020-06-01 13:22:04 -07:00
parent 5d000e3d53
commit c1987ceb38
12 changed files with 1519 additions and 1 deletions
--- a/lms.cpp
+++ b/lms.cpp
@@ -0,0 +1,88 @@
+/*
+Copyright 2019 Intel Corporation
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+#include "lms.h"
+#include <string.h>
+
+#ifdef _WIN32
+// Windows
+#include <Ws2tcpip.h>
+#else
+// Linux
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#endif
+
+SOCKET lmsConnect()
+{
+    std::string lmsAddress = "localhost";
+    std::string lmsPort = "16992";
+    SOCKET s = INVALID_SOCKET;
+    struct addrinfo *addr, hints;
+
+#ifdef _WIN32
+    WSADATA wsa;
+    if (WSAStartup(MAKEWORD(2, 2), &wsa) != 0)
+    {
+        throw std::runtime_error("error: unable to connect to LMS");
+    }
+#endif
+
+    memset(&hints, 0, sizeof(hints));
+    
+    hints.ai_family = AF_UNSPEC;
+    hints.ai_socktype = SOCK_STREAM;
+    hints.ai_protocol = IPPROTO_TCP;
+
+    if (getaddrinfo(lmsAddress.c_str(), lmsPort.c_str(), &hints, &addr) != 0)
+    {
+        throw std::runtime_error("error: unable to connect to LMS");
+    }
+
+    if (addr == NULL)
+    {
+        throw std::runtime_error("error: unable to connect to LMS");
+    }
+
+    for (addr; addr != NULL; addr = addr->ai_next)
+    {
+        s = socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol);
+        if (s == INVALID_SOCKET)
+        {
+            continue;
+        }
+
+        if (connect(s, addr->ai_addr, (int)addr->ai_addrlen) == 0)
+        {
+            break;
+        }
+
+        closesocket(s);
+        s = INVALID_SOCKET;
+    }
+
+    if (addr != NULL)
+    {
+        freeaddrinfo(addr);
+    }
+
+    if (s == INVALID_SOCKET)
+    {
+        throw std::runtime_error("error: unable to connect to LMS");
+    }
+
+    return s;
+}