.

Connection.cpp

@@ -47,22 +47,37 @@ Connection::~Connection(){
 	close(fd);
 }
 
-bool Connection::send(std::vector<Payload> payloads){
+bool Connection::send(std::vector<Payload> payloads, bool size){
+	if(size){
+		unsigned char _size = payloads.size();
+		::send(fd, &_size , sizeof(_size), 0);
+	}
 	for (Payload& p : payloads) {
 		::send(fd, p.getData() , p.getSize(), 0);
 	}
 	return true;
 }
 
+bool Connection::send(Payload payload, bool size){
+	return send(std::vector<Payload>{payload}, size);
+}
+
 Payload Connection::recv(){
 	Payload ret;
-	char temp[100];
+
-	memset(temp, 0, 100);
+	ssize_t size = 0, bytes;
-	ssize_t size;
+	std::vector<unsigned char> received(100);
-	while((size = ::recv(fd, temp, sizeof(temp), 0)) > 0){
+	while((bytes = ::recv(fd, received.data() + size, 100, 0)) > 0){
-		for(char& in: temp){
+		size += bytes;
-			ret.push_char(in);
+		received.resize(size+100);
 	}
+	if(bytes == 0){
+		received.shrink_to_fit();
+		ret.setData(received);
+		return ret;
 	}
+
+	//Error Handling
+	error = ErrorTypes::recv;
 	return ret;
 }

Connection.h

@@ -5,8 +5,6 @@
 #include <vector>
 #include "Payload.h"
 
-class Payload;
-
 class Connection
 {
 private:
@@ -18,7 +16,10 @@ private:
 		socket_creation = 10,
 		get_ip,
 		connect,
-	}error;
+
+		recv = 20,
+		send
+	}error = ErrorTypes::none;
 
 	void p_HandleError();
 
@@ -26,7 +27,8 @@ public:
 	Connection(std::string server_ip, int port);
 	~Connection();
 
-	bool send(std::vector<Payload> payload);
+	bool send(std::vector<Payload> payload, bool size);
+	bool send(Payload payload, bool size);
 	Payload recv();
 
 	/**

Payload.cpp

@@ -1,13 +1,23 @@
 #include "Payload.h"
 
-void Payload::push_char(char load){
+void Payload::push_char(unsigned char load){
 	Data.push_back(load);
 }
 
-char* Payload::getData(){
+unsigned char* Payload::getData(){
 	return Data.data();
 }
 
+void Payload::setData(std::vector<unsigned char> m_data){
+	Data = m_data;
+}
+
 size_t Payload::getSize(){
 	return Data.size();
 }
+
+Payload::Payload(){ }
+
+Payload::Payload(std::vector<unsigned char> m_data)
+: Data(m_data){}
+

Payload.h

@@ -7,15 +7,17 @@
 class Payload
 {
 private:
-	std::vector<char> Data;
+	std::vector<unsigned char> Data;
 public:
 	size_t getSize();
-	char* getData();
+	unsigned char* getData();
 
-	void push_char(char);
+	void setData(std::vector<unsigned char>);
+
+	void push_char(unsigned char);
 
 	Payload();
-	virtual ~Payload();
+	Payload(std::vector<unsigned char>);
 };
 
 #endif /* PAYLOAD_H */

main.cpp

@@ -1,8 +1,29 @@
 #include "Connection.h"
+#include "Payload.h"
+#include <thread>
+#include <iostream>
 
 int main(int argc, char *argv[])
 {
-	Connection c("145.239.73.162", 25565);
+	Connection* c;
+	if(argc == 1){
+		c = new Connection("145.239.73.162", 25565);
+	}else{
+		c = new Connection(argv[1], atoi(argv[2]));
+	}
+	if(c->status() != 0){
+		return 1;
+	}
+	std::vector<unsigned char> p_data{0xFE, 0x01, 0xFA, 0x00, 0x08, 0x00, 0x4D, 0x00, 0x43, 0x00, 0x7C, 0x00, 0x50, 0x00, 0x69, 0x00, 0x6E, 0x00, 0x67, 0x00, 0x48, 0x00, 0x6F, 0x00, 0x73, 0x00, 0x74};
+	Payload p(p_data);
+
+	c->send(p, true);
+	Payload response = c->recv();
+	std::cout << response.getData() << std::endl;
 	
 	return 0;
 }
+
+void test_server(std::string addr, int port, Payload p){
+	Connection c(addr, port);
+}