project-base/server-skynet-source-3rd-jemalloc
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add ASSURED_WRITE in mallctl

Browse Source
This commit is contained in:
Yinan Zhang 2020-02-04 16:36:02 -08:00
parent 2476889195
commit 7014f81e17
1 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/ctl.c
View File

@ -1496,6 +1496,14 @@ ctl_mtx_assert_held(tsdn_t *tsdn) {
} \ } \
} while (0) } while (0)
#define ASSURED_WRITE(v, t) do { \
if (newp == NULL || newlen != sizeof(t)) { \
ret = EINVAL; \
goto label_return; \
} \
(v) = *(t *)newp; \
} while (0)
#define MIB_UNSIGNED(v, i) do { \ #define MIB_UNSIGNED(v, i) do { \
if (mib[i] > UINT_MAX) { \ if (mib[i] > UINT_MAX) { \
ret = EFAULT; \ ret = EFAULT; \
@ -2048,12 +2056,7 @@ tcache_flush_ctl(tsd_t *tsd, const size_t *mib, size_t miblen,
unsigned tcache_ind; unsigned tcache_ind;
WRITEONLY(); WRITEONLY();
tcache_ind = UINT_MAX; ASSURED_WRITE(tcache_ind, unsigned);
WRITE(tcache_ind, unsigned);
if (tcache_ind == UINT_MAX) {
ret = EFAULT;
goto label_return;
}
tcaches_flush(tsd, tcache_ind); tcaches_flush(tsd, tcache_ind);
ret = 0; ret = 0;
@ -2068,12 +2071,7 @@ tcache_destroy_ctl(tsd_t *tsd, const size_t *mib, size_t miblen,
unsigned tcache_ind; unsigned tcache_ind;
WRITEONLY(); WRITEONLY();
tcache_ind = UINT_MAX; ASSURED_WRITE(tcache_ind, unsigned);
WRITE(tcache_ind, unsigned);
if (tcache_ind == UINT_MAX) {
ret = EFAULT;
goto label_return;
}
tcaches_destroy(tsd, tcache_ind); tcaches_destroy(tsd, tcache_ind);
ret = 0; ret = 0;