Initial commit
This commit is contained in:
commit
165a19da21
|
|
@ -0,0 +1,17 @@
|
|||
SHELL = /bin/sh
|
||||
CC = clang
|
||||
CFLAGS = -ggdb -Wall -Wextra -std=c99
|
||||
LDFLAGS = `pkg-config --libs libssl`
|
||||
|
||||
.PHONY: all clean
|
||||
|
||||
targets = zyklonb
|
||||
|
||||
all: $(targets)
|
||||
|
||||
clean:
|
||||
rm -f $(targets)
|
||||
|
||||
zyklonb: src/zyklonb.c src/siphash.c
|
||||
$(CC) $^ -o $@ $(CFLAGS) $(LDFLAGS)
|
||||
|
||||
|
|
@ -0,0 +1,21 @@
|
|||
ZyklonB
|
||||
=======
|
||||
|
||||
`ZyklonB' is an experimental IRC bot, building upon the concept of my other
|
||||
VitaminA IRC bot. The main characteristic of these two bots is that they run
|
||||
plugins as coprocesses, which allows for enhanced reliability and programming
|
||||
language freedom.
|
||||
|
||||
While originally intended to be a simple C99 rewrite of the original bot, which
|
||||
was written in the GNU dialect of AWK, it fairly quickly became a playground
|
||||
where I added everything that seemed nice.
|
||||
|
||||
License
|
||||
-------
|
||||
`ZyklonB' is written by Přemysl Janouch <p.janouch@gmail.com>.
|
||||
|
||||
You may use the software under the terms of the ISC license, the text of which
|
||||
is included within the package, or, at your option, you may relicense the work
|
||||
under the MIT or the Modified BSD License, as listed at the following site:
|
||||
|
||||
http://www.gnu.org/licenses/license-list.html
|
||||
|
|
@ -0,0 +1,312 @@
|
|||
#!/usr/bin/awk -f
|
||||
#
|
||||
# ZyklonB eval plugin, LISP-like expression evaluator
|
||||
#
|
||||
# Copyright 2013, 2014 Přemysl Janouch. All rights reserved.
|
||||
# See the file LICENSE for licensing information.
|
||||
#
|
||||
|
||||
BEGIN \
|
||||
{
|
||||
RS = "\r"
|
||||
ORS = "\r\n"
|
||||
IGNORECASE = 1
|
||||
srand()
|
||||
|
||||
prefix = get_config("prefix")
|
||||
|
||||
print "ZYKLONB register"
|
||||
fflush("")
|
||||
|
||||
# All functions have to be in this particular array
|
||||
min_args["int"] = 1
|
||||
min_args["+"] = 1
|
||||
min_args["-"] = 1
|
||||
min_args["*"] = 1
|
||||
min_args["/"] = 1
|
||||
min_args["%"] = 1
|
||||
min_args["^"] = 1
|
||||
min_args["**"] = 1
|
||||
min_args["exp"] = 1
|
||||
min_args["sin"] = 1
|
||||
min_args["cos"] = 1
|
||||
min_args["atan2"] = 2
|
||||
min_args["log"] = 1
|
||||
min_args["rand"] = 0
|
||||
min_args["sqrt"] = 1
|
||||
|
||||
min_args["pi"] = 0
|
||||
min_args["e"] = 0
|
||||
|
||||
min_args["min"] = 1
|
||||
min_args["max"] = 1
|
||||
|
||||
# Whereas here their presence is only optional
|
||||
max_args["int"] = 1
|
||||
max_args["sin"] = 1
|
||||
max_args["cos"] = 1
|
||||
max_args["atan2"] = 2
|
||||
max_args["log"] = 1
|
||||
max_args["rand"] = 0
|
||||
max_args["sqrt"] = 1
|
||||
|
||||
max_args["pi"] = 0
|
||||
max_args["e"] = 0
|
||||
}
|
||||
|
||||
{
|
||||
parse($0)
|
||||
}
|
||||
|
||||
msg_command == "PRIVMSG" \
|
||||
{
|
||||
# Context = either channel or user nickname
|
||||
match(msg_prefix, /^[^!]+/)
|
||||
ctx = substr(msg_prefix, RSTART, RLENGTH)
|
||||
if (msg_param[0] ~ /^[#&!+]/)
|
||||
{
|
||||
ctxquote = ctx ": "
|
||||
ctx = msg_param[0]
|
||||
}
|
||||
else
|
||||
ctxquote = ""
|
||||
|
||||
|
||||
if (substr(msg_param[1], 1, length(prefix)) == prefix)
|
||||
{
|
||||
keyword = "eval"
|
||||
text = substr(msg_param[1], 1 + length(prefix))
|
||||
if (match(text, "^" keyword "([^A-Za-z0-9].*|$)"))
|
||||
process_request(substr(text, 1 + length(keyword)))
|
||||
}
|
||||
}
|
||||
|
||||
{
|
||||
fflush("")
|
||||
}
|
||||
|
||||
function pmrespond (text)
|
||||
{
|
||||
print "PRIVMSG " ctx " :" ctxquote text
|
||||
}
|
||||
|
||||
function process_request (input, res, x)
|
||||
{
|
||||
delete funs
|
||||
delete accumulator
|
||||
delete n_args
|
||||
|
||||
res = ""
|
||||
fun_top = 0
|
||||
funs[0] = ""
|
||||
accumulator[0] = 0
|
||||
n_args[0] = 0
|
||||
|
||||
if (match(input, "^[ \t]*"))
|
||||
input = substr(input, RLENGTH + 1)
|
||||
if (input == "")
|
||||
res = "expression missing"
|
||||
|
||||
while (res == "" && input != "") {
|
||||
if (match(input, "^-?[0-9]+\\.?[0-9]*")) {
|
||||
x = substr(input, RSTART, RLENGTH)
|
||||
input = substr(input, RLENGTH + 1)
|
||||
|
||||
match(input, "^ *")
|
||||
input = substr(input, RLENGTH + 1)
|
||||
|
||||
res = process_argument(x)
|
||||
} else if (match(input, "^[(]([^ ()]+)")) {
|
||||
x = substr(input, RSTART + 1, RLENGTH - 1)
|
||||
input = substr(input, RLENGTH + 1)
|
||||
|
||||
match(input, "^ *")
|
||||
input = substr(input, RLENGTH + 1)
|
||||
|
||||
if (!(x in min_args)) {
|
||||
res = "undefined function '" x "'"
|
||||
} else {
|
||||
fun_top++
|
||||
funs[fun_top] = x
|
||||
accumulator[fun_top] = 636363
|
||||
n_args[fun_top] = 0
|
||||
}
|
||||
} else if (match(input, "^[)] *")) {
|
||||
input = substr(input, RLENGTH + 1)
|
||||
res = process_end()
|
||||
} else
|
||||
res = "invalid input at '" substr(input, 1, 10) "...'"
|
||||
}
|
||||
|
||||
if (res == "") {
|
||||
if (fun_top != 0)
|
||||
res = "unclosed '" funs[fun_top] "'"
|
||||
else if (n_args[0] != 1)
|
||||
res = "internal error, expected one result" \
|
||||
", got " n_args[0] " instead"
|
||||
}
|
||||
|
||||
if (res == "")
|
||||
pmrespond(accumulator[0])
|
||||
else
|
||||
pmrespond(res)
|
||||
}
|
||||
|
||||
function process_argument (arg)
|
||||
{
|
||||
if (fun_top == 0) {
|
||||
if (n_args[0]++ != 0)
|
||||
return "too many results, I only expect one"
|
||||
|
||||
accumulator[0] = arg
|
||||
return ""
|
||||
}
|
||||
|
||||
fun = funs[fun_top]
|
||||
if (fun in max_args && max_args[fun] <= n_args[fun_top])
|
||||
return "too many operands for " fun
|
||||
|
||||
if (fun == "int") {
|
||||
accumulator[fun_top] = int(arg)
|
||||
} else if (fun == "+") {
|
||||
if (n_args[fun_top] == 0)
|
||||
accumulator[fun_top] = arg
|
||||
else
|
||||
accumulator[fun_top] += arg
|
||||
} else if (fun == "-") {
|
||||
if (n_args[fun_top] == 0)
|
||||
accumulator[fun_top] = arg
|
||||
else
|
||||
accumulator[fun_top] -= arg
|
||||
} else if (fun == "*") {
|
||||
if (n_args[fun_top] == 0)
|
||||
accumulator[fun_top] = arg
|
||||
else
|
||||
accumulator[fun_top] *= arg
|
||||
} else if (fun == "/") {
|
||||
if (n_args[fun_top] == 0)
|
||||
accumulator[fun_top] = arg
|
||||
else if (arg == 0)
|
||||
return "division by zero"
|
||||
else
|
||||
accumulator[fun_top] /= arg
|
||||
} else if (fun == "%") {
|
||||
if (n_args[fun_top] == 0)
|
||||
accumulator[fun_top] = arg
|
||||
else if (arg == 0)
|
||||
return "division by zero"
|
||||
else
|
||||
accumulator[fun_top] %= arg
|
||||
} else if (fun == "^" || fun == "**" || fun == "exp") {
|
||||
if (n_args[fun_top] == 0)
|
||||
accumulator[fun_top] = arg
|
||||
else
|
||||
accumulator[fun_top] ^= arg
|
||||
} else if (fun == "sin") {
|
||||
accumulator[fun_top] = sin(arg)
|
||||
} else if (fun == "cos") {
|
||||
accumulator[fun_top] = cos(arg)
|
||||
} else if (fun == "atan2") {
|
||||
if (n_args[fun_top] == 0)
|
||||
accumulator[fun_top] = arg
|
||||
else
|
||||
accumulator[fun_top] = atan2(accumulator[fun_top], arg)
|
||||
} else if (fun == "log") {
|
||||
accumulator[fun_top] = log(arg)
|
||||
} else if (fun == "rand") {
|
||||
# Just for completeness, execution never gets here
|
||||
} else if (fun == "sqrt") {
|
||||
accumulator[fun_top] = sqrt(arg)
|
||||
} else if (fun == "min") {
|
||||
if (n_args[fun_top] == 0)
|
||||
accumulator[fun_top] = arg
|
||||
else if (accumulator[fun_top] > arg)
|
||||
accumulator[fun_top] = arg
|
||||
} else if (fun == "max") {
|
||||
if (n_args[fun_top] == 0)
|
||||
accumulator[fun_top] = arg
|
||||
else if (accumulator[fun_top] < arg)
|
||||
accumulator[fun_top] = arg
|
||||
} else
|
||||
return "internal error, unhandled operands for " fun
|
||||
|
||||
n_args[fun_top]++
|
||||
return ""
|
||||
}
|
||||
|
||||
function process_end ()
|
||||
{
|
||||
if (fun_top <= 0)
|
||||
return "extraneous ')'"
|
||||
|
||||
fun = funs[fun_top]
|
||||
if (!(fun in min_args))
|
||||
return "internal error, unhandled ')' for '" fun "'"
|
||||
if (min_args[fun] > n_args[fun_top])
|
||||
return "not enough operands for '" fun "'"
|
||||
|
||||
# There's no 'init' function to do it in
|
||||
if (fun == "rand")
|
||||
accumulator[fun_top] = rand()
|
||||
else if (fun == "pi")
|
||||
accumulator[fun_top] = 3.141592653589793
|
||||
else if (fun == "e")
|
||||
accumulator[fun_top] = 2.718281828459045
|
||||
|
||||
return process_argument(accumulator[fun_top--])
|
||||
}
|
||||
|
||||
function get_config (key)
|
||||
{
|
||||
print "ZYKLONB get_config :" key
|
||||
fflush("")
|
||||
|
||||
getline
|
||||
parse($0)
|
||||
return msg_param[0]
|
||||
}
|
||||
|
||||
function parse (line, s, n, id, token)
|
||||
{
|
||||
s = 1
|
||||
id = 0
|
||||
|
||||
# NAWK only uses the first character of RS
|
||||
if (line ~ /^\n/)
|
||||
line = substr(line, 2)
|
||||
|
||||
msg_prefix = ""
|
||||
msg_command = ""
|
||||
delete msg_param
|
||||
|
||||
n = match(substr(line, s), / |$/)
|
||||
while (n)
|
||||
{
|
||||
token = substr(line, s, n - 1)
|
||||
if (token ~ /^:/)
|
||||
{
|
||||
if (s == 1)
|
||||
msg_prefix = substr(token, 2)
|
||||
else
|
||||
{
|
||||
msg_param[id] = substr(line, s + 1)
|
||||
break
|
||||
}
|
||||
}
|
||||
else if (!msg_command)
|
||||
msg_command = toupper(token)
|
||||
else
|
||||
msg_param[id++] = token
|
||||
|
||||
s = s + n
|
||||
n = index(substr(line, s), " ")
|
||||
|
||||
if (!n)
|
||||
{
|
||||
n = length(substr(line, s)) + 1
|
||||
if (n == 1)
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,86 @@
|
|||
// Code taken from https://github.com/floodyberry/siphash with some edits.
|
||||
//
|
||||
// To the extent possible under law, the author(s) have dedicated all copyright
|
||||
// and related and neighboring rights to this software to the public domain
|
||||
// worldwide. This software is distributed without any warranty.
|
||||
//
|
||||
// You should have received a copy of the CC0 Public Domain Dedication along
|
||||
// with this software. If not, see <http://creativecommons.org/publicdomain/zero/1.0/>.
|
||||
|
||||
#include "siphash.h"
|
||||
|
||||
static uint64_t inline
|
||||
u8to64_le (const unsigned char *p)
|
||||
{
|
||||
return
|
||||
(uint64_t) p[0] |
|
||||
(uint64_t) p[1] << 8 |
|
||||
(uint64_t) p[2] << 16 |
|
||||
(uint64_t) p[3] << 24 |
|
||||
(uint64_t) p[4] << 32 |
|
||||
(uint64_t) p[5] << 40 |
|
||||
(uint64_t) p[6] << 48 |
|
||||
(uint64_t) p[7] << 56 ;
|
||||
}
|
||||
|
||||
uint64_t
|
||||
siphash (const unsigned char key[16], const unsigned char *m, size_t len)
|
||||
{
|
||||
uint64_t v0, v1, v2, v3;
|
||||
uint64_t mi, k0, k1;
|
||||
uint64_t last7;
|
||||
size_t i, blocks;
|
||||
|
||||
k0 = u8to64_le (key + 0);
|
||||
k1 = u8to64_le (key + 8);
|
||||
v0 = k0 ^ 0x736f6d6570736575ull;
|
||||
v1 = k1 ^ 0x646f72616e646f6dull;
|
||||
v2 = k0 ^ 0x6c7967656e657261ull;
|
||||
v3 = k1 ^ 0x7465646279746573ull;
|
||||
|
||||
last7 = (uint64_t) (len & 0xff) << 56;
|
||||
|
||||
#define ROTL64(a,b) (((a)<<(b))|((a)>>(64-b)))
|
||||
|
||||
#define COMPRESS \
|
||||
v0 += v1; v2 += v3; \
|
||||
v1 = ROTL64 (v1,13); v3 = ROTL64 (v3,16); \
|
||||
v1 ^= v0; v3 ^= v2; \
|
||||
v0 = ROTL64 (v0,32); \
|
||||
v2 += v1; v0 += v3; \
|
||||
v1 = ROTL64 (v1,17); v3 = ROTL64 (v3,21); \
|
||||
v1 ^= v2; v3 ^= v0; \
|
||||
v2 = ROTL64(v2,32);
|
||||
|
||||
for (i = 0, blocks = (len & ~7); i < blocks; i += 8) {
|
||||
mi = u8to64_le (m + i);
|
||||
v3 ^= mi;
|
||||
COMPRESS
|
||||
COMPRESS
|
||||
v0 ^= mi;
|
||||
}
|
||||
|
||||
switch (len - blocks)
|
||||
{
|
||||
case 7: last7 |= (uint64_t) m[i + 6] << 48;
|
||||
case 6: last7 |= (uint64_t) m[i + 5] << 40;
|
||||
case 5: last7 |= (uint64_t) m[i + 4] << 32;
|
||||
case 4: last7 |= (uint64_t) m[i + 3] << 24;
|
||||
case 3: last7 |= (uint64_t) m[i + 2] << 16;
|
||||
case 2: last7 |= (uint64_t) m[i + 1] << 8;
|
||||
case 1: last7 |= (uint64_t) m[i + 0] ;
|
||||
default:;
|
||||
};
|
||||
v3 ^= last7;
|
||||
COMPRESS
|
||||
COMPRESS
|
||||
v0 ^= last7;
|
||||
v2 ^= 0xff;
|
||||
COMPRESS
|
||||
COMPRESS
|
||||
COMPRESS
|
||||
COMPRESS
|
||||
|
||||
return v0 ^ v1 ^ v2 ^ v3;
|
||||
}
|
||||
|
||||
|
|
@ -0,0 +1,10 @@
|
|||
#ifndef SIPHASH_H
|
||||
#define SIPHASH_H
|
||||
|
||||
#include <stdint.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
uint64_t siphash (const unsigned char key[16],
|
||||
const unsigned char *m, size_t len);
|
||||
|
||||
#endif // SIPHASH_H
|
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue