mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-02-01 12:03:41 +00:00
b107d2709f
redbean improvements:
- Explicitly disable corking
- Simulate Python regex API for Lua
- Send warmup requests in main process on startup
- Add Class-A granular IPv4 network classification
- Add /statusz page so you can monitor your redbean's health
- Fix regressions on OpenBSD/NetBSD caused by recent changes
- Plug Authorization header into Lua GetUser and GetPass APIs
- Recognize X-Forwarded-{For,Host} from local reverse proxies
- Add many additional functions to redbean Lua server page API
- Report resource usage of child processes on `/` listing page
- Introduce `-a` flag for logging child process resource usage
- Introduce `-t MILLIS` flag and `ProgramTimeout(ms)` init API
- Introduce `-H "Header: value"` flag and `ProgramHeader(k,v)` API
Cosmopolitan Libc improvements:
- Make strerror() simpler
- Make inet_pton() not depend on sscanf()
- Fix OpenExecutable() which broke .data section earlier
- Fix stdio in cases where it overflows kernel tty buffer
- Fix bugs in crash reporting w/o .com.dbg binary present
- Add polyfills for SO_LINGER, SO_RCVTIMEO, and SO_SNDTIMEO
- Polyfill TCP_CORK on BSD and XNU using TCP_NOPUSH magnums
New netcat clone in examples/nc.c:
While testing some of the failure conditions for redbean, I noticed that
BusyBox's `nc` command is pretty busted, if you use it as an interactive
tool, rather than having it be part of a pipeline. Unfortunately this'll
only work on UNIX since Windows doesn't let us poll on stdio and sockets
at the same time because I don't think they want tools like this running
on their platform. So if you want forbidden fruit, it's here so enjoy it
167 lines
8.3 KiB
C
167 lines
8.3 KiB
C
/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
|
||
│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8 :vi│
|
||
╞══════════════════════════════════════════════════════════════════════════════╡
|
||
│ Copyright 2021 Justine Alexandra Roberts Tunney │
|
||
│ │
|
||
│ Permission to use, copy, modify, and/or distribute this software for │
|
||
│ any purpose with or without fee is hereby granted, provided that the │
|
||
│ above copyright notice and this permission notice appear in all copies. │
|
||
│ │
|
||
│ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL │
|
||
│ WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED │
|
||
│ WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE │
|
||
│ AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL │
|
||
│ DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR │
|
||
│ PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER │
|
||
│ TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR │
|
||
│ PERFORMANCE OF THIS SOFTWARE. │
|
||
╚─────────────────────────────────────────────────────────────────────────────*/
|
||
#include "libc/mem/mem.h"
|
||
#include "libc/rand/rand.h"
|
||
#include "libc/runtime/gc.internal.h"
|
||
#include "libc/str/str.h"
|
||
#include "libc/testlib/ezbench.h"
|
||
#include "libc/testlib/hyperion.h"
|
||
#include "libc/testlib/testlib.h"
|
||
#include "net/http/escape.h"
|
||
|
||
size_t i, n, m;
|
||
char *p, *q, b[32];
|
||
|
||
TEST(EncodeBase64, test) {
|
||
EXPECT_STREQ("", gc(EncodeBase64("", 0, 0)));
|
||
EXPECT_STREQ("AA==", gc(EncodeBase64("\0", 1, 0)));
|
||
EXPECT_STREQ("AAA=", gc(EncodeBase64("\0\0", 2, 0)));
|
||
EXPECT_STREQ("AAAA", gc(EncodeBase64("\0\0\0", 3, 0)));
|
||
EXPECT_STREQ("/w==", gc(EncodeBase64("\377", 1, 0)));
|
||
EXPECT_STREQ("//8=", gc(EncodeBase64("\377\377", 2, 0)));
|
||
EXPECT_STREQ("////", gc(EncodeBase64("\377\377\377", 3, 0)));
|
||
EXPECT_STREQ("aGVsbG8=", gc(EncodeBase64("hello", 5, 0)));
|
||
EXPECT_STREQ(
|
||
"AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIjJCUmJygpKissLS4vMDEy"
|
||
"MzQ1Njc4OTo7PD0+P0BBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWltcXV5fYGFiY2Rl"
|
||
"ZmdoaWprbG1ub3BxcnN0dXZ3eHl6e3x9fn+AgYKDhIWGh4iJiouMjY6PkJGSk5SVlpeY"
|
||
"mZqbnJ2en6ChoqOkpaanqKmqq6ytrq+wsbKztLW2t7i5uru8vb6/wMHCw8TFxsfIycrL"
|
||
"zM3Oz9DR0tPU1dbX2Nna29zd3t/g4eLj5OXm5+jp6uvs7e7v8PHy8/T19vf4+fr7/P3+"
|
||
"/w==",
|
||
gc(EncodeBase64(
|
||
"\000\001\002\003\004\005\006\007\010\011\012\013\014\015\016\017\020"
|
||
"\021\022\023\024\025\026\027\030\031\032\033\034\035\036\037\040\041"
|
||
"\042\043\044\045\046\047\050\051\052\053\054\055\056\057\060\061\062"
|
||
"\063\064\065\066\067\070\071\072\073\074\075\076\077\100\101\102\103"
|
||
"\104\105\106\107\110\111\112\113\114\115\116\117\120\121\122\123\124"
|
||
"\125\126\127\130\131\132\133\134\135\136\137\140\141\142\143\144\145"
|
||
"\146\147\150\151\152\153\154\155\156\157\160\161\162\163\164\165\166"
|
||
"\167\170\171\172\173\174\175\176\177\200\201\202\203\204\205\206\207"
|
||
"\210\211\212\213\214\215\216\217\220\221\222\223\224\225\226\227\230"
|
||
"\231\232\233\234\235\236\237\240\241\242\243\244\245\246\247\250\251"
|
||
"\252\253\254\255\256\257\260\261\262\263\264\265\266\267\270\271\272"
|
||
"\273\274\275\276\277\300\301\302\303\304\305\306\307\310\311\312\313"
|
||
"\314\315\316\317\320\321\322\323\324\325\326\327\330\331\332\333\334"
|
||
"\335\336\337\340\341\342\343\344\345\346\347\350\351\352\353\354\355"
|
||
"\356\357\360\361\362\363\364\365\366\367\370\371\372\373\374\375\376"
|
||
"\377",
|
||
256, &n)));
|
||
EXPECT_EQ(344, n);
|
||
}
|
||
|
||
TEST(DecodeBase64, test) {
|
||
EXPECT_BINEQ(u" ", gc(DecodeBase64("", 0, 0)));
|
||
EXPECT_BINEQ(u" ", gc(DecodeBase64("AA==", 4, 0)));
|
||
EXPECT_BINEQ(u"♦ ", gc(DecodeBase64("BB==", 4, 0)));
|
||
EXPECT_BINEQ(u" ", gc(DecodeBase64("AAA=", 4, 0)));
|
||
EXPECT_BINEQ(u" ", gc(DecodeBase64("AAAA", 4, 0)));
|
||
EXPECT_BINEQ(u"λ ", gc(DecodeBase64("/w==", 4, 0)));
|
||
EXPECT_BINEQ(u"λλ ", gc(DecodeBase64("//8=", 4, 0)));
|
||
EXPECT_BINEQ(u"λλλ ", gc(DecodeBase64("////", 4, 0)));
|
||
EXPECT_BINEQ(u"hello ", gc(DecodeBase64("aGVsbG8=", 8, 0)));
|
||
EXPECT_BINEQ(u"hello ", gc(DecodeBase64("aGVsbG8=", -1, 0)));
|
||
EXPECT_EQ(
|
||
0,
|
||
memcmp(
|
||
"\000\001\002\003\004\005\006\007\010\011\012\013\014\015\016\017\020"
|
||
"\021\022\023\024\025\026\027\030\031\032\033\034\035\036\037\040\041"
|
||
"\042\043\044\045\046\047\050\051\052\053\054\055\056\057\060\061\062"
|
||
"\063\064\065\066\067\070\071\072\073\074\075\076\077\100\101\102\103"
|
||
"\104\105\106\107\110\111\112\113\114\115\116\117\120\121\122\123\124"
|
||
"\125\126\127\130\131\132\133\134\135\136\137\140\141\142\143\144\145"
|
||
"\146\147\150\151\152\153\154\155\156\157\160\161\162\163\164\165\166"
|
||
"\167\170\171\172\173\174\175\176\177\200\201\202\203\204\205\206\207"
|
||
"\210\211\212\213\214\215\216\217\220\221\222\223\224\225\226\227\230"
|
||
"\231\232\233\234\235\236\237\240\241\242\243\244\245\246\247\250\251"
|
||
"\252\253\254\255\256\257\260\261\262\263\264\265\266\267\270\271\272"
|
||
"\273\274\275\276\277\300\301\302\303\304\305\306\307\310\311\312\313"
|
||
"\314\315\316\317\320\321\322\323\324\325\326\327\330\331\332\333\334"
|
||
"\335\336\337\340\341\342\343\344\345\346\347\350\351\352\353\354\355"
|
||
"\356\357\360\361\362\363\364\365\366\367\370\371\372\373\374\375\376"
|
||
"\377",
|
||
gc(DecodeBase64(
|
||
"AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIjJCUm\r\n"
|
||
"JygpKissLS4vMDEy\r\n"
|
||
"MzQ1Njc4OTo7PD0+\r\n"
|
||
"P0BBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWltcXV5fYGFiY2Rl\r\n"
|
||
"ZmdoaWprbG1ub3BxcnN0dXZ3eHl6e3x9fn+\r\n"
|
||
"AgYKDhIWGh4iJiouMjY6PkJGSk5SVlpeY\r\n"
|
||
"mZqbnJ2en6ChoqOkpaanqKmqq6ytrq+wsbKztLW2t7i5uru8vb6/\r\n"
|
||
"wMHCw8TFxsfIycrL\r\n"
|
||
"zM3Oz9DR0tPU1dbX2Nna29zd3t/\r\n"
|
||
"g4eLj5OXm5+jp6uvs7e7v8PHy8/T19vf4+fr7/P3+\r\n"
|
||
"/w==\r\n",
|
||
-1, &n)),
|
||
256));
|
||
EXPECT_EQ(256, n);
|
||
}
|
||
|
||
TEST(DecodeBase64, testSeparators_skipsOverThemAtAnyState) {
|
||
EXPECT_BINEQ(u" ", gc(DecodeBase64(" ", 1, 0)));
|
||
EXPECT_BINEQ(u" ", gc(DecodeBase64(" A A = = ", 9, 0)));
|
||
EXPECT_BINEQ(u"♦ ", gc(DecodeBase64(" B B = = ", 4, 0)));
|
||
EXPECT_BINEQ(u"hello ", gc(DecodeBase64("a\nG\nV\ns\nb\nG\n8\n=\n", 16, 0)));
|
||
}
|
||
|
||
TEST(DecodeBase64, testInvalidSequences_skipsOverThem) {
|
||
EXPECT_BINEQ(u" ", gc(DecodeBase64("A===", 4, 0)));
|
||
EXPECT_BINEQ(u" ", gc(DecodeBase64("B===", 4, 0)));
|
||
EXPECT_BINEQ(u"♦ ", gc(DecodeBase64("B===BB==", 8, 0)));
|
||
EXPECT_BINEQ(u"♦ ", gc(DecodeBase64("====BB==", 8, 0)));
|
||
}
|
||
|
||
TEST(DecodeBase64, testOom_returnsNullAndSetsSizeToZero) {
|
||
n = 31337;
|
||
EXPECT_EQ(NULL, DecodeBase64("hello", 0x1000000000000, &n));
|
||
EXPECT_EQ(0, n);
|
||
}
|
||
|
||
TEST(EncodeBase64, testOom_returnsNullAndSetsSizeToZero) {
|
||
n = 31337;
|
||
EXPECT_EQ(NULL, EncodeBase64("hello", 0x1000000000000, &n));
|
||
EXPECT_EQ(0, n);
|
||
}
|
||
|
||
TEST(Base64, RoundTrip) {
|
||
for (i = 0; i < 1000; ++i) {
|
||
n = rand() % 32;
|
||
rngset(b, n, rand64, -1);
|
||
p = EncodeBase64(b, n, &m);
|
||
q = DecodeBase64(p, m, &m);
|
||
ASSERT_EQ(n, m);
|
||
ASSERT_EQ(0, memcmp(b, q, n));
|
||
free(p);
|
||
free(q);
|
||
}
|
||
}
|
||
|
||
TEST(Base64, Fuzz) {
|
||
for (i = 0; i < 1000; ++i) {
|
||
n = rand() % 32;
|
||
rngset(b, n, rand64, -1);
|
||
free(DecodeBase64(p, m, 0));
|
||
}
|
||
}
|
||
|
||
BENCH(EncodeBase64, bench) {
|
||
EZBENCH2("EncodeBase64", donothing,
|
||
free(EncodeBase64(kHyperion, kHyperionSize, 0)));
|
||
p = gc(EncodeBase64(kHyperion, kHyperionSize, &n));
|
||
EZBENCH2("DecodeBase64", donothing, free(DecodeBase64(p, n, 0)));
|
||
}
|