diff options
Diffstat (limited to 'protocols/WhatsApp/src/WASocketConnection.cpp')
| -rw-r--r-- | protocols/WhatsApp/src/WASocketConnection.cpp | 138 |
1 files changed, 138 insertions, 0 deletions
diff --git a/protocols/WhatsApp/src/WASocketConnection.cpp b/protocols/WhatsApp/src/WASocketConnection.cpp new file mode 100644 index 0000000000..f4e6ef8ae4 --- /dev/null +++ b/protocols/WhatsApp/src/WASocketConnection.cpp @@ -0,0 +1,138 @@ +#include "WASocketConnection.h"
+
+
+HANDLE WASocketConnection::hNetlibUser = NULL;
+
+void WASocketConnection::initNetwork(HANDLE hNetlibUser) throw (WAException) {
+ WASocketConnection::hNetlibUser = hNetlibUser;
+}
+
+void WASocketConnection::quitNetwork() {
+}
+
+WASocketConnection::WASocketConnection(const std::string& dir, int port) throw (WAException)
+{
+ NETLIBOPENCONNECTION noc = {sizeof(noc)};
+ noc.szHost = dir.c_str();
+ noc.wPort = port;
+ noc.flags = NLOCF_V2; // | NLOCF_SSL;
+ this->hConn = (HANDLE) CallService(MS_NETLIB_OPENCONNECTION, reinterpret_cast<WPARAM>(this->hNetlibUser),
+ reinterpret_cast<LPARAM>(&noc));
+ if (this->hConn == NULL)
+ {
+ throw WAException(getLastErrorMsg(), WAException::SOCKET_EX, WAException::SOCKET_EX_OPEN);
+ }
+
+ this->connected = true;
+}
+
+void WASocketConnection::write(int i) {
+ char buffer;
+ buffer = (char) i;
+
+ NETLIBBUFFER nlb;
+ nlb.buf = &buffer;
+ nlb.len = 1;
+ nlb.flags = MSG_NOHTTPGATEWAYWRAP | MSG_NODUMP;
+
+ int result = CallService(MS_NETLIB_SEND, reinterpret_cast<WPARAM>(this->hConn), reinterpret_cast<LPARAM>(&nlb));
+ if (result < 1) {
+ throw WAException(getLastErrorMsg(), WAException::SOCKET_EX, WAException::SOCKET_EX_SEND);
+ }
+}
+
+void WASocketConnection::makeNonBlock() {
+ //if (fcntl(socket->channel, F_SETFL, O_NONBLOCK) == -1) // #TODO !?
+ throw WAException("Error setting socket nonblocking!", WAException::SOCKET_EX, WAException::SOCKET_EX_OPEN);
+}
+
+int WASocketConnection::waitForRead() {
+ // #TODO Is this called at all?
+ return 0;
+
+ fd_set rfds;
+ struct timeval tv;
+ struct timeval* tvp;
+ int fd = 0;
+
+ FD_ZERO(&rfds);
+ // _LOGDATA("preparando select");
+ //fd = (this->socket)->channel; //#!?
+ // _LOGDATA("socket %d", fd);
+ FD_SET(fd, &rfds);
+ tv.tv_sec = 600; //ApplicationData::SELECT_TIMEOUT;
+ tv.tv_usec = 0; // 5000000;
+ tvp = &tv;
+ //if (ApplicationData::SELECT_TIMEOUT == -1) #TODO
+ // tvp = NULL;
+
+ int retval = select(/*fd + 1*/ 0, &rfds, NULL, NULL, tvp);
+ if (!FD_ISSET(fd, &rfds))
+ retval = 0;
+
+ return retval;
+}
+
+void WASocketConnection::flush() {}
+
+void WASocketConnection::write(const std::vector<unsigned char>& bytes, int offset, int length)
+{
+ NETLIBBUFFER nlb;
+ std::string tmpBuf = std::string(bytes.begin(), bytes.end());
+ nlb.buf = (char*) &(tmpBuf.c_str()[offset]);
+ nlb.len = length;
+ nlb.flags = 0; //MSG_NOHTTPGATEWAYWRAP | MSG_NODUMP;
+
+ int result = CallService(MS_NETLIB_SEND, reinterpret_cast<WPARAM>(this->hConn),
+ reinterpret_cast<LPARAM>(&nlb));
+ if (result < length) {
+ throw WAException(getLastErrorMsg(), WAException::SOCKET_EX, WAException::SOCKET_EX_SEND);
+ }
+}
+
+void WASocketConnection::write(const std::vector<unsigned char>& bytes, int length)
+{
+ this->write(bytes, 0, length);
+}
+
+unsigned char WASocketConnection::read() {
+ char c;
+
+ SetLastError(0);
+ int result;
+ //do {
+ result = Netlib_Recv(this->hConn, &c, 1, 0 /*MSG_NOHTTPGATEWAYWRAP | MSG_NODUMP*/);
+ //} while (WSAGetLastError() == EINTR);
+ if (result <= 0) {
+ throw WAException(getLastErrorMsg(), WAException::SOCKET_EX, WAException::SOCKET_EX_RECV);
+ }
+ return c;
+}
+
+int WASocketConnection::read(std::vector<unsigned char>& b, int off, int length) {
+ if (off < 0 || length < 0) {
+ throw new WAException("Out of bounds", WAException::SOCKET_EX, WAException::SOCKET_EX_RECV);
+ }
+ char* buffer = new char[length];
+ int result = Netlib_Recv(this->hConn, buffer, length, MSG_NOHTTPGATEWAYWRAP | MSG_NODUMP);
+
+ if (result <= 0) {
+ throw WAException(getLastErrorMsg(), WAException::SOCKET_EX, WAException::SOCKET_EX_RECV);
+ }
+
+ for (int i = 0; i < result; i++)
+ b[off + i] = buffer[i];
+
+ delete[] buffer;
+
+ return result;
+}
+
+void WASocketConnection::forceShutdown() {
+ Netlib_Shutdown(this->hConn);
+}
+
+WASocketConnection::~WASocketConnection() {
+ this->forceShutdown();
+ //SDLNet_TCP_Close(this->socket);
+}
\ No newline at end of file |