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

util.strbitop: Library for bitwise operations on strings

Browse source
This commit is contained in:
Kim Alvefur 2019-09-07 13:37:47 +02:00
parent d69cf3b007
commit fb1808b185
3 changed files with 93 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
util-src/GNUmakefile
View file

@ -7,7 +7,7 @@ INSTALL_DATA=install -m644
TARGET?=../util/
ALL=encodings.so hashes.so net.so pposix.so signal.so table.so \
ringbuffer.so time.so poll.so compat.so
ringbuffer.so time.so poll.so compat.so strbitop.so
ifdef RANDOM
ALL+=crand.so

2
util-src/makefile
View file

@ -6,7 +6,7 @@ INSTALL_DATA=install -m644
TARGET?=../util/
ALL=encodings.so hashes.so net.so pposix.so signal.so table.so \
ringbuffer.so time.so poll.so compat.so
ringbuffer.so time.so poll.so compat.so strbitop.so
.ifdef $(RANDOM)
ALL+=crand.so

91
util-src/strbitop.c Normal file
View file

@ -0,0 +1,91 @@
/*
* 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;
}