PeterCxy
/
irqbalance
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #198 from Irqbalance/unix_trunc 

Ensure that we get all of a unix message
This commit is contained in:
Neil Horman 2022-01-26 09:50:40 -05:00 committed by GitHub
parent b85854ac17 0f594a6f19
commit b3adb5fd14
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
ui/irqbalance-ui.c
View File

@ -21,6 +21,7 @@ GList *tree = NULL;
setup_t setup;
GMainLoop *main_loop;
int is_tree = 1;
static int default_bufsz = 8192;
struct msghdr * create_credentials_msg()
{
@ -103,7 +104,10 @@ char * get_data(char *string)
/* Send "setup" to get sleep interval, banned IRQs and banned CPUs,
* "stats" to get CPU tree statistics
*/
int socket_fd = init_connection();
int socket_fd;
try_again:
socket_fd = init_connection();
if(!socket_fd) {
return NULL;
}
@ -115,17 +119,18 @@ char * get_data(char *string)
msg->msg_iov = &iov;
sendmsg(socket_fd, msg, 0);
/*
* This is just...horrible. Mental note to replace this
* With a select, ioctl to determine size, and malloc based
* on that
*/
char *data = malloc(8192);
int len = recv(socket_fd, data, 8192, 0);
char *data = malloc(default_bufsz);
int len = recv(socket_fd, data, default_bufsz, MSG_TRUNC);
close(socket_fd);
data[len] = '\0';
free(msg->msg_control);
free(msg);
if (len >= default_bufsz) {
/* msg was truncated, increase bufsz and try again */
default_bufsz += 8192;
free(data);
goto try_again;
}
return data;
}