|
@@ -9,20 +9,7 @@
|
|
#include <stdio.h>
|
|
#include <stdio.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/socket.h>
|
|
#include <stdlib.h>
|
|
#include <stdlib.h>
|
|
-#include <netinet/in.h>
|
|
|
|
-#include <string.h>
|
|
|
|
-#include <arpa/inet.h>
|
|
|
|
-#include <unistd.h>
|
|
|
|
-
|
|
|
|
-struct socket_exception : public std::exception{
|
|
|
|
- std::string msg;
|
|
|
|
- socket_exception(std::string&& _msg){
|
|
|
|
- msg = std::move(_msg);
|
|
|
|
- }
|
|
|
|
- const char* what() const throw (){
|
|
|
|
- return msg.length() == 0 ? "Connection creation failure" : msg.c_str();
|
|
|
|
- }
|
|
|
|
-};
|
|
|
|
|
|
+#include <netinet/in.h>
|
|
class cppsocket{
|
|
class cppsocket{
|
|
private:
|
|
private:
|
|
struct sockaddr_in serv_addr;
|
|
struct sockaddr_in serv_addr;
|
|
@@ -30,59 +17,13 @@ private:
|
|
std::vector<char> buffer;
|
|
std::vector<char> buffer;
|
|
public:
|
|
public:
|
|
const static constexpr unsigned int buffersize = 16;
|
|
const static constexpr unsigned int buffersize = 16;
|
|
- cppsocket(sockaddr_in _serv_addr,int _sock) : sock(_sock), serv_addr(_serv_addr){
|
|
|
|
- buffer = std::vector<char>(buffersize + 1,0);
|
|
|
|
- }
|
|
|
|
- cppsocket(){};
|
|
|
|
- cppsocket(const std::string& addr, unsigned int PORT){
|
|
|
|
- struct sockaddr_in address;
|
|
|
|
- sock = 0;
|
|
|
|
- int valread;
|
|
|
|
- buffer = std::vector<char>(buffersize + 1, 0);
|
|
|
|
-
|
|
|
|
- if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
|
|
|
|
- printf("\n Socket creation error \n");
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- memset(&serv_addr, '0', sizeof(serv_addr));
|
|
|
|
- serv_addr.sin_family = AF_INET;
|
|
|
|
- serv_addr.sin_port = htons(PORT);
|
|
|
|
-
|
|
|
|
- if(inet_pton(AF_INET, addr.c_str(), &serv_addr.sin_addr)<=0){
|
|
|
|
- throw std::invalid_argument("Invalid address: " + addr);
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) {
|
|
|
|
- throw socket_exception("Could not reach server " + addr);
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- void write(const std::string& message){
|
|
|
|
- for(size_t i = 0;i < message.length();i += buffersize){
|
|
|
|
- char cs[buffersize + 1] = {0};
|
|
|
|
- std::memcpy(cs, message.c_str() + i,buffersize);
|
|
|
|
- if((i + buffersize) < message.length()){
|
|
|
|
- cs[buffersize] = 'c';
|
|
|
|
- }
|
|
|
|
- else{
|
|
|
|
- cs[buffersize] = (char)0;
|
|
|
|
- }
|
|
|
|
- send(sock, message.c_str(), message.size(), 0);
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- std::vector<char> receive(){
|
|
|
|
- std::vector<char> stor;
|
|
|
|
- while(true){
|
|
|
|
- std::fill(buffer.begin(), buffer.end(), (char)0);
|
|
|
|
- size_t val = read(sock, buffer.data(), buffersize + 1);
|
|
|
|
- if(val == 0)throw socket_exception("Connection reset by peer");
|
|
|
|
- stor.insert(stor.end(),buffer.begin(),buffer.begin() + std::min(val, (size_t)buffersize));
|
|
|
|
- if(buffer.data()[buffersize] == (char)0)break;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- return std::move(stor);
|
|
|
|
- }
|
|
|
|
|
|
+ cppsocket(sockaddr_in _serv_addr,int _sock);
|
|
|
|
+ cppsocket();
|
|
|
|
+ cppsocket(const std::string& addr, unsigned int PORT);
|
|
|
|
+ void write(const std::string& message);
|
|
|
|
+ std::vector<char> receive();
|
|
};
|
|
};
|
|
|
|
+
|
|
class server_socket{
|
|
class server_socket{
|
|
private:
|
|
private:
|
|
int port_;
|
|
int port_;
|
|
@@ -90,34 +31,9 @@ private:
|
|
struct sockaddr_in address;
|
|
struct sockaddr_in address;
|
|
socklen_t addrlen;
|
|
socklen_t addrlen;
|
|
public:
|
|
public:
|
|
- int port(){return port_;}
|
|
|
|
- server_socket(int _port) : port_(_port){
|
|
|
|
- int opt = 1;
|
|
|
|
- addrlen = sizeof(address);
|
|
|
|
- if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0) {
|
|
|
|
- throw socket_exception("server_socket creation failed");
|
|
|
|
- }
|
|
|
|
- if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt))) {
|
|
|
|
- throw socket_exception("server_socket creation failed");
|
|
|
|
- }
|
|
|
|
- address.sin_family = AF_INET;
|
|
|
|
- address.sin_addr.s_addr = INADDR_ANY;
|
|
|
|
- address.sin_port = htons(port());
|
|
|
|
- if(bind(server_fd, (struct sockaddr *)&address, sizeof(address))<0){
|
|
|
|
- throw socket_exception("Couldn't bind to port " + std::to_string(port()) + ", is it already in use?");
|
|
|
|
- }
|
|
|
|
- if (listen(server_fd, 16) < 0){
|
|
|
|
- throw socket_exception("Error in listen() call, no clue why");
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- cppsocket accept_connection(){
|
|
|
|
- int new_socket;
|
|
|
|
- sockaddr client_addr;
|
|
|
|
- if ((new_socket = accept(server_fd, (struct sockaddr *)&client_addr,(socklen_t*)&addrlen))<0){
|
|
|
|
- throw socket_exception("Socket accept failed");
|
|
|
|
- }
|
|
|
|
- return cppsocket(*((sockaddr_in*)&client_addr), new_socket);
|
|
|
|
- }
|
|
|
|
|
|
+ int port();
|
|
|
|
+ server_socket(int _port);
|
|
|
|
+ cppsocket accept_connection();
|
|
};
|
|
};
|
|
|
|
|
|
|
|
|