RepoMirrors
/
mars
mirror of https://github.com/schoebel/mars
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

server: safeguard interim realloc and len changes

This commit is contained in:
Thomas Schoebel-Theuer 2022-05-26 13:18:29 +02:00
parent eefd950f26
commit 9a8ae5e78a
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
kernel/mars_server.c
View File

@ -245,6 +245,8 @@ int server_io(struct server_brick *brick, struct mars_socket *sock, struct mars_
mars_free_mref(mref);
goto done;
}
mref_a->brick = brick;
mref_a->data = mref->ref_data;
status = mars_recv_mref(sock, mref, cmd);
if (status < 0) {
@ -252,7 +254,6 @@ int server_io(struct server_brick *brick, struct mars_socket *sock, struct mars_
goto done;
}
mref_a->brick = brick;
mref_a->data = mref->ref_data;
mref_a->len = mref->ref_len;
SETUP_CALLBACK(mref, server_endio, mref_a);
@ -269,6 +270,8 @@ int server_io(struct server_brick *brick, struct mars_socket *sock, struct mars_
status = 0; // continue serving requests
goto done;
}
mref_a->data = mref->ref_data;
mref_a->len = mref->ref_len;
mref_a->do_put = true;
atomic_inc(&brick->in_flight);
GENERIC_INPUT_CALL_VOID(brick->inputs[0], mref_io, mref);