mirror of https://github.com/nextcloud/desktop
130 lines
2.8 KiB
C++
130 lines
2.8 KiB
C++
/**
|
|
* Copyright (c) 2000-2013 Liferay, Inc. All rights reserved.
|
|
*
|
|
* This library is free software; you can redistribute it and/or modify it under
|
|
* the terms of the GNU Lesser General Public License as published by the Free
|
|
* Software Foundation; either version 2.1 of the License, or (at your option)
|
|
* any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
* FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
|
|
* details.
|
|
*/
|
|
|
|
#include "CommunicationSocket.h"
|
|
#include "UtilConstants.h"
|
|
#include "StringUtil.h"
|
|
|
|
#include <WinSock2.h>
|
|
#include <Ws2def.h>
|
|
#include <windows.h>
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
#include <fstream>
|
|
|
|
#define BUFSIZE 1024
|
|
|
|
using namespace std;
|
|
|
|
#define DEFAULT_BUFLEN 4096
|
|
|
|
CommunicationSocket::CommunicationSocket(int port)
|
|
: _port(port), _clientSocket(INVALID_SOCKET)
|
|
{
|
|
}
|
|
|
|
CommunicationSocket::~CommunicationSocket()
|
|
{
|
|
Close();
|
|
}
|
|
|
|
bool CommunicationSocket::Close()
|
|
{
|
|
WSACleanup();
|
|
bool closed = (closesocket(_clientSocket) == 0);
|
|
shutdown(_clientSocket, SD_BOTH);
|
|
_clientSocket = INVALID_SOCKET;
|
|
return closed;
|
|
}
|
|
|
|
|
|
bool CommunicationSocket::Connect()
|
|
{
|
|
WSADATA wsaData;
|
|
|
|
HRESULT iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
|
|
|
|
if (iResult != NO_ERROR) {
|
|
int error = WSAGetLastError();
|
|
}
|
|
|
|
|
|
_clientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
|
|
|
if (_clientSocket == INVALID_SOCKET) {
|
|
//int error = WSAGetLastError();
|
|
Close();
|
|
return false;
|
|
}
|
|
|
|
struct sockaddr_in clientService;
|
|
|
|
clientService.sin_family = AF_INET;
|
|
clientService.sin_addr.s_addr = inet_addr(PLUG_IN_SOCKET_ADDRESS);
|
|
clientService.sin_port = htons(_port);
|
|
|
|
iResult = connect(_clientSocket, (SOCKADDR*)&clientService, sizeof(clientService));
|
|
DWORD timeout = 500; // ms
|
|
setsockopt(_clientSocket, SOL_SOCKET, SO_RCVTIMEO, (const char*) &timeout, sizeof(DWORD));
|
|
|
|
if (iResult == SOCKET_ERROR) {
|
|
//int error = WSAGetLastError();
|
|
Close();
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool CommunicationSocket::SendMsg(const wchar_t* message)
|
|
{
|
|
const char* utf8_msg = StringUtil::toUtf8(message);
|
|
size_t result = send(_clientSocket, utf8_msg, (int)strlen(utf8_msg), 0);
|
|
delete[] utf8_msg;
|
|
|
|
if (result == SOCKET_ERROR) {
|
|
//int error = WSAGetLastError();
|
|
closesocket(_clientSocket);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
bool CommunicationSocket::ReadLine(wstring* response)
|
|
{
|
|
if (!response) {
|
|
return false;
|
|
}
|
|
|
|
vector<char> resp_utf8;
|
|
char buffer;
|
|
while (true) {
|
|
int bytesRead = recv(_clientSocket, &buffer, 1, 0);
|
|
if (bytesRead <= 0) {
|
|
response = 0;
|
|
return false;
|
|
}
|
|
|
|
if (buffer == '\n') {
|
|
resp_utf8.push_back(0);
|
|
*response = StringUtil::toUtf16(&resp_utf8[0], resp_utf8.size());
|
|
return true;
|
|
} else {
|
|
resp_utf8.push_back(buffer);
|
|
}
|
|
}
|
|
}
|