mirror of https://github.com/bjc/prosody
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
1.6 KiB
91 lines
1.6 KiB
/*
|
|
* This project is MIT licensed. Please see the
|
|
* COPYING file in the source package for more information.
|
|
*
|
|
* Copyright (C) 2016 Kim Alvefur
|
|
*/
|
|
|
|
#include <lua.h>
|
|
#include <lauxlib.h>
|
|
|
|
#if (LUA_VERSION_NUM == 501)
|
|
#define luaL_setfuncs(L, R, N) luaL_register(L, NULL, R)
|
|
#endif
|
|
|
|
/* TODO Deduplicate code somehow */
|
|
|
|
int strop_and(lua_State *L) {
|
|
luaL_Buffer buf;
|
|
size_t a, b, i;
|
|
const char *str_a = luaL_checklstring(L, 1, &a);
|
|
const char *str_b = luaL_checklstring(L, 2, &b);
|
|
|
|
luaL_buffinit(L, &buf);
|
|
|
|
if(a == 0 || b == 0) {
|
|
lua_settop(L, 1);
|
|
return 1;
|
|
}
|
|
|
|
for(i = 0; i < a; i++) {
|
|
luaL_addchar(&buf, str_a[i] & str_b[i % b]);
|
|
}
|
|
|
|
luaL_pushresult(&buf);
|
|
return 1;
|
|
}
|
|
|
|
int strop_or(lua_State *L) {
|
|
luaL_Buffer buf;
|
|
size_t a, b, i;
|
|
const char *str_a = luaL_checklstring(L, 1, &a);
|
|
const char *str_b = luaL_checklstring(L, 2, &b);
|
|
|
|
luaL_buffinit(L, &buf);
|
|
|
|
if(a == 0 || b == 0) {
|
|
lua_settop(L, 1);
|
|
return 1;
|
|
}
|
|
|
|
for(i = 0; i < a; i++) {
|
|
luaL_addchar(&buf, str_a[i] | str_b[i % b]);
|
|
}
|
|
|
|
luaL_pushresult(&buf);
|
|
return 1;
|
|
}
|
|
|
|
int strop_xor(lua_State *L) {
|
|
luaL_Buffer buf;
|
|
size_t a, b, i;
|
|
const char *str_a = luaL_checklstring(L, 1, &a);
|
|
const char *str_b = luaL_checklstring(L, 2, &b);
|
|
|
|
luaL_buffinit(L, &buf);
|
|
|
|
if(a == 0 || b == 0) {
|
|
lua_settop(L, 1);
|
|
return 1;
|
|
}
|
|
|
|
for(i = 0; i < a; i++) {
|
|
luaL_addchar(&buf, str_a[i] ^ str_b[i % b]);
|
|
}
|
|
|
|
luaL_pushresult(&buf);
|
|
return 1;
|
|
}
|
|
|
|
LUA_API int luaopen_util_strbitop(lua_State *L) {
|
|
luaL_Reg exports[] = {
|
|
{ "sand", strop_and },
|
|
{ "sor", strop_or },
|
|
{ "sxor", strop_xor },
|
|
{ NULL, NULL }
|
|
};
|
|
|
|
lua_newtable(L);
|
|
luaL_setfuncs(L, exports, 0);
|
|
return 1;
|
|
}
|
|
|