88 lines
1.9 KiB
C
88 lines
1.9 KiB
C
#include "net.h"
|
|
#include <unistd.h>
|
|
|
|
static volatile int socket_lock __attribute__((section(".data"))) = 0;
|
|
|
|
int32_t recvwait(int32_t sock, void *buffer, int32_t len) {
|
|
while (socket_lock) {
|
|
usleep(1000);
|
|
}
|
|
socket_lock = 1;
|
|
int32_t 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;
|
|
}
|
|
|
|
uint8_t recvbyte(int32_t sock) {
|
|
unsigned char buffer[1];
|
|
int32_t ret;
|
|
|
|
ret = recvwait(sock, buffer, 1);
|
|
if (ret < 0)
|
|
return ret;
|
|
return buffer[0];
|
|
}
|
|
|
|
uint32_t recvword(int32_t sock) {
|
|
uint32_t result;
|
|
int32_t ret;
|
|
|
|
ret = recvwait(sock, &result, 4);
|
|
if (ret < 0)
|
|
return ret;
|
|
return result;
|
|
}
|
|
|
|
int32_t checkbyte(int32_t sock) {
|
|
while (socket_lock) {
|
|
usleep(1000);
|
|
}
|
|
socket_lock = 1;
|
|
unsigned char buffer[1];
|
|
int32_t ret;
|
|
|
|
ret = recv(sock, buffer, 1, MSG_DONTWAIT);
|
|
socket_lock = 0;
|
|
if (ret < 0)
|
|
return ret;
|
|
if (ret == 0)
|
|
return -1;
|
|
return buffer[0];
|
|
}
|
|
|
|
int32_t sendwait(int32_t sock, const void *buffer, int32_t len) {
|
|
while (socket_lock) {
|
|
usleep(1000);
|
|
}
|
|
socket_lock = 1;
|
|
int32_t 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;
|
|
}
|
|
|
|
int32_t sendbyte(int32_t sock, unsigned char byte) {
|
|
unsigned char buffer[1];
|
|
buffer[0] = byte;
|
|
return sendwait(sock, buffer, 1);
|
|
}
|