mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-04 21:57:45 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

util.pposix: Use Lua enum API for resource limit name argument

Browse source 
Because diffstat.
This commit is contained in:
Kim Alvefur 2023-06-14 13:39:39 +02:00
parent 23273ed228
commit b7b65435f7
1 changed files with 34 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

82
util-src/pposix.c
View file

@ -507,55 +507,43 @@ static int lc_mkdir(lua_State *L) {
* Example usage: * Example usage:
* pposix.setrlimit("NOFILE", 1000, 2000) * pposix.setrlimit("NOFILE", 1000, 2000)
*/ */
static int string2resource(const char *s) {
if(!strcmp(s, "CORE")) {
return RLIMIT_CORE;
}
if(!strcmp(s, "CPU")) {
return RLIMIT_CPU;
}
if(!strcmp(s, "DATA")) {
return RLIMIT_DATA;
}
if(!strcmp(s, "FSIZE")) {
return RLIMIT_FSIZE;
}
if(!strcmp(s, "NOFILE")) {
return RLIMIT_NOFILE;
}
if(!strcmp(s, "STACK")) {
return RLIMIT_STACK;
}
static const char *const resource_strings[] = {
/* Defined by POSIX */
"CORE",
"CPU",
"DATA",
"FSIZE",
"NOFILE",
"STACK",
#if !(defined(sun) || defined(__sun) || defined(__APPLE__)) #if !(defined(sun) || defined(__sun) || defined(__APPLE__))
"MEMLOCK",
if(!strcmp(s, "MEMLOCK")) { "NPROC",
return RLIMIT_MEMLOCK; "RSS",
}
if(!strcmp(s, "NPROC")) {
return RLIMIT_NPROC;
}
if(!strcmp(s, "RSS")) {
return RLIMIT_RSS;
}
#endif #endif
#ifdef RLIMIT_NICE #ifdef RLIMIT_NICE
"NICE",
if(!strcmp(s, "NICE")) {
return RLIMIT_NICE;
}
#endif #endif
return -1; NULL
} };
static int resource_constants[] = {
RLIMIT_CORE,
RLIMIT_CPU,
RLIMIT_DATA,
RLIMIT_FSIZE,
RLIMIT_NOFILE,
RLIMIT_STACK,
#if !(defined(sun) || defined(__sun) || defined(__APPLE__))
RLIMIT_MEMLOCK,
RLIMIT_NPROC,
RLIMIT_RSS,
#endif
#ifdef RLIMIT_NICE
RLIMIT_NICE,
#endif
-1,
};
static rlim_t arg_to_rlimit(lua_State *L, int idx, rlim_t current) { static rlim_t arg_to_rlimit(lua_State *L, int idx, rlim_t current) {
switch(lua_type(L, idx)) { switch(lua_type(L, idx)) {
@ -589,7 +577,7 @@ static int lc_setrlimit(lua_State *L) {
return 2; return 2;
} }
rid = string2resource(luaL_checkstring(L, 1)); rid = resource_constants[luaL_checkoption(L, 1,NULL, resource_strings)];
if(rid == -1) { if(rid == -1) {
lua_pushboolean(L, 0); lua_pushboolean(L, 0);
@ -619,7 +607,6 @@ static int lc_setrlimit(lua_State *L) {
static int lc_getrlimit(lua_State *L) { static int lc_getrlimit(lua_State *L) {
int arguments = lua_gettop(L); int arguments = lua_gettop(L);
const char *resource = NULL;
int rid = -1; int rid = -1;
struct rlimit lim; struct rlimit lim;
@ -629,8 +616,7 @@ static int lc_getrlimit(lua_State *L) {
return 2; return 2;
} }
resource = luaL_checkstring(L, 1); rid = resource_constants[luaL_checkoption(L, 1, NULL, resource_strings)];
rid = string2resource(resource);
if(rid != -1) { if(rid != -1) {
if(getrlimit(rid, &lim)) { if(getrlimit(rid, &lim)) {