more socket stuff

GloryOfNight · Mar 1, 2024 · 4d1aab1 · 4d1aab1
1 parent 3a91198
commit 4d1aab1
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 6 deletions.
diff --git a/src/relay/relay.cxx b/src/relay/relay.cxx
@@ -19,6 +19,10 @@ bool relay::init()
 	m_socket->bind(mainPort);
 	m_socket->setNonBlocking(true);
 
+	int32_t bufferSize{};
+	m_socket->setSendBufferSize(0x20000, bufferSize);
+	m_socket->setRecvBufferSize(0x20000, bufferSize);
+
 	return true;
 }
 

diff --git a/src/socket/udpsocket.hxx b/src/socket/udpsocket.hxx
@@ -24,5 +24,9 @@ public:
 
 	virtual bool setNonBlocking(bool value) = 0;
 
+	virtual bool setSendBufferSize(int32_t size, int32_t& newSize) = 0;
+
+	virtual bool setRecvBufferSize(int32_t size, int32_t& newSize) = 0;
+
 	virtual bool isValid() = 0;
 };
diff --git a/src/socket/unix/udpsocketUnix.cxx b/src/socket/unix/udpsocketUnix.cxx
@@ -54,17 +54,35 @@ int32_t udpsocketUnix::recvFrom(void* buffer, size_t bufferSize, internetaddr* a
 	if (unixAddr == nullptr) [[unlikely]]
 		return -1;
 
-    socklen_t socklen = sizeof(sockaddr_in);
-    return ::recvfrom(m_socket, buffer, bufferSize, 0, (struct sockaddr*)&unixAddr->getAddr(), &socklen);
+	socklen_t socklen = sizeof(sockaddr_in);
+	return ::recvfrom(m_socket, buffer, bufferSize, 0, (struct sockaddr*)&unixAddr->getAddr(), &socklen);
 }
 
-bool udpsocketUnix::setNonBlocking(bool value)
+bool udpsocketUnix::setNonBlocking(bool bNonBlocking)
 {
 	int flags = fcntl(m_socket, F_GETFL, 0);
 	if (flags == -1) [[unlikely]]
 		return false;
+	flags = bNonBlocking ? flags | O_NONBLOCK : flags ^ (flags & O_NONBLOCK);
+	return fcntl(m_socket, F_SETFL, ) != -1;
+}
 
-	return fcntl(m_socket, F_SETFL, flags | O_NONBLOCK) != -1;
+bool udpsocketUnix::setSendBufferSize(int32_t size, int32_t& newSize)
+{
+	//int32_t sizeSize = sizeof(size);
+	//bool bOk = setsockopt(m_socket, SOL_SOCKET, SO_SNDBUF, (char*)&size, sizeof(int32_t)) == 0;
+	//getsockopt(m_socket, SOL_SOCKET, SO_SNDBUF, (char*)&newSize, &sizeSize);
+	//return bOk;
+	return false;
+}
+
+bool udpsocketUnix::setRecvBufferSize(int32_t size, int32_t& newSize)
+{
+	//int32_t sizeSize = sizeof(size);
+	//bool bOk = setsockopt(m_socket, SOL_SOCKET, SO_SNDBUF, (char*)&size, sizeof(int32_t)) == 0;
+	//getsockopt(m_socket, SOL_SOCKET, SO_RCVBUF, (char*)&newSize, &sizeSize);
+	//return bOk;
+	return false;
 }
 
 bool udpsocketUnix::isValid()

diff --git a/src/socket/unix/udpsocketUnix.hxx b/src/socket/unix/udpsocketUnix.hxx
@@ -9,13 +9,17 @@ public:
 
 	virtual ~udpsocketUnix();
 
-    virtual bool bind(int32_t port) override;
+	virtual bool bind(int32_t port) override;
 
 	virtual int32_t sendTo(void* buffer, size_t bufferSize, const internetaddr* addr) override;
 
 	virtual int32_t recvFrom(void* buffer, size_t bufferSize, internetaddr* addr) override;
 
-	virtual bool setNonBlocking(bool value) override;
+	virtual bool setNonBlocking(bool bNonBlocking) override;
+
+	virtual bool setSendBufferSize(int32_t size, int32_t& newSize) override;
+
+	virtual bool setRecvBufferSize(int32_t size, int32_t& newSize) override;
 
 	virtual bool isValid() override;