magidoor/MD_Getc.c
Andrew Pamment 099abdcbf1 md_getstring
2019-12-01 08:08:59 +10:00

51 lines
1.1 KiB
C

#include <errno.h>
#include <stdio.h>
#include <sys/socket.h>
#include <unistd.h>
#include "MagiDoor.h"
char md_getc() {
char c;
ssize_t ret;
while (1) {
if (mdcontrol.socket == -1) {
ret = read(STDIN_FILENO, &c, 1);
} else {
ret = recv(mdcontrol.socket, &c, 1, 0);
}
if (ret == 0) {
md_exit(0);
}
if (ret == -1 && (errno == EAGAIN || errno == EWOULDBLOCK)) {
sleep(1);
continue;
}
return c;
}
}
int md_getstring(char *ptr, int maxlen) {
char c;
int len = 0;
while (len < maxlen) {
c = md_getc();
if (c == '\n' || c == '\0') {
continue;
}
if (c == '\r') {
return len;
} else if (c == '\b' || c == 127) {
if (len > 0) {
md_printf("\027[D \027[D");
len--;
ptr[len] = '\0';
}
continue;
}
ptr[len++] = c;
ptr[len] = '\0';
md_putchar(c);
}
return len;
}