30.11.2010 - initial commit

piethernet.cpp

@@ -0,0 +1,128 @@
+#include "piethernet.h"
+
+
+PIEthernet::PIEthernet(string ip, int port, void * data_, EthernetFunc slot): PIThread() {
+	setPriority(piHigh);
+	data = data_;
+	ip_ = ip_s = ip;
+	port_ = port_s = port;
+	sock = sock_s = -1;
+	ret_func = slot;
+	buffer = new char[BUFFER_SIZE];
+#ifdef WINDOWS
+	WSADATA wsaData;
+	WSAStartup(MAKEWORD(2, 2), &wsaData);
+#endif
+	sock = socket(PF_INET, SOCK_DGRAM, 0);
+	sock_s = socket(PF_INET, SOCK_DGRAM, 0);
+	if (sock == -1 || sock_s == -1)
+		cout << "[PIEthernet] Unable to create socket" << endl;
+}
+
+
+PIEthernet::~PIEthernet() {
+	if (isRunning()) {
+		stop();
+		pthread_cancel(thread);
+	}
+	if (sock != -1) {
+		shutdown(sock, SHUT_RDWR);
+		close(sock);
+	}
+	if (sock_s != -1) {
+		shutdown(sock_s, SHUT_RDWR);
+		close(sock_s);
+	}
+#ifdef WINDOWS
+	WSACleanup();
+#endif
+	delete buffer;
+}
+
+
+void PIEthernet::begin() {
+	if (!init()) stop();
+}
+
+
+void PIEthernet::run() {
+#ifdef WINDOWS
+	int addr_len;
+#else
+	socklen_t addr_len;
+#endif
+	//cout << "[PIEthernet] reading ... ";
+	readed = recvfrom(sock, buffer, BUFFER_SIZE, 0, (sockaddr * )&addr_, &addr_len);
+	//cout << WSAGetLastError() << endl;
+	if (readed < 0) {
+		//cout << "[PIEthernet] Error while reading" << endl;
+		//stop();
+		//init();
+		return;
+	}
+	if (ret_func != 0) ret_func(data, buffer);
+}
+
+
+void PIEthernet::end() {
+	if (sock != -1) {
+		shutdown(sock, SHUT_RDWR);
+		close(sock);
+		sock = -1;
+	}
+	if (sock_s != -1) {
+		shutdown(sock_s, SHUT_RDWR);
+		close(sock_s);
+		sock_s = -1;
+	}
+	//cout << "[PIEthernet] end" << endl;
+}
+
+
+bool PIEthernet::init() {
+	addr_.sin_addr.s_addr = inet_addr(ip_.c_str());
+	addr_.sin_family = PF_INET;
+	addr_.sin_port = htons(port_);
+	if (bind(sock, (sockaddr * )&addr_, sizeof(addr_)) == -1) {
+		cout << "[PIEthernet] Cant`t bind to " << ip_ << ":" << port_ << endl;
+		return false;
+	}
+	return true;
+}
+
+
+bool PIEthernet::send(string ip, int port, char * data, int size) {
+	if (sock_s == -1) {
+		//cout << "[PIEthernet] Can`t send to uninitialized socket" << endl;
+		return false;
+	}
+	saddr_.sin_port = htons(port);
+	saddr_.sin_addr.s_addr = inet_addr(ip.c_str());
+	saddr_.sin_family = PF_INET;
+	wrote = sendto(sock_s, data, size, 0, (sockaddr * )&saddr_, sizeof(saddr_));
+	if (wrote != size) {
+		//cout << "[PIEthernet] Error while sending" << endl;
+		return false;
+	}
+	//cout << "[PIEthernet] Wrote " << wrote << " bytes in " << devName << endl;
+	return true;
+}
+
+
+bool PIEthernet::send(char * data, int size) {
+	if (sock_s == -1) {
+		//cout << "[PIEthernet] Can`t send to uninitialized socket" << endl;
+		return false;
+	}
+	saddr_.sin_port = htons(port_s);
+	saddr_.sin_addr.s_addr = inet_addr(ip_s.data());
+	saddr_.sin_family = PF_INET;
+	//cout << "[PIEthernet] sending in " << sock_s << endl;
+	wrote = sendto(sock_s, data, size, 0, (sockaddr * )&saddr_, sizeof(saddr_));
+	if (wrote != size) {
+		//cout << "[PIEthernet] Error while sending" << endl;
+		return false;
+	}
+	//cout << "[PIEthernet] Wrote " << wrote << " bytes" << endl;
+	return true;
+}