libutils/source/utils/net.c

82 lines
1.8 KiB
C
Raw Normal View History

#include "net.h"
#include <dynamic_libs/os_functions.h>
#include <dynamic_libs/socket_functions.h>
static volatile int socket_lock __attribute__((section(".data"))) = 0;
s32 recvwait(s32 sock, void *buffer, s32 len) {
while(socket_lock) {
os_usleep(1000);
}
s32 ret;
while (len > 0) {
ret = recv(sock, buffer, len, 0);
if(ret < 0) {
socket_lock = 0;
return ret;
}
len -= ret;
buffer = (void *)(((char *) buffer) + ret);
}
socket_lock = 0;
return 0;
}
u8 recvbyte(s32 sock) {
unsigned char buffer[1];
s32 ret;
ret = recvwait(sock, buffer, 1);
if (ret < 0) return ret;
return buffer[0];
}
u32 recvword(s32 sock) {
u32 result;
s32 ret;
ret = recvwait(sock, &result, 4);
if (ret < 0) return ret;
return result;
}
s32 checkbyte(s32 sock) {
while(socket_lock) {
os_usleep(1000);
}
unsigned char buffer[1];
s32 ret;
ret = recv(sock, buffer, 1, MSG_DONTWAIT);
socket_lock = 0;
if (ret < 0) return ret;
if (ret == 0) return -1;
return buffer[0];
}
s32 sendwait(s32 sock, const void *buffer, s32 len) {
while(socket_lock) {
os_usleep(1000);
}
s32 ret;
while (len > 0) {
// For some reason the send blocks/crashes if the buffer is too big..
int cur_length = len <= 0x30 ? len : 0x30;
ret = send(sock, buffer, cur_length, 0);
if(ret < 0) {
socket_lock = 0;
return ret;
}
len -= ret;
buffer = (void *)(((char *) buffer) + ret);
}
socket_lock = 0;
return 0;
}
s32 sendbyte(s32 sock, unsigned char byte) {
unsigned char buffer[1];
buffer[0] = byte;
return sendwait(sock, buffer, 1);
}