Compare commits
436 Commits
Author | SHA1 | Date | |
---|---|---|---|
af62a3065a
|
|||
c36ad8c113
|
|||
7f7a07c4d2
|
|||
4ff22bd0c6
|
|||
83ca2467de
|
|||
0a1fb977d6
|
|||
1405036cf2
|
|||
5f3cd03a6f
|
|||
819b71f285
|
|||
ceaa384fce
|
|||
99c4eb1c8a
|
|||
fb59b0d8c4
|
|||
946adb54d7
|
|||
b369a1116e
|
|||
950bf19331
|
|||
993cb65724
|
|||
c7be0adc66
|
|||
4782707049
|
|||
3d1451448d
|
|||
1619e01174
|
|||
6112e719e5
|
|||
6f2751f69b
|
|||
925ff2b9e4
|
|||
c67edd4195
|
|||
d8fd552b40
|
|||
ec6457465c
|
|||
490c8a2ae0
|
|||
f87fb74ced
|
|||
7b9844e267
|
|||
308534b883
|
|||
b93d97258c
|
|||
cd80a194dd
|
|||
ffc5c80b65
|
|||
896486b64a
|
|||
c1706edcea
|
|||
f8d3253a9f
|
|||
ec3183e99b
|
|||
27d514ae12
|
|||
e82869e993
|
|||
a54a6af497
|
|||
c13bea5ab4
|
|||
fd1630a554
|
|||
f7f245c931
|
|||
e72cb57b4a
|
|||
1cf9172194
|
|||
f416c2f05c
|
|||
300e3dc0aa
|
|||
26869c0df5
|
|||
c7488b0f59
|
|||
4e6930692e
|
|||
24cb993cb8
|
|||
fc9538fa6d
|
|||
9d8120966d
|
|||
3c5ecacecc
|
|||
606865e5dc
|
|||
b79c9c710b
|
|||
4b4e7bd257
|
|||
8d6b7105f3
|
|||
95e2ddb24c
|
|||
535e1add96
|
|||
51d85cc99f
|
|||
a4c0093fbe
|
|||
6ff6f0c73b
|
|||
555e8f62a1
|
|||
03c2c7b364
|
|||
abb40a659c
|
|||
6eb3b99bfb
|
|||
9cb278eb2a
|
|||
33feabdec0
|
|||
7995a9699e
|
|||
680344fbec
|
|||
e667a79ab2
|
|||
816c8c0bfc
|
|||
1547805e4c
|
|||
96f3225f51
|
|||
9ad1ecf1da
|
|||
28d7cf68df
|
|||
204317f46a
|
|||
a56002a409
|
|||
db966b3a66
|
|||
5d27cf0398
|
|||
7a2acb0e66
|
|||
cff5d558d6
|
|||
665e4ceabc
|
|||
f241913620
|
|||
2969e435d1
|
|||
c4289ac018
|
|||
672745f6df
|
|||
f92c26c350
|
|||
f9b3cc29ab
|
|||
5c72a0cb60
|
|||
cf3cff0746
|
|||
b90ed61e03
|
|||
dfc659dec0
|
|||
096aeae23b
|
|||
cd97eca7d3
|
|||
ce658ac965
|
|||
dd4f768cc4
|
|||
45514f90ca
|
|||
44913c1e0e
|
|||
b244f86c72
|
|||
a3c1ecc0bf
|
|||
3227e615fe
|
|||
3ce72975b8
|
|||
1f20c70772
|
|||
fabb55d94b
|
|||
6d473bfa49
|
|||
8e83d6aa5f
|
|||
b422b37806
|
|||
74c97a512f
|
|||
7653c3117e
|
|||
483b386100
|
|||
bfa9cf4fcd
|
|||
9ac67dbfd3
|
|||
2efe65fc74
|
|||
2937bdaded
|
|||
782c7440b0
|
|||
88346fe722
|
|||
0f75aeea7a
|
|||
933aac0f09
|
|||
7f1299feb4
|
|||
8435048150
|
|||
206ae3264d
|
|||
5e050512ad
|
|||
db053121f2
|
|||
89a9d4b9d6
|
|||
bc7c3591a2
|
|||
1859c432c8
|
|||
33ec943e8b
|
|||
b30f9fa56d
|
|||
90e324cf87
|
|||
e7e1e7b18f
|
|||
63b1ca5d6a | |||
54313551fc | |||
92779e5dba | |||
100eb1597d | |||
f4c3345445
|
|||
893316ebfa
|
|||
6a511732af
|
|||
bb895c5bca
|
|||
a2f4bf57e0
|
|||
635271ec50
|
|||
2b4569aabe
|
|||
f361fce561
|
|||
c92742275a
|
|||
041e4d43a7
|
|||
170337d4d5
|
|||
f0b27b3b37
|
|||
ee8aedce91
|
|||
0648c75baa | |||
b6ba58d406
|
|||
abe0e326cb
|
|||
4fe067ed7d
|
|||
41e12d6293
|
|||
a738f1abfe
|
|||
b6309eec39
|
|||
9923a76ba7
|
|||
ea4cdff233
|
|||
6526b5cbcb
|
|||
0119945e03
|
|||
3fe1fe023a
|
|||
bc1c6d3498
|
|||
50bb537074
|
|||
c060ee5bb6
|
|||
4062883cb3
|
|||
e0e44e9c26
|
|||
f0d8a3db4c
|
|||
557e176d3d
|
|||
cd3bc9aa90
|
|||
6ab65abec9
|
|||
7d6fa4682d
|
|||
6f4cbb6e24
|
|||
174865b71c
|
|||
4a002d6c31
|
|||
2a3f74e825
|
|||
9eaa644fa1
|
|||
ce91fecc80
|
|||
15c160a60a
|
|||
fd5e2302d4
|
|||
8c13eecc51
|
|||
ae751925fa
|
|||
a1fee3d2ec
|
|||
8721f07d00
|
|||
81ad3fea3c
|
|||
877ee12351
|
|||
dfbe3bbb95
|
|||
6c13922f2f
|
|||
171bca55fa
|
|||
fe0cdc9b1a
|
|||
930d4db2a2
|
|||
b37757fc73
|
|||
61c645eca8
|
|||
2fd71116ec
|
|||
495a3a6aaf
|
|||
69dfc562af
|
|||
a2383825ed
|
|||
55f2318b57
|
|||
eeb0ce7035
|
|||
b5749ae991
|
|||
9adb4791df
|
|||
eba916fad4
|
|||
ac2d17f23a
|
|||
433cc7861f
|
|||
52ea670d36
|
|||
b4b5d77985
|
|||
aebc731c4e
|
|||
4994b4375b
|
|||
43c512dc5a
|
|||
3f5eee236d
|
|||
0b157bcb74
|
|||
30b163c6fa
|
|||
cf2c0de697
|
|||
7aa47cac61
|
|||
456deeae20
|
|||
b08481818c
|
|||
49ad349775
|
|||
2b823cabd6
|
|||
ecd4f16afe
|
|||
45c5f20345
|
|||
c42f27e961
|
|||
de8ab406f6
|
|||
22d50ed4bd
|
|||
de44f4a3fe
|
|||
cc29250d76
|
|||
c2f8f4c962
|
|||
ff708230bd
|
|||
52ea541833
|
|||
f4bd426f3c
|
|||
ffbbcc6490
|
|||
80986325ce
|
|||
10d405e745
|
|||
28f163f97a
|
|||
a8914aa981
|
|||
2ada22481d
|
|||
17b25a3596
|
|||
d130474989
|
|||
8dea0cd3fc
|
|||
0406cad0d8
|
|||
21b7ab585a
|
|||
7587e15749
|
|||
1b44752f91
|
|||
5eeb9ef3c1
|
|||
dd15b9d906
|
|||
12922a0661
|
|||
c0799101b1
|
|||
c1d076db04
|
|||
10464f3f30
|
|||
63781472fa
|
|||
81931d287d
|
|||
531ddb4880
|
|||
e0d8ab31d5
|
|||
3a36d54e9d
|
|||
33d9aa3a5d
|
|||
f4d30206b0
|
|||
5b094ba98d
|
|||
f60cdc8228
|
|||
ab1c4d6fd4
|
|||
26d54e9968
|
|||
b6c7d8f58e
|
|||
53fcceeafb
|
|||
96567909db
|
|||
4b3c067a75
|
|||
77b80ca67b
|
|||
70e76d8783
|
|||
6b1bc54cf3
|
|||
e1edb48a3c
|
|||
dc5d1bebcc
|
|||
7b562c4b78
|
|||
413ab2aa5b
|
|||
68315b9765
|
|||
820a232a96
|
|||
b676388018
|
|||
debac11f90
|
|||
9297788cdf
|
|||
e61d16fb41
|
|||
ebf3258092
|
|||
59d0c485fd
|
|||
c90df2397f
|
|||
c4eb6709cf
|
|||
a73cbac7a1
|
|||
95946b1666
|
|||
785ab31890
|
|||
5481d314c9
|
|||
5986f39802
|
|||
a972340209
|
|||
e6dd8b84f9
|
|||
c3942b3382
|
|||
fde2b7aabb
|
|||
d0be587a36
|
|||
351568004f
|
|||
be757add02
|
|||
037c150868
|
|||
576d0a0dba
|
|||
ee6f9115a8
|
|||
6bbf79245b
|
|||
080d729f31
|
|||
2f6ba62715
|
|||
4d4d94fc81
|
|||
55028bd9cd
|
|||
8b0eb45854
|
|||
df49236130
|
|||
c7ca62a7e9
|
|||
5d01f7e219
|
|||
ceb0167742
|
|||
8fa9f2528f
|
|||
2e16fdee96
|
|||
f9b7e83ac8
|
|||
9b9ddbb913
|
|||
566ac0ca93
|
|||
bcdf36527f
|
|||
75ef4110c8
|
|||
f1064692a8
|
|||
f0ec64b629
|
|||
0db781e823
|
|||
798c41f1c8
|
|||
bc4a764bd5
|
|||
105e11d31d
|
|||
15530b642a
|
|||
e856f3f091
|
|||
b04c787df4
|
|||
ee7d1e086b
|
|||
cf8862100a
|
|||
4cc7b1c18a
|
|||
6fa62aeb4d
|
|||
3663a6064b
|
|||
ed233f8b5e
|
|||
6e8875d528
|
|||
640f3abc1f
|
|||
7d8f065ec9
|
|||
dd44ffee95
|
|||
c0962b90a6
|
|||
161952441d
|
|||
09b2118863
|
|||
b64828b01a
|
|||
8a640acd24
|
|||
609fd34ab7
|
|||
7cb4f40d22
|
|||
156c7d6621
|
|||
f39e1a6cb6
|
|||
c708633197
|
|||
b7f574e2ed
|
|||
be4475c336
|
|||
c6fbf25192
|
|||
b6ecb68b1a
|
|||
492e6a94cb
|
|||
1733ad3c1d
|
|||
885ec2226f
|
|||
1429961e44
|
|||
d88c70c978
|
|||
bfe08623c2
|
|||
2ed8451db1
|
|||
495381ab3f
|
|||
9d1594dc01
|
|||
463568182d
|
|||
fbd9a2667f
|
|||
d39f69d665
|
|||
c6f364d6e8
|
|||
8cccf48826
|
|||
7e2c9bea6d
|
|||
562cfb04e6
|
|||
14bcc47db2
|
|||
815fa32d14
|
|||
31fa9195a1
|
|||
caa09c3f05
|
|||
00abf345b9
|
|||
f0338209f5
|
|||
f6ac55adaf
|
|||
5e9c98d67f
|
|||
ba57509461
|
|||
4e079e094f
|
|||
a4901f4c23
|
|||
f5a42b8c28
|
|||
4f33e1d809
|
|||
c7902ea5b5
|
|||
50f5438654
|
|||
a5c1516468
|
|||
db224c7452
|
|||
812d5ab384
|
|||
1be4929c6c
|
|||
80ed6dea3e
|
|||
14b4f71af7
|
|||
a6789a7f3e
|
|||
7c2e3f2f9b
|
|||
93973c6e77
|
|||
97b04173ee
|
|||
8002ab3ed7
|
|||
35b924eea5
|
|||
3dbe356b87
|
|||
84c3e87a08
|
|||
51531ff78e
|
|||
e0ecc35d0c
|
|||
c2de0ce439
|
|||
2bcfa97e1b
|
|||
7c5a57b3d8
|
|||
99328996fa
|
|||
8249b7ed89
|
|||
db94388e69
|
|||
a5919807a3
|
|||
cc53b43c8a
|
|||
1ea0c95ebb
|
|||
9f3d8cc0c0
|
|||
8b9b32b9af
|
|||
81b5a569db
|
|||
dcdf91d5bf
|
|||
2d6277dc23
|
|||
8335a20e32
|
|||
8ac29b3815
|
|||
687d918677
|
|||
b4a4f12e9c
|
|||
38448401d0
|
|||
352742d26c
|
|||
e71786716e
|
|||
bce6f64c6e
|
|||
43bf946cce
|
|||
49f1a5f429
|
|||
3ba1848e40
|
|||
f7fc4fa801
|
|||
e116b940b2
|
|||
37f3e4eae1
|
|||
bbbde82a46
|
|||
77852d9626
|
|||
13fb362d12
|
|||
0bf8c0ccc3
|
|||
f98ffc7077
|
|||
e6760cf665
|
|||
ae430c340c
|
|||
42b8aee4bd
|
|||
24ce5d8cd6
|
|||
00382a71e2
|
|||
feaadf31bf
|
|||
a16c66c7e7
|
|||
7500476dc5
|
|||
a417efdece
|
|||
d104a43d1b
|
|||
e68e0239c3
|
|||
f148578154
|
10
.gitignore
vendored
10
.gitignore
vendored
@ -1,8 +1,12 @@
|
||||
*
|
||||
!src
|
||||
!src/**
|
||||
!run.sh
|
||||
!res
|
||||
!res/**
|
||||
!doc
|
||||
!doc/**
|
||||
!test
|
||||
!test/**
|
||||
!Makefile
|
||||
!.gitignore
|
||||
!CppNet
|
||||
!CppNet/**
|
||||
!*.md
|
||||
|
141
Makefile
141
Makefile
@ -1,22 +1,127 @@
|
||||
.DEFAULT_GOAL := install
|
||||
|
||||
packages:
|
||||
@echo "Installing packages..."
|
||||
sudo apt-get install g++ libmagic-dev libssl-dev php-cgi
|
||||
@echo "Finished downloading!"
|
||||
CC=gcc
|
||||
CFLAGS=-std=gnu11 -Wno-unused-but-set-variable -D_DEFAULT_SOURCE -D_BSD_SOURCE -D_SVID_SOURCE -D_POSIX_C_SOURCE=200809L
|
||||
LDFLAGS=-lssl -lcrypto -lmagic -lz -lmaxminddb -lbrotlienc
|
||||
|
||||
update:
|
||||
@echo "Updating imported git repos..."
|
||||
cd CppNet
|
||||
git pull
|
||||
cd ..
|
||||
@echo "Finished updating!"
|
||||
DEBIAN_OPTS=-D CACHE_MAGIC_FILE="\"/usr/share/file/magic.mgc\"" -D PHP_FPM_SOCKET="\"/var/run/php/php7.4-fpm.sock\""
|
||||
|
||||
compile:
|
||||
@echo "Compiling..."
|
||||
@mkdir -p bin
|
||||
g++ src/necronda-server.cpp -o bin/necronda-server -std=c++17 -fPIC -pthread -lz -lmagic -lssl -ldl -lcrypto
|
||||
@echo "Finished compiling!"
|
||||
.PHONY: all prod debug default debian permit clean test
|
||||
all: prod
|
||||
default: bin bin/lib bin/worker bin/res bin/sesimos
|
||||
|
||||
install: | packages update compile
|
||||
@echo "Finished!"
|
||||
prod: CFLAGS += -O3
|
||||
prod: default
|
||||
|
||||
debug: CFLAGS += -Wall -pedantic
|
||||
debug: default
|
||||
|
||||
debian: CFLAGS += $(DEBIAN_OPTS)
|
||||
debian: prod
|
||||
|
||||
test: CFLAGS += -include test/mock_*.h
|
||||
test: bin bin/test
|
||||
bin/test
|
||||
|
||||
|
||||
bin:
|
||||
mkdir -p bin
|
||||
|
||||
bin/lib:
|
||||
mkdir -p bin/lib
|
||||
|
||||
bin/worker:
|
||||
mkdir -p bin/worker
|
||||
|
||||
bin/res:
|
||||
mkdir -p bin/res
|
||||
|
||||
bin/test: test/mock_*.c test/test_*.c \
|
||||
src/lib/utils.c src/lib/sock.c src/lib/list.c src/lib/http.c src/lib/http_static.c src/logger.c
|
||||
$(CC) -o $@ $(CFLAGS) $^ -lcriterion
|
||||
|
||||
|
||||
bin/%.o: src/%.c
|
||||
$(CC) -c -o $@ $(CFLAGS) $<
|
||||
|
||||
bin/lib/%.o: src/lib/%.c
|
||||
$(CC) -c -o $@ $(CFLAGS) $<
|
||||
|
||||
bin/worker/%.o: src/worker/%.c
|
||||
$(CC) -c -o $@ $(CFLAGS) $<
|
||||
|
||||
bin/res/%.o: bin/res/%.txt
|
||||
objcopy -I binary --rename-section .data=.rodata -O elf64-x86-64 $^ $@
|
||||
|
||||
bin/res/%.txt: res/%.*
|
||||
cp $^ $@
|
||||
echo -ne "\x00" >> $@
|
||||
|
||||
bin/sesimos: bin/server.o bin/logger.o bin/cache_handler.o bin/async.o bin/workers.o \
|
||||
bin/worker/request_handler.o bin/worker/tcp_acceptor.o \
|
||||
bin/worker/fastcgi_handler.o bin/worker/local_handler.o bin/worker/proxy_handler.o \
|
||||
bin/worker/ws_frame_handler.o \
|
||||
bin/lib/http_static.o bin/res/default.o bin/res/proxy.o bin/res/style.o \
|
||||
bin/res/icon_error.o bin/res/icon_info.o bin/res/icon_success.o bin/res/icon_warning.o \
|
||||
bin/res/globe.o \
|
||||
bin/lib/compress.o bin/lib/config.o bin/lib/fastcgi.o bin/lib/geoip.o bin/lib/error.o \
|
||||
bin/lib/http.o bin/lib/proxy.o bin/lib/sock.o bin/lib/uri.o \
|
||||
bin/lib/utils.o bin/lib/websocket.o bin/lib/mpmc.o bin/lib/list.o
|
||||
$(CC) -o $@ $^ $(CFLAGS) $(LDFLAGS)
|
||||
|
||||
|
||||
bin/server.o: src/server.h src/defs.h src/cache_handler.h src/lib/config.h src/lib/sock.h \
|
||||
src/lib/proxy.h src/lib/geoip.h src/lib/utils.h src/logger.h
|
||||
|
||||
bin/logger.o: src/logger.h
|
||||
|
||||
bin/cache_handler.o: src/cache_handler.h src/lib/utils.h src/lib/uri.h src/lib/compress.h src/logger.h
|
||||
|
||||
bin/async.o: src/async.h src/logger.h
|
||||
|
||||
bin/workers.o: src/workers.h src/lib/mpmc.h src/worker/func.h
|
||||
|
||||
bin/worker/request_handler.o: src/worker/func.h
|
||||
|
||||
bin/worker/tcp_acceptor.o: src/worker/func.h
|
||||
|
||||
bin/worker/fastcgi_handler.o: src/worker/func.h
|
||||
|
||||
bin/worker/local_handler.o: src/worker/func.h
|
||||
|
||||
bin/worker/proxy_handler.o: src/worker/func.h
|
||||
|
||||
bin/worker/ws_frame_handler.o: src/worker/func.h
|
||||
|
||||
bin/lib/compress.o: src/lib/compress.h
|
||||
|
||||
bin/lib/config.o: src/lib/config.h src/lib/utils.h src/lib/uri.h src/logger.h
|
||||
|
||||
bin/lib/fastcgi.o: src/lib/fastcgi.h src/server.h src/lib/utils.h src/lib/compress.h src/lib/http.h \
|
||||
src/lib/uri.h src/lib/include/fastcgi.h src/logger.h
|
||||
|
||||
bin/lib/error.o: src/lib/error.h
|
||||
|
||||
bin/lib/geoip.o: src/lib/geoip.h
|
||||
|
||||
bin/lib/http.o: src/lib/http.h src/lib/utils.h src/lib/compress.h src/lib/sock.h src/logger.h
|
||||
|
||||
bin/lib/list.o: src/lib/list.h
|
||||
|
||||
bin/lib/mpmc.o: src/lib/mpmc.h src/logger.h
|
||||
|
||||
bin/lib/proxy.o: src/lib/proxy.h src/defs.h src/server.h src/lib/compress.h src/logger.h
|
||||
|
||||
bin/lib/sock.o: src/lib/sock.h
|
||||
|
||||
bin/lib/uri.o: src/lib/uri.h src/lib/utils.h
|
||||
|
||||
bin/lib/utils.o: src/lib/utils.h
|
||||
|
||||
bin/lib/websocket.o: src/lib/websocket.h src/defs.h src/lib/utils.h src/lib/sock.h src/logger.h
|
||||
|
||||
|
||||
permit:
|
||||
sudo setcap 'cap_net_bind_service=+ep' "$(shell pwd)/bin/sesimos"
|
||||
|
||||
clean:
|
||||
rm -rf bin/*
|
||||
|
50
README.md
Normal file
50
README.md
Normal file
@ -0,0 +1,50 @@
|
||||
|
||||
Sesimos – Secure, simple, modern web server
|
||||
===========================================
|
||||
|
||||
## Features
|
||||
|
||||
* Full IPv4 and IPv6 support
|
||||
* TLS Server Name Inspection (SNI)
|
||||
* Serving local files via HTTP and HTTPS
|
||||
* File compression ([gzip](https://www.gzip.org/), [Brotli](https://www.brotli.org/))
|
||||
* Disk cache for compressed files
|
||||
* Reverse proxy for other HTTP and HTTPS servers
|
||||
* Transparent WebSocket reverse proxy
|
||||
* FastCGI support (e.g. [PHP-FPM](https://php-fpm.org/))
|
||||
* Automatic path info detection (e.g. `/my/file/extra/path` -> script: `/my/file.php`, path info: `extra/path`)
|
||||
* Support for [MaxMind's GeoIP Database](https://www.maxmind.com/en/geoip2-services-and-databases)
|
||||
* Optional DNS reverse lookup for connecting hosts
|
||||
* Automatic URL rewrite (e.g. `/index.html` -> `/`, `/test.php` -> `/test`)
|
||||
* Modern looking and responsive error documents
|
||||
|
||||
|
||||
## Configuration
|
||||
|
||||
See [doc/example.conf](doc/example.conf) for more details.
|
||||
|
||||
|
||||
### Global directives
|
||||
|
||||
* `geoip_dir` (optional) - path to a directory containing GeoIP databases
|
||||
* `dns_server` (optional) - address of a DNS server
|
||||
|
||||
|
||||
### Configuration
|
||||
|
||||
* `[cert <cert-name>]` - begins section for a certificate
|
||||
* `certificate` - path to SSL certificate (or certificate chain)
|
||||
* `private_key` - path to SSL private key
|
||||
* `[host <host>]` - begins section for the virtual host `<host>`
|
||||
* `cert` - the name of the certificate to use
|
||||
* Local
|
||||
* `webroot` - path to the root directory
|
||||
* `dir_mode` - specify the behaviour for directories without an `index.html` or `index.php`
|
||||
* `forbidden` - the server will respond with `403 Forbidden`
|
||||
* `info` - try passing *path info* to an upper `.php` file.
|
||||
* `list` - list contents of directory (**not implemented yet**)
|
||||
* Reverse proxy
|
||||
* `hostname` - hostname of server to be reverse proxy of
|
||||
* `port` - port to be used
|
||||
* `http` - use HTTP to communicate with server
|
||||
* `https` - use HTTPS to communicate with server
|
22
architecture.md
Normal file
22
architecture.md
Normal file
@ -0,0 +1,22 @@
|
||||
|
||||
# Architecture
|
||||
|
||||
* logger (1)
|
||||
* listener (2) - 80, 443
|
||||
* cache_handler (1)
|
||||
* connection_initializer
|
||||
* request_handler
|
||||
* local_handler
|
||||
* proxy_handler
|
||||
* ws_handler
|
||||
* fastcgi_handler
|
||||
|
||||
|
||||
* -> logger
|
||||
* main -> listener
|
||||
* listener -> connection_handler
|
||||
* connection_initializer -> request_handler
|
||||
* request_handler -> local_handler -> request_handler
|
||||
* local_handler -> fastcgi_handler -> request_handler
|
||||
* request_handler -> rp_handler -> request_handler
|
||||
* proxy_handler -> ws_handler -> request_handler
|
24
doc/example.conf
Normal file
24
doc/example.conf
Normal file
@ -0,0 +1,24 @@
|
||||
|
||||
#geoip_dir /var/dir
|
||||
#dns_server 192.168.0.1
|
||||
|
||||
[cert cert1]
|
||||
certificate /var/cert/cert.pem
|
||||
private_key /var/cert/cert.key
|
||||
|
||||
[host localhost]
|
||||
webroot /var/www/localhost
|
||||
dir_mode forbidden
|
||||
cert cert1
|
||||
|
||||
[host me.local]
|
||||
hostname www.example.com
|
||||
port 80
|
||||
cert cert1
|
||||
http
|
||||
|
||||
[host secure.local]
|
||||
hostname www.example.com
|
||||
port 443
|
||||
cert cert1
|
||||
https
|
22
res/default.html
Normal file
22
res/default.html
Normal file
@ -0,0 +1,22 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<title>%1$i %2$s - %7$s</title>
|
||||
<meta charset="UTF-8"/>
|
||||
<meta name="theme-color" content="%6$s"/>
|
||||
<meta name="color-scheme" content="light dark"/>
|
||||
<meta name="apple-mobile-web-app-status-bar-style" content="black-translucent"/>
|
||||
<meta name="viewport" content="width=device-width,initial-scale=1.0"/>
|
||||
<link rel="shortcut icon" type="image/x-icon" href="/favicon.ico"/>
|
||||
<link rel="stylesheet" type="text/css" href="/.sesimos/res/style.css"/>
|
||||
<link rel="alternate icon" type="image/svg+xml" sizes="any" href="%5$s"/>
|
||||
<style>html{--color:var(--%4$s);}</style>
|
||||
</head>
|
||||
<body>
|
||||
<main>
|
||||
<section>
|
||||
%3$s%9$s <div class="footer"><a href="https://%7$s/">%7$s</a> - %10$s</div>
|
||||
</section>
|
||||
%8$s </main>
|
||||
</body>
|
||||
</html>
|
3
res/globe.svg
Normal file
3
res/globe.svg
Normal file
@ -0,0 +1,3 @@
|
||||
<svg width="64" height="64" xmlns="http://www.w3.org/2000/svg">
|
||||
<path d="M2,32 a30,30,0,1,0,60,0 a30,30,0,1,0,-60,0 L62,32 M6,16 L58,16 M6,48 L58,48 M32,2 L32,62 a15,30,0,1,0,0,-60 a15,30,0,1,0,0,60 Z" stroke="#008000" stroke-width="2" fill="#00000000"/>
|
||||
</svg>
|
After Width: | Height: | Size: 266 B |
3
res/icon_error.svg
Normal file
3
res/icon_error.svg
Normal file
@ -0,0 +1,3 @@
|
||||
<svg width="16" height="16" xmlns="http://www.w3.org/2000/svg">
|
||||
<text x="4" y="12" fill="#C00000" style="font-family:'Arial',sans-serif">:(</text>
|
||||
</svg>
|
After Width: | Height: | Size: 158 B |
3
res/icon_info.svg
Normal file
3
res/icon_info.svg
Normal file
@ -0,0 +1,3 @@
|
||||
<svg width="16" height="16" xmlns="http://www.w3.org/2000/svg">
|
||||
<text x="4" y="12" fill="#606060" style="font-family:'Arial',sans-serif">:)</text>
|
||||
</svg>
|
After Width: | Height: | Size: 158 B |
3
res/icon_success.svg
Normal file
3
res/icon_success.svg
Normal file
@ -0,0 +1,3 @@
|
||||
<svg width="16" height="16" xmlns="http://www.w3.org/2000/svg">
|
||||
<text x="4" y="12" fill="#008000" style="font-family:'Arial',sans-serif">:)</text>
|
||||
</svg>
|
After Width: | Height: | Size: 158 B |
3
res/icon_warning.svg
Normal file
3
res/icon_warning.svg
Normal file
@ -0,0 +1,3 @@
|
||||
<svg width="16" height="16" xmlns="http://www.w3.org/2000/svg">
|
||||
<text x="4" y="12" fill="#E0C000" style="font-family:'Arial',sans-serif">:)</text>
|
||||
</svg>
|
After Width: | Height: | Size: 158 B |
31
res/proxy.html
Normal file
31
res/proxy.html
Normal file
@ -0,0 +1,31 @@
|
||||
<section class="error-ctx">
|
||||
<div class="box%1$s">
|
||||
<div class="content">
|
||||
<span>Client</span>
|
||||
<img src="/.sesimos/res/globe.svg"/>
|
||||
<span>Your Browser</span>
|
||||
</div>
|
||||
<div class="arrow request%2$s"></div>
|
||||
</div>
|
||||
<div class="border%8$s"></div>
|
||||
<div class="box%3$s">
|
||||
<div class="content">
|
||||
<span>Reverse Proxy</span>
|
||||
<h3>%10$03i</h3>
|
||||
<h4>%11$s</h4>
|
||||
<span>%15$s</span>
|
||||
</div>
|
||||
<div class="arrow request%4$s"></div>
|
||||
<div class="arrow response%5$s"></div>
|
||||
</div>
|
||||
<div class="border%9$s"></div>
|
||||
<div class="box%6$s">
|
||||
<div class="content">
|
||||
<span>Server</span>
|
||||
<h3>%12$s</h3>
|
||||
<h4>%13$s</h4>
|
||||
<span>%14$s</span>
|
||||
</div>
|
||||
<div class="arrow response%7$s"></div>
|
||||
</div>
|
||||
</section>
|
66
res/style.css
Normal file
66
res/style.css
Normal file
@ -0,0 +1,66 @@
|
||||
html{
|
||||
font-family:"Arial",sans-serif;
|
||||
--error:#C00000;
|
||||
--warning:#E0C000;
|
||||
--success:#008000;
|
||||
--info:#606060;
|
||||
--soft:#808080;
|
||||
}
|
||||
|
||||
body{background-color:#F0F0F0;margin:0;}
|
||||
main{max-width:650px;margin:2em auto;}
|
||||
section{margin:2em 1em;background-color:#FFFFFF;border: 1px solid var(--color);border-radius:4px;padding:1em;}
|
||||
h1,h2,h3,h4,h5,h6{text-align:center;color:var(--color);font-weight:normal;}
|
||||
h1{font-size:3em;margin:0.125em 0;}
|
||||
h2{font-size:1.5em;margin:0.25em 0 1em 0;}
|
||||
p{text-align:center;font-size:0.875em;}
|
||||
div.footer{color:var(--soft);font-size:0.75em;text-align:center;margin:2em 0 0.5em 0;}
|
||||
div.footer a{color:var(--soft);}
|
||||
ul,ol{width:fit-content;margin:auto;}
|
||||
pre{width:fit-content;margin:2em auto 0 auto;}
|
||||
|
||||
section.error-ctx{display:flex;padding:0;border:none;}
|
||||
div.box{flex:100% 1 1;border:1px solid var(--info);color:var(--info);position:relative;padding:1em;box-sizing:border-box;text-align:center;}
|
||||
div.box.error{border-color:var(--error);color:var(--error);}
|
||||
div.box.success{border-color:var(--success);color:var(--success);}
|
||||
div.arrow{position:absolute;height:20px;width:30px;z-index:10;background-repeat:no-repeat;background-size:contain;}
|
||||
div.arrow.response{left:-17.5px;bottom:calc(33.3333% - 10px);}
|
||||
div.arrow.request{right:-17.5px;top:calc(33.3333% - 10px);}
|
||||
div.border{flex:1px 0 0;background-color:var(--info);}
|
||||
div.border.error{background-color:var(--error);}
|
||||
div.border.success{background-color:var(--success);}
|
||||
div.content>span{display:block;color:var(--soft);font-size:0.75em;}
|
||||
div.content>img{height:3.75rem;margin:0.75rem auto;display:block;}
|
||||
h3{font-size:2.25em;margin:0.75rem 0 0 0;color:unset;height:2.5rem;}
|
||||
h4{font-size:1em;margin:0 0 0.75rem 0;color:unset;height:1.25rem;}
|
||||
|
||||
div.arrow.request.success{background-image:url('data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMzAiIGhlaWdodD0iMjAiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+PHBhdGggZD0iTTEsMSBMMjUsMSBMMjksMTAgTDI1LDE5IEwxLDE5IiBmaWxsPSIjRkZGRkZGIiBzdHJva2U9IiMwMDgwMDAiIHN0cm9rZS13aWR0aD0iMiIvPjwvc3ZnPgo=');}
|
||||
div.arrow.request.error{background-image:url('data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMzAiIGhlaWdodD0iMjAiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+PHBhdGggZD0iTTEsMSBMMjUsMSBMMjksMTAgTDI1LDE5IEwxLDE5IiBmaWxsPSIjRkZGRkZGIiBzdHJva2U9IiNDMDAwMDAiIHN0cm9rZS13aWR0aD0iMiIvPjwvc3ZnPgo=');}
|
||||
div.arrow.response.success{background-image:url('data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMzAiIGhlaWdodD0iMjAiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+PHBhdGggZD0iTTI5LDE5IEw1LDE5IEwxLDEwIEw1LDEgTDI5LDEiIGZpbGw9IiNGRkZGRkYiIHN0cm9rZT0iIzAwODAwMCIgc3Ryb2tlLXdpZHRoPSIyIi8+PC9zdmc+Cg==');}
|
||||
div.arrow.response.error{background-image:url('data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMzAiIGhlaWdodD0iMjAiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+PHBhdGggZD0iTTI5LDE5IEw1LDE5IEwxLDEwIEw1LDEgTDI5LDEiIGZpbGw9IiNGRkZGRkYiIHN0cm9rZT0iI0MwMDAwMCIgc3Ryb2tlLXdpZHRoPSIyIi8+PC9zdmc+Cg==');}
|
||||
|
||||
@media(prefers-color-scheme:dark){
|
||||
html{color:#FFFFFF;--soft:#404040;}
|
||||
body{background-color:#101010;}
|
||||
section{background-color:#181818;}
|
||||
|
||||
div.arrow.request.success{background-image:url('data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMzAiIGhlaWdodD0iMjAiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+PHBhdGggZD0iTTEsMSBMMjUsMSBMMjksMTAgTDI1LDE5IEwxLDE5IiBmaWxsPSIjMTgxODE4IiBzdHJva2U9IiMwMDgwMDAiIHN0cm9rZS13aWR0aD0iMiIvPjwvc3ZnPgo=');}
|
||||
div.arrow.request.error{background-image:url('data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMzAiIGhlaWdodD0iMjAiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+PHBhdGggZD0iTTEsMSBMMjUsMSBMMjksMTAgTDI1LDE5IEwxLDE5IiBmaWxsPSIjMTgxODE4IiBzdHJva2U9IiNDMDAwMDAiIHN0cm9rZS13aWR0aD0iMiIvPjwvc3ZnPgo=');}
|
||||
div.arrow.response.success{background-image:url('data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMzAiIGhlaWdodD0iMjAiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+PHBhdGggZD0iTTI5LDE5IEw1LDE5IEwxLDEwIEw1LDEgTDI5LDEiIGZpbGw9IiMxODE4MTgiIHN0cm9rZT0iIzAwODAwMCIgc3Ryb2tlLXdpZHRoPSIyIi8+PC9zdmc+Cg==');}
|
||||
div.arrow.response.error{background-image:url('data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMzAiIGhlaWdodD0iMjAiIHhtbG5zPSJodHRwOi8vd3d3LnczLm9yZy8yMDAwL3N2ZyI+PHBhdGggZD0iTTI5LDE5IEw1LDE5IEwxLDEwIEw1LDEgTDI5LDEiIGZpbGw9IiMxODE4MTgiIHN0cm9rZT0iI0MwMDAwMCIgc3Ryb2tlLXdpZHRoPSIyIi8+PC9zdmc+Cg==');}
|
||||
}
|
||||
|
||||
@media(min-width:650px){
|
||||
div.box:first-child{border-top-left-radius:4px;border-bottom-left-radius:4px;border-right:none;}
|
||||
div.box:last-child{border-top-right-radius:4px;border-bottom-right-radius:4px;border-left:none;}
|
||||
div.box:not(:last-child):not(:first-child){border-left:none;border-right:none;}
|
||||
}
|
||||
|
||||
@media(max-width:650px){
|
||||
section.error-ctx{flex-direction:column;height:unset;}
|
||||
div.box:first-child{border-top-left-radius:4px;border-top-right-radius:4px;border-bottom:none;padding-top:1em;}
|
||||
div.box:last-child{border-bottom-right-radius:4px;border-bottom-left-radius:4px;border-top:none;padding-bottom:1em;}
|
||||
div.box:not(:last-child):not(:first-child){border-top:none;border-bottom:none;}
|
||||
div.arrow.response{transform:rotate(90deg);top:-10px;left:calc(33.3333% - 22.5px);right:unset;}
|
||||
div.arrow.request{transform:rotate(90deg);bottom:-10px;right:calc(33.3333% - 22.5px);top:unset;}
|
||||
}
|
7
run.sh
7
run.sh
@ -1,7 +0,0 @@
|
||||
#!/bin/bash
|
||||
echo "-- Building and starting Necronda Server..."
|
||||
make update && make compile && \
|
||||
echo -e "-- Successfully finished compiling!\n" && \
|
||||
sleep 0.0625 && \
|
||||
echo -e "-- Starting Server...\n" && \
|
||||
authbind ./bin/necronda-server
|
196
src/URI.cpp
196
src/URI.cpp
@ -1,196 +0,0 @@
|
||||
|
||||
#include "URI.h"
|
||||
#include "necronda-server.h"
|
||||
#include <utility>
|
||||
#include <sys/stat.h>
|
||||
|
||||
using namespace std;
|
||||
|
||||
string getExtension(string path) {
|
||||
long pos = path.find_last_of('.');
|
||||
if (pos == string::npos) {
|
||||
return "";
|
||||
}
|
||||
return path.substr(pos + 1, path.length() - pos);
|
||||
}
|
||||
|
||||
string getFilename(string path) {
|
||||
long pos = path.find_last_of('/');
|
||||
if (pos == string::npos) {
|
||||
return "";
|
||||
}
|
||||
return path.substr(pos + 1, path.length() - pos);
|
||||
}
|
||||
|
||||
bool isDirectory(string path) {
|
||||
struct stat statbuf;
|
||||
return stat(path.c_str(), &statbuf) == 0 && S_ISDIR(statbuf.st_mode) != 0;
|
||||
}
|
||||
|
||||
bool isFile(string path) {
|
||||
struct stat statbuf;
|
||||
return stat(path.c_str(), &statbuf) == 0 && S_ISDIR(statbuf.st_mode) == 0;
|
||||
}
|
||||
|
||||
bool fileExists(string path) {
|
||||
struct stat statbuf;
|
||||
return stat(path.c_str(), &statbuf) == 0;
|
||||
}
|
||||
|
||||
URI::URI() = default;
|
||||
|
||||
URI::URI(string webroot, string reqpath) {
|
||||
unsigned long pos = reqpath.find('?');
|
||||
if (pos != string::npos) {
|
||||
queryinit = true;
|
||||
query = reqpath.substr(pos + 1, reqpath.length() - pos);
|
||||
reqpath.erase(pos, reqpath.length() - pos);
|
||||
} else {
|
||||
query = "";
|
||||
queryinit = false;
|
||||
}
|
||||
if (webroot.length() >= 1 && webroot[webroot.length() - 1] == '/') {
|
||||
webroot.erase(webroot.length() - 1);
|
||||
}
|
||||
reqpath = url_decode(reqpath);
|
||||
if (reqpath.find("/../") != string::npos) {
|
||||
throw (char *) "Invalid path";
|
||||
}
|
||||
if (reqpath[0] != '/') {
|
||||
reqpath = '/' + reqpath;
|
||||
}
|
||||
this->webroot = webroot;
|
||||
this->reqpath = reqpath;
|
||||
|
||||
info = "";
|
||||
relpath = reqpath;
|
||||
|
||||
while ((!fileExists(webroot + relpath) || (isDirectory(webroot + relpath) && !fileExists(webroot + relpath + "/index.php")))
|
||||
&& (!fileExists(webroot + relpath + ".php") || (isDirectory(webroot + relpath + ".php") && !fileExists(webroot + relpath + ".php/index.php")))
|
||||
&& (!fileExists(webroot + relpath + ".html") || (isDirectory(webroot + relpath + ".html") && !fileExists(webroot + relpath + ".html/index.php")))) {
|
||||
long slash = relpath.find_last_of('/');
|
||||
if (slash == string::npos || relpath == "/") {
|
||||
break;
|
||||
}
|
||||
info = relpath.substr(slash) + info;
|
||||
relpath.erase(slash);
|
||||
}
|
||||
|
||||
if (!info.empty() && isDirectory(webroot + relpath)) {
|
||||
relpath.append("/");
|
||||
}
|
||||
|
||||
string abs = relpath;
|
||||
if (fileExists(webroot + abs)) {
|
||||
string ext = getExtension(abs);
|
||||
if (ext == "php" || ext == "html") {
|
||||
abs.erase(abs.length() - ext.length() - 1, abs.length());
|
||||
}
|
||||
}
|
||||
|
||||
string fname = getFilename(abs);
|
||||
if (fname == "index") {
|
||||
abs.erase(abs.length() - fname.length() - 1, abs.length());
|
||||
}
|
||||
|
||||
this->filepath = webroot + relpath;
|
||||
|
||||
if (isDirectory(webroot + abs)) {
|
||||
if (abs[abs.length() - 1] != '/') {
|
||||
abs += "/";
|
||||
}
|
||||
this->relpath = abs;
|
||||
abs += "index";
|
||||
if (fileExists(webroot + abs + ".php")) {
|
||||
this->filepath = webroot + abs + ".php";
|
||||
} else if (fileExists(webroot + abs + ".html")) {
|
||||
this->filepath = webroot + abs + ".html";
|
||||
}
|
||||
} else {
|
||||
if (abs[abs.length() - 1] == '/') {
|
||||
abs.erase(abs.length() - 1, abs.length() - 1);
|
||||
}
|
||||
this->relpath = abs;
|
||||
if (fileExists(webroot + abs + ".php")) {
|
||||
this->filepath = webroot + abs + ".php";
|
||||
} else if (fileExists(webroot + abs + ".html")) {
|
||||
this->filepath = webroot + abs + ".html";
|
||||
}
|
||||
}
|
||||
|
||||
if (isStatic() && !info.empty()) {
|
||||
if (relpath[relpath.length() - 1] == '/') {
|
||||
relpath.erase(relpath.length() - 1);
|
||||
}
|
||||
newpath = relpath + info;
|
||||
filepath = "";
|
||||
} else if (relpath != reqpath) {
|
||||
if (!info.empty() && relpath[relpath.length() - 1] == '/') {
|
||||
info.erase(0,1);
|
||||
}
|
||||
newpath = relpath + info;
|
||||
} else {
|
||||
newpath = "";
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
string URI::getWebRoot() {
|
||||
return webroot;
|
||||
}
|
||||
|
||||
string URI::getRelativePath() {
|
||||
return relpath;
|
||||
}
|
||||
|
||||
string URI::getAbsolutePath() {
|
||||
return webroot + relpath;
|
||||
}
|
||||
|
||||
string URI::getFilePath() {
|
||||
return filepath;
|
||||
}
|
||||
|
||||
string URI::getRelativeFilePath() {
|
||||
string str = getFilePath();
|
||||
long len = getWebRoot().length();
|
||||
return str.substr(len, str.length() - len);
|
||||
}
|
||||
|
||||
string URI::getNewPath() {
|
||||
if (isStatic()) {
|
||||
if (hasQuery()) {
|
||||
return getRelativePath();
|
||||
}
|
||||
}
|
||||
if (!newpath.empty() && newpath != reqpath) {
|
||||
return url_encode(newpath) + (queryinit? "?" + query : "");
|
||||
} else {
|
||||
return "";
|
||||
}
|
||||
}
|
||||
|
||||
FILE *URI::openFile() {
|
||||
return fopen64(getFilePath().c_str(), "rb");
|
||||
}
|
||||
|
||||
string URI::getFilePathInfo() {
|
||||
return info; //getAbsolutePath().erase(getFilePath().length(), getAbsolutePath().length());
|
||||
}
|
||||
|
||||
string URI::getFileType() {
|
||||
return getMimeType(getFilePath());
|
||||
}
|
||||
|
||||
bool URI::isStatic() {
|
||||
return getExtension(filepath) != "php";
|
||||
}
|
||||
|
||||
string URI::getQuery() {
|
||||
return query;
|
||||
}
|
||||
|
||||
bool URI::hasQuery() {
|
||||
return queryinit;
|
||||
}
|
||||
|
51
src/URI.h
51
src/URI.h
@ -1,51 +0,0 @@
|
||||
|
||||
#include <iostream>
|
||||
|
||||
#ifndef NECRONDA_PATH
|
||||
#define NECRONDA_PATH
|
||||
|
||||
using namespace std;
|
||||
|
||||
class URI {
|
||||
private:
|
||||
string webroot;
|
||||
string reqpath;
|
||||
string relpath;
|
||||
string query;
|
||||
string info;
|
||||
string filepath;
|
||||
string newpath;
|
||||
bool queryinit{};
|
||||
|
||||
public:
|
||||
URI();
|
||||
|
||||
URI(string webroot, string reqpath);
|
||||
|
||||
string getWebRoot();
|
||||
|
||||
string getRelativePath();
|
||||
|
||||
string getAbsolutePath();
|
||||
|
||||
string getFilePath();
|
||||
|
||||
string getRelativeFilePath();
|
||||
|
||||
string getNewPath();
|
||||
|
||||
FILE *openFile();
|
||||
|
||||
string getFilePathInfo();
|
||||
|
||||
string getFileType();
|
||||
|
||||
bool isStatic();
|
||||
|
||||
string getQuery();
|
||||
|
||||
bool hasQuery();
|
||||
|
||||
};
|
||||
|
||||
#endif
|
329
src/async.c
Normal file
329
src/async.c
Normal file
@ -0,0 +1,329 @@
|
||||
/**
|
||||
* Sesimos - secure, simple, modern web server
|
||||
* @brief Async handler
|
||||
* @file src/async.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-28
|
||||
*/
|
||||
|
||||
#include "async.h"
|
||||
#include "logger.h"
|
||||
#include "lib/list.h"
|
||||
#include "lib/utils.h"
|
||||
|
||||
#include <poll.h>
|
||||
#include <sys/epoll.h>
|
||||
#include <signal.h>
|
||||
#include <errno.h>
|
||||
#include <memory.h>
|
||||
#include <pthread.h>
|
||||
#include <semaphore.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#define ASYNC_MAX_EVENTS 16
|
||||
|
||||
typedef struct {
|
||||
int fd;
|
||||
sock *socket;
|
||||
async_evt_t events;
|
||||
int flags;
|
||||
void *arg;
|
||||
void (*cb)(void *);
|
||||
void (*to_cb)(void *);
|
||||
void (*err_cb)(void *);
|
||||
} evt_listen_t;
|
||||
|
||||
typedef struct {
|
||||
int n;
|
||||
evt_listen_t *q[ASYNC_MAX_EVENTS];
|
||||
} listen_queue_t;
|
||||
|
||||
static listen_queue_t listen1, listen2, *listen_q = &listen1;
|
||||
static volatile sig_atomic_t alive = 1;
|
||||
static pthread_t thread = -1;
|
||||
static sem_t lock;
|
||||
static int epoll_fd;
|
||||
|
||||
static short async_a2p(async_evt_t events) {
|
||||
short ret = 0;
|
||||
if (events & ASYNC_IN) ret |= POLLIN;
|
||||
if (events & ASYNC_PRI) ret |= POLLPRI;
|
||||
if (events & ASYNC_OUT) ret |= POLLOUT;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static unsigned int async_a2e(async_evt_t events) {
|
||||
unsigned int ret = 0;
|
||||
if (events & ASYNC_IN) ret |= EPOLLIN;
|
||||
if (events & ASYNC_PRI) ret |= EPOLLPRI;
|
||||
if (events & ASYNC_OUT) ret |= EPOLLOUT;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static async_evt_t async_p2a(short events) {
|
||||
async_evt_t ret = 0;
|
||||
if (events & POLLIN) ret |= ASYNC_IN;
|
||||
if (events & POLLPRI) ret |= ASYNC_PRI;
|
||||
if (events & POLLOUT) ret |= ASYNC_OUT;
|
||||
if (events & POLLERR) ret |= ASYNC_ERR;
|
||||
if (events & POLLHUP) ret |= ASYNC_HUP;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static async_evt_t async_e2a(unsigned int events) {
|
||||
async_evt_t ret = 0;
|
||||
if (events & EPOLLIN) ret |= ASYNC_IN;
|
||||
if (events & EPOLLPRI) ret |= ASYNC_PRI;
|
||||
if (events & EPOLLOUT) ret |= ASYNC_OUT;
|
||||
if (events & EPOLLERR) ret |= ASYNC_ERR;
|
||||
if (events & EPOLLHUP) ret |= ASYNC_HUP;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static int async_add_to_queue(evt_listen_t *evt) {
|
||||
try_again:
|
||||
if (sem_wait(&lock) != 0) {
|
||||
if (errno == EINTR) {
|
||||
goto try_again;
|
||||
} else {
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
evt_listen_t *ptr = malloc(sizeof(evt_listen_t));
|
||||
if (ptr == NULL) {
|
||||
sem_post(&lock);
|
||||
return -1;
|
||||
}
|
||||
|
||||
memcpy(ptr, evt, sizeof(*evt));
|
||||
listen_q->q[listen_q->n++] = ptr;
|
||||
|
||||
sem_post(&lock);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int async_exec(evt_listen_t *evt, async_evt_t r_events) {
|
||||
int ret, e = errno;
|
||||
if (r_events & evt->events) {
|
||||
// specified event(s) occurred
|
||||
if (evt->socket && !sock_has_pending(evt->socket)) {
|
||||
evt->err_cb(evt->arg);
|
||||
ret = 0;
|
||||
} else {
|
||||
evt->cb(evt->arg);
|
||||
ret = (evt->flags & ASYNC_KEEP) ? 1 : 0;
|
||||
}
|
||||
} else if (r_events & (POLLERR | POLLHUP | POLLNVAL)) {
|
||||
// error occurred
|
||||
evt->err_cb(evt->arg);
|
||||
ret = 0;
|
||||
} else {
|
||||
// no event occurred
|
||||
ret = -1;
|
||||
}
|
||||
|
||||
logger_set_prefix("");
|
||||
errno = e;
|
||||
return ret;
|
||||
}
|
||||
|
||||
static int async_check(evt_listen_t *evt) {
|
||||
struct pollfd fds[1] = {{
|
||||
.fd = evt->fd,
|
||||
.events = async_a2p(evt->events)
|
||||
}};
|
||||
|
||||
// check, if fd is already ready
|
||||
if (poll(fds, 1, 0) == 1) {
|
||||
// fd already ready
|
||||
if (async_exec(evt, async_p2a(fds[0].revents)) == 0)
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int async_add(evt_listen_t *evt) {
|
||||
if (async_check(evt) == 1)
|
||||
return 0;
|
||||
|
||||
int ret = async_add_to_queue(evt);
|
||||
if (ret == 0 && thread != -1)
|
||||
pthread_kill(thread, SIGUSR1);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int async_fd(int fd, async_evt_t events, int flags, void *arg, void cb(void *), void to_cb(void *), void err_cb(void *)) {
|
||||
evt_listen_t evt = {
|
||||
.fd = fd,
|
||||
.socket = NULL,
|
||||
.events = events,
|
||||
.flags = flags,
|
||||
.arg = arg,
|
||||
.cb = cb,
|
||||
.to_cb = to_cb,
|
||||
.err_cb = err_cb,
|
||||
};
|
||||
return async_add(&evt);
|
||||
}
|
||||
|
||||
int async(sock *s, async_evt_t events, int flags, void *arg, void cb(void *), void to_cb(void *), void err_cb(void *)) {
|
||||
evt_listen_t evt = {
|
||||
.fd = s->socket,
|
||||
.socket = s,
|
||||
.events = events,
|
||||
.flags = flags,
|
||||
.arg = arg,
|
||||
.cb = cb,
|
||||
.to_cb = to_cb,
|
||||
.err_cb = err_cb,
|
||||
};
|
||||
return async_add(&evt);
|
||||
}
|
||||
|
||||
int async_init(void) {
|
||||
if (sem_init(&lock, 0, 1) != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
listen1.n = 0;
|
||||
listen2.n = 0;
|
||||
|
||||
if ((epoll_fd = epoll_create1(0)) == -1) {
|
||||
async_free();
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void async_free(void) {
|
||||
int e = errno;
|
||||
sem_destroy(&lock);
|
||||
close(epoll_fd);
|
||||
errno = e;
|
||||
}
|
||||
|
||||
void async_thread(void) {
|
||||
struct epoll_event ev, events[ASYNC_MAX_EVENTS];
|
||||
int num_fds;
|
||||
long ts, min_ts, cur_ts;
|
||||
listen_queue_t *l;
|
||||
evt_listen_t **local = list_create(sizeof(evt_listen_t *), 16);
|
||||
if (local == NULL) {
|
||||
critical("Unable to create async local list");
|
||||
return;
|
||||
}
|
||||
|
||||
thread = pthread_self();
|
||||
|
||||
// main event loop
|
||||
while (alive) {
|
||||
// swap listen queue
|
||||
l = listen_q;
|
||||
listen_q = (listen_q == &listen1) ? &listen2 : &listen1;
|
||||
|
||||
// fill local list and epoll instance with previously added queue entries
|
||||
for (int i = 0; i < l->n; i++) {
|
||||
evt_listen_t *evt = l->q[i];
|
||||
local = list_append(local, &evt);
|
||||
if (local == NULL) {
|
||||
critical("Unable to resize async local list");
|
||||
return;
|
||||
}
|
||||
|
||||
ev.events = async_a2e(evt->events);
|
||||
ev.data.ptr = evt;
|
||||
|
||||
if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, evt->fd, &ev) == -1) {
|
||||
critical("Unable to add file descriptor to epoll instance");
|
||||
return;
|
||||
}
|
||||
}
|
||||
// reset size of queue
|
||||
l->n = 0;
|
||||
|
||||
// calculate wait timeout
|
||||
min_ts = -1000, cur_ts = clock_micros();;
|
||||
for (int i = 0; i < list_size(local); i++) {
|
||||
evt_listen_t *evt = local[i];
|
||||
if (!evt->socket) continue;
|
||||
|
||||
ts = evt->socket->ts_last + evt->socket->timeout_us - cur_ts;
|
||||
if (min_ts == -1000 || ts < min_ts) min_ts = ts;
|
||||
}
|
||||
|
||||
if ((num_fds = epoll_wait(epoll_fd, events, ASYNC_MAX_EVENTS, (int) (min_ts / 1000))) == -1) {
|
||||
if (errno == EINTR) {
|
||||
// interrupt
|
||||
errno = 0;
|
||||
continue;
|
||||
} else {
|
||||
// other error
|
||||
critical("Unable to poll for events");
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
for (int i = 0; i < num_fds; i++) {
|
||||
evt_listen_t *evt = events[i].data.ptr;
|
||||
if (async_exec(evt, async_e2a(events[i].events)) == 0) {
|
||||
if (epoll_ctl(epoll_fd, EPOLL_CTL_DEL, evt->fd, NULL) == -1) {
|
||||
if (errno == EBADF) {
|
||||
// already closed fd, do not die
|
||||
errno = 0;
|
||||
} else {
|
||||
critical("Unable to remove file descriptor from epoll instance");
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
local = list_delete(local, &evt);
|
||||
if (local == NULL) {
|
||||
critical("Unable to resize async local list");
|
||||
return;
|
||||
}
|
||||
|
||||
free(evt);
|
||||
}
|
||||
}
|
||||
|
||||
// check, if some socket ran into a timeout
|
||||
cur_ts = clock_micros();
|
||||
for (int i = 0; i < list_size(local); i++) {
|
||||
evt_listen_t *evt = local[i];
|
||||
if (!evt->socket) continue;
|
||||
|
||||
if ((cur_ts - evt->socket->ts_last) >= evt->socket->timeout_us) {
|
||||
evt->to_cb(evt->arg);
|
||||
|
||||
if (epoll_ctl(epoll_fd, EPOLL_CTL_DEL, evt->fd, NULL) == -1) {
|
||||
if (errno == EBADF) {
|
||||
// already closed fd, do not die
|
||||
errno = 0;
|
||||
} else {
|
||||
critical("Unable to remove file descriptor from epoll instance");
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
local = list_remove(local, i--);
|
||||
}
|
||||
}
|
||||
logger_set_prefix("");
|
||||
errno = 0;
|
||||
}
|
||||
|
||||
// cleanup
|
||||
for (int i = 0; i < list_size(local); i++) {
|
||||
free(local[i]);
|
||||
}
|
||||
list_free(local);
|
||||
}
|
||||
|
||||
void async_stop(void) {
|
||||
alive = 0;
|
||||
}
|
39
src/async.h
Normal file
39
src/async.h
Normal file
@ -0,0 +1,39 @@
|
||||
/**
|
||||
* Sesimos - secure, simple, modern web server
|
||||
* @brief Async handler (header file)
|
||||
* @file src/async.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-28
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_ASYNC_H
|
||||
#define SESIMOS_ASYNC_H
|
||||
|
||||
#include "lib/sock.h"
|
||||
|
||||
#define ASYNC_KEEP 1
|
||||
|
||||
#define ASYNC_IN 0x01
|
||||
#define ASYNC_PRI 0x02
|
||||
#define ASYNC_OUT 0x04
|
||||
#define ASYNC_ERR 0x08
|
||||
#define ASYNC_HUP 0x10
|
||||
|
||||
#define ASYNC_WAIT_READ ASYNC_IN
|
||||
#define ASYNC_WAIT_WRITE ASYNC_OUT
|
||||
|
||||
typedef unsigned int async_evt_t;
|
||||
|
||||
int async(sock *s, async_evt_t events, int flags, void *arg, void cb(void *), void to_cb(void *), void err_cb(void *));
|
||||
|
||||
int async_fd(int fd, async_evt_t events, int flags, void *arg, void cb(void *), void to_cb(void *), void err_cb(void *));
|
||||
|
||||
int async_init(void);
|
||||
|
||||
void async_free(void);
|
||||
|
||||
void async_thread(void);
|
||||
|
||||
void async_stop(void);
|
||||
|
||||
#endif //SESIMOS_ASYNC_H
|
397
src/cache_handler.c
Normal file
397
src/cache_handler.c
Normal file
@ -0,0 +1,397 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief File cache implementation
|
||||
* @file src/cache_handler.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-19
|
||||
*/
|
||||
|
||||
#include "logger.h"
|
||||
#include "cache_handler.h"
|
||||
#include "lib/utils.h"
|
||||
#include "lib/compress.h"
|
||||
#include "lib/config.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <magic.h>
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
#include <openssl/evp.h>
|
||||
#include <sys/mman.h>
|
||||
#include <fcntl.h>
|
||||
#include <pthread.h>
|
||||
#include <semaphore.h>
|
||||
#include <signal.h>
|
||||
|
||||
#define CACHE_BUF_SIZE 16
|
||||
|
||||
|
||||
static magic_t magic;
|
||||
static pthread_t thread;
|
||||
static sem_t sem_free, sem_used, sem_lock;
|
||||
volatile sig_atomic_t alive = 1;
|
||||
|
||||
typedef struct {
|
||||
int rd;
|
||||
int wr;
|
||||
cache_entry_t *msgs[CACHE_BUF_SIZE];
|
||||
} buf_t;
|
||||
|
||||
static buf_t buffer;
|
||||
|
||||
static int magic_init(void) {
|
||||
if ((magic = magic_open(MAGIC_MIME)) == NULL) {
|
||||
critical("Unable to open magic cookie");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (magic_load(magic, CACHE_MAGIC_FILE) != 0) {
|
||||
critical("Unable to load magic cookie: %s", magic_error(magic));
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void cache_free(void) {
|
||||
for (int i = 0; i < CONFIG_MAX_HOST_CONFIG; i++) {
|
||||
host_config_t *hc = &config.hosts[i];
|
||||
if (hc->type == CONFIG_TYPE_UNSET) break;
|
||||
if (hc->type != CONFIG_TYPE_LOCAL) continue;
|
||||
|
||||
munmap(hc->cache, sizeof(cache_t));
|
||||
}
|
||||
}
|
||||
|
||||
static cache_entry_t *cache_get_entry(cache_t *cache, const char *filename) {
|
||||
// search entry
|
||||
cache_entry_t *entry;
|
||||
for (int i = 0; i < CACHE_ENTRIES; i++) {
|
||||
entry = &cache->entries[i];
|
||||
if (entry->filename[0] == 0) break;
|
||||
if (strcmp(entry->filename, filename) == 0) {
|
||||
// found
|
||||
return entry;
|
||||
}
|
||||
}
|
||||
|
||||
// not found
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static cache_entry_t *cache_get_new_entry(cache_t *cache) {
|
||||
// search empty slot
|
||||
cache_entry_t *entry;
|
||||
for (int i = 0; i < CACHE_ENTRIES; i++) {
|
||||
entry = &cache->entries[i];
|
||||
if (entry->filename[0] == 0)
|
||||
return entry;
|
||||
}
|
||||
|
||||
// not found
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static void cache_process_entry(cache_entry_t *entry) {
|
||||
char buf[16384], comp_buf[16384], filename_comp_gz[256], filename_comp_br[256];
|
||||
|
||||
info("Hashing file %s", entry->filename);
|
||||
|
||||
EVP_MD_CTX *ctx = EVP_MD_CTX_new();
|
||||
EVP_DigestInit(ctx, EVP_sha1());
|
||||
FILE *file = fopen(entry->filename, "rb");
|
||||
int compress = mime_is_compressible(entry->meta.type);
|
||||
|
||||
compress_ctx comp_ctx;
|
||||
FILE *comp_file_gz = NULL, *comp_file_br = NULL;
|
||||
if (compress) {
|
||||
sprintf(buf, "%.*s/.sesimos", entry->webroot_len, entry->filename);
|
||||
if (mkdir(buf, 0755) != 0 && errno != EEXIST) {
|
||||
error("Unable to create directory %s", buf);
|
||||
goto comp_err;
|
||||
}
|
||||
|
||||
sprintf(buf, "%.*s/.sesimos/cache", entry->webroot_len, entry->filename);
|
||||
if (mkdir(buf, 0700) != 0 && errno != EEXIST) {
|
||||
error("Unable to create directory %s", buf);
|
||||
goto comp_err;
|
||||
}
|
||||
errno = 0;
|
||||
|
||||
char *rel_path = entry->filename + entry->webroot_len + 1;
|
||||
for (int j = 0; j < strlen(rel_path); j++) {
|
||||
char ch = rel_path[j];
|
||||
if (ch == '/') ch = '_';
|
||||
buf[j] = ch;
|
||||
}
|
||||
buf[strlen(rel_path)] = 0;
|
||||
|
||||
int p_len_gz = snprintf(filename_comp_gz, sizeof(filename_comp_gz),
|
||||
"%.*s/.sesimos/cache/%s.gz",
|
||||
entry->webroot_len, entry->filename, buf);
|
||||
int p_len_br = snprintf(filename_comp_br, sizeof(filename_comp_br),
|
||||
"%.*s/.sesimos/cache/%s.br",
|
||||
entry->webroot_len, entry->filename, buf);
|
||||
if (p_len_gz < 0 || p_len_gz >= sizeof(filename_comp_gz) || p_len_br < 0 || p_len_br >= sizeof(filename_comp_br)) {
|
||||
error("Unable to open cached file: File name for compressed file too long");
|
||||
goto comp_err;
|
||||
}
|
||||
|
||||
info("Compressing file %s", entry->filename);
|
||||
|
||||
comp_file_gz = fopen(filename_comp_gz, "wb");
|
||||
comp_file_br = fopen(filename_comp_br, "wb");
|
||||
if (comp_file_gz == NULL || comp_file_br == NULL) {
|
||||
error("Unable to open cached file");
|
||||
comp_err:
|
||||
compress = 0;
|
||||
} else {
|
||||
if ((compress_init(&comp_ctx, COMPRESS_GZ | COMPRESS_BR)) != 0) {
|
||||
error("Unable to init compression");
|
||||
compress = 0;
|
||||
fclose(comp_file_gz);
|
||||
fclose(comp_file_br);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
unsigned long read;
|
||||
while ((read = fread(buf, 1, sizeof(buf), file)) > 0) {
|
||||
EVP_DigestUpdate(ctx, buf, read);
|
||||
if (compress) {
|
||||
unsigned long avail_in, avail_out;
|
||||
avail_in = read;
|
||||
do {
|
||||
avail_out = sizeof(comp_buf);
|
||||
compress_compress_mode(&comp_ctx, COMPRESS_GZ, buf + read - avail_in, &avail_in, comp_buf, &avail_out, feof(file));
|
||||
fwrite(comp_buf, 1, sizeof(comp_buf) - avail_out, comp_file_gz);
|
||||
} while (avail_in != 0 || avail_out != sizeof(comp_buf));
|
||||
avail_in = read;
|
||||
do {
|
||||
avail_out = sizeof(comp_buf);
|
||||
compress_compress_mode(&comp_ctx, COMPRESS_BR, buf + read - avail_in, &avail_in, comp_buf, &avail_out, feof(file));
|
||||
fwrite(comp_buf, 1, sizeof(comp_buf) - avail_out, comp_file_br);
|
||||
} while (avail_in != 0 || avail_out != sizeof(comp_buf));
|
||||
}
|
||||
}
|
||||
|
||||
if (compress) {
|
||||
compress_free(&comp_ctx);
|
||||
fclose(comp_file_gz);
|
||||
fclose(comp_file_br);
|
||||
info("Finished compressing file %s", entry->filename);
|
||||
strcpy(entry->meta.filename_comp_gz, filename_comp_gz);
|
||||
strcpy(entry->meta.filename_comp_br, filename_comp_br);
|
||||
} else {
|
||||
memset(entry->meta.filename_comp_gz, 0, sizeof(entry->meta.filename_comp_gz));
|
||||
memset(entry->meta.filename_comp_br, 0, sizeof(entry->meta.filename_comp_br));
|
||||
}
|
||||
|
||||
unsigned int md_len;
|
||||
unsigned char hash[EVP_MAX_MD_SIZE];
|
||||
EVP_DigestFinal(ctx, hash, &md_len);
|
||||
EVP_MD_CTX_free(ctx);
|
||||
|
||||
memset(entry->meta.etag, 0, sizeof(entry->meta.etag));
|
||||
for (int j = 0; j < md_len; j++) {
|
||||
sprintf(entry->meta.etag + j * 2, "%02x", hash[j]);
|
||||
}
|
||||
fclose(file);
|
||||
entry->flags &= !CACHE_DIRTY;
|
||||
|
||||
info("Finished hashing file %s", entry->filename);
|
||||
}
|
||||
|
||||
static void *cache_thread(void *arg) {
|
||||
logger_set_name("cache");
|
||||
|
||||
while (alive) {
|
||||
pthread_testcancel();
|
||||
if (sem_wait(&sem_used) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
continue;
|
||||
} else {
|
||||
error("Unable to lock semaphore");
|
||||
errno = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
cache_entry_t *entry = buffer.msgs[buffer.wr];
|
||||
buffer.wr = (buffer.wr + 1) % CACHE_BUF_SIZE;
|
||||
|
||||
cache_process_entry(entry);
|
||||
|
||||
// unlock slot in buffer
|
||||
sem_post(&sem_free);
|
||||
}
|
||||
|
||||
cache_free();
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
int cache_init(void) {
|
||||
char buf[512];
|
||||
int ret, fd;
|
||||
if ((ret = magic_init()) != 0)
|
||||
return ret;
|
||||
|
||||
for (int i = 0; i < CONFIG_MAX_HOST_CONFIG; i++) {
|
||||
host_config_t *hc = &config.hosts[i];
|
||||
if (hc->type == CONFIG_TYPE_UNSET) break;
|
||||
if (hc->type != CONFIG_TYPE_LOCAL) continue;
|
||||
|
||||
sprintf(buf, "%s/.sesimos/metadata", hc->local.webroot);
|
||||
if ((fd = open(buf, O_CREAT | O_RDWR, 0600)) == -1) {
|
||||
critical("Unable to open file %s", buf);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (ftruncate(fd, sizeof(cache_t)) == -1) {
|
||||
critical("Unable to truncate file %s", buf);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if ((hc->cache = mmap(NULL, sizeof(cache_t), PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0)) == NULL) {
|
||||
critical("Unable to map file %s", buf);
|
||||
close(fd);
|
||||
return 1;
|
||||
}
|
||||
|
||||
close(fd);
|
||||
errno = 0;
|
||||
}
|
||||
|
||||
// try to initialize all three semaphores
|
||||
if (sem_init(&sem_lock, 0, 1) != 0 ||
|
||||
sem_init(&sem_free, 0, 1) != 0 ||
|
||||
sem_init(&sem_used, 0, 0) != 0)
|
||||
{
|
||||
critical("Unable to initialize semaphore");
|
||||
return -1;
|
||||
}
|
||||
|
||||
// initialize read/write heads
|
||||
buffer.rd = 0;
|
||||
buffer.wr = 0;
|
||||
|
||||
pthread_create(&thread, NULL, cache_thread, NULL);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void cache_stop(void) {
|
||||
alive = 0;
|
||||
pthread_kill(thread, SIGUSR1);
|
||||
}
|
||||
|
||||
int cache_join(void) {
|
||||
return pthread_join(thread, NULL);
|
||||
}
|
||||
|
||||
static void cache_mark_entry_dirty(cache_entry_t *entry) {
|
||||
if (entry->flags & CACHE_DIRTY)
|
||||
return;
|
||||
|
||||
memset(entry->meta.etag, 0, sizeof(entry->meta.etag));
|
||||
memset(entry->meta.filename_comp_gz, 0, sizeof(entry->meta.filename_comp_gz));
|
||||
memset(entry->meta.filename_comp_br, 0, sizeof(entry->meta.filename_comp_br));
|
||||
entry->flags |= CACHE_DIRTY;
|
||||
|
||||
try_again_free:
|
||||
if (sem_wait(&sem_free) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
goto try_again_free;
|
||||
} else {
|
||||
error("Unable to lock semaphore");
|
||||
errno = 0;
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
// try to lock buffer
|
||||
try_again_lock:
|
||||
if (sem_wait(&sem_lock) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
goto try_again_lock;
|
||||
} else {
|
||||
error("Unable to lock semaphore");
|
||||
errno = 0;
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
// write to buffer
|
||||
buffer.msgs[buffer.rd] = entry;
|
||||
buffer.rd = (buffer.rd + 1) % CACHE_BUF_SIZE;
|
||||
|
||||
// unlock buffer
|
||||
sem_post(&sem_lock);
|
||||
|
||||
// unlock slot in buffer for logger
|
||||
sem_post(&sem_used);
|
||||
}
|
||||
|
||||
static void cache_update_entry(cache_entry_t *entry, const char *filename, const char *webroot) {
|
||||
struct stat statbuf;
|
||||
stat(filename, &statbuf);
|
||||
memcpy(&entry->meta.stat, &statbuf, sizeof(statbuf));
|
||||
|
||||
entry->webroot_len = (unsigned char) strlen(webroot);
|
||||
strcpy(entry->filename, filename);
|
||||
|
||||
magic_setflags(magic, MAGIC_MIME_TYPE);
|
||||
const char *type = magic_file(magic, filename);
|
||||
char type_new[URI_TYPE_SIZE];
|
||||
sprintf(type_new, "%s", type);
|
||||
if (strncmp(type, "text/", 5) == 0) {
|
||||
if (strcmp(filename + strlen(filename) - 4, ".css") == 0) {
|
||||
sprintf(type_new, "text/css");
|
||||
} else if (strcmp(filename + strlen(filename) - 3, ".js") == 0) {
|
||||
sprintf(type_new, "application/javascript");
|
||||
}
|
||||
}
|
||||
strcpy(entry->meta.type, type_new);
|
||||
|
||||
magic_setflags(magic, MAGIC_MIME_ENCODING);
|
||||
strcpy(entry->meta.charset, magic_file(magic, filename));
|
||||
|
||||
cache_mark_entry_dirty(entry);
|
||||
}
|
||||
|
||||
void cache_mark_dirty(cache_t *cache, const char *filename) {
|
||||
cache_entry_t *entry = cache_get_entry(cache, filename);
|
||||
if (entry) cache_mark_entry_dirty(entry);
|
||||
}
|
||||
|
||||
void cache_init_uri(cache_t *cache, http_uri *uri) {
|
||||
if (!uri->filename)
|
||||
return;
|
||||
|
||||
cache_entry_t *entry = cache_get_entry(cache, uri->filename);
|
||||
if (!entry) {
|
||||
// no entry found -> create new entry
|
||||
if ((entry = cache_get_new_entry(cache))) {
|
||||
cache_update_entry(entry, uri->filename, uri->webroot);
|
||||
uri->meta = &entry->meta;
|
||||
} else {
|
||||
warning("No empty cache entry slot found");
|
||||
}
|
||||
} else {
|
||||
uri->meta = &entry->meta;
|
||||
if (entry->flags & CACHE_DIRTY)
|
||||
return;
|
||||
|
||||
// check, if file has changed
|
||||
struct stat statbuf;
|
||||
stat(uri->filename, &statbuf);
|
||||
if (memcmp(&uri->meta->stat.st_mtime, &statbuf.st_mtime, sizeof(statbuf.st_mtime)) != 0) {
|
||||
// modify time has changed
|
||||
cache_update_entry(entry, uri->filename, uri->webroot);
|
||||
}
|
||||
}
|
||||
}
|
46
src/cache_handler.h
Normal file
46
src/cache_handler.h
Normal file
@ -0,0 +1,46 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief File cache implementation (header file)
|
||||
* @file src/cache_handler.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-19
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_CACHE_HANDLER_H
|
||||
#define SESIMOS_CACHE_HANDLER_H
|
||||
|
||||
#include "lib/uri.h"
|
||||
|
||||
#define CACHE_ENTRIES 1024
|
||||
|
||||
#define CACHE_DIRTY 1
|
||||
|
||||
#ifndef CACHE_MAGIC_FILE
|
||||
# define CACHE_MAGIC_FILE "/usr/share/file/misc/magic.mgc"
|
||||
#endif
|
||||
|
||||
|
||||
typedef struct {
|
||||
char filename[256];
|
||||
unsigned char webroot_len;
|
||||
unsigned char flags;
|
||||
metadata_t meta;
|
||||
} cache_entry_t;
|
||||
|
||||
typedef struct {
|
||||
char sig[6];
|
||||
unsigned char ver;
|
||||
cache_entry_t entries[CACHE_ENTRIES];
|
||||
} cache_t;
|
||||
|
||||
int cache_init(void);
|
||||
|
||||
void cache_stop(void);
|
||||
|
||||
int cache_join(void);
|
||||
|
||||
void cache_mark_dirty(cache_t *cache, const char *filename);
|
||||
|
||||
void cache_init_uri(cache_t *cache, http_uri *uri);
|
||||
|
||||
#endif //SESIMOS_CACHE_HANDLER_H
|
673
src/client.cpp
673
src/client.cpp
@ -1,673 +0,0 @@
|
||||
#include <utility>
|
||||
|
||||
#include <utility>
|
||||
|
||||
/**
|
||||
* Necronda Web Server 3.0
|
||||
* client.cpp - Client and Connection handler
|
||||
* Lorenz Stechauner, 2018-05-16
|
||||
*/
|
||||
|
||||
|
||||
#include <string>
|
||||
#include <iostream>
|
||||
#include <zlib.h>
|
||||
#include <cassert>
|
||||
#include <fstream>
|
||||
#include <openssl/md5.h>
|
||||
#include <cstring>
|
||||
#include <fcntl.h>
|
||||
#include <sstream>
|
||||
#include <netinet/in.h>
|
||||
#include <netinet/tcp.h>
|
||||
|
||||
#include "network/Socket.h"
|
||||
#include "network/http/HttpRequest.h"
|
||||
#include "network/http/HttpConnection.h"
|
||||
#include "necronda-server.h"
|
||||
#include "network/http/HttpStatusCode.h"
|
||||
#include "URI.h"
|
||||
#include "procopen.h"
|
||||
#include "network/Address.h"
|
||||
|
||||
|
||||
typedef struct {
|
||||
string host;
|
||||
string cc;
|
||||
string country;
|
||||
string prov;
|
||||
string provname;
|
||||
string city;
|
||||
string timezone;
|
||||
string localdate;
|
||||
} IpAddressInfo;
|
||||
|
||||
|
||||
void log_to_file(const char *prefix, const string &str, string host) {
|
||||
//FILE *file = fopen((getWebRoot(std::move(host)) + ".access.log").c_str(), "a");
|
||||
//fprintf(file, "%s%s\r\n", prefix, str.c_str());
|
||||
//fflush(file);
|
||||
//fclose(file);
|
||||
}
|
||||
|
||||
void log_error_to_file(const char *prefix, const string &str, string host) {
|
||||
log_to_file(prefix, "\x1B[1;31m" + str + "\x1B[0m", std::move(host));
|
||||
}
|
||||
|
||||
/**
|
||||
* Writes log messages to the console
|
||||
* @param prefix The connection prefix
|
||||
* @param str The string to be written
|
||||
*/
|
||||
void log(const char *prefix, const string &str) {
|
||||
printf("%s%s\r\n", prefix, str.c_str());
|
||||
flush(cout);
|
||||
}
|
||||
|
||||
void log_error(const char *prefix, const string &str) {
|
||||
log(prefix, "\x1B[1;31m" + str + "\x1B[0m");
|
||||
}
|
||||
|
||||
void php_error_handler(const char *prefix, FILE *stderr) {
|
||||
string line;
|
||||
while (!(line = read_line(stderr)).empty()) {
|
||||
log_error(prefix, line);
|
||||
}
|
||||
fclose(stderr);
|
||||
}
|
||||
|
||||
IpAddressInfo get_ip_address_info(Address* addr) {
|
||||
FILE *name = popen(("/opt/ipinfo/ipinfo.py " + addr->toString()).c_str(), "r");
|
||||
char hostbuffer[1024];
|
||||
memset(hostbuffer, 0, 1024);
|
||||
size_t size = fread(hostbuffer, 1, 1024, name);
|
||||
istringstream buffer(hostbuffer);
|
||||
string line;
|
||||
|
||||
IpAddressInfo info;
|
||||
int num = 0;
|
||||
while (std::getline(buffer, line)) {
|
||||
switch (num) {
|
||||
case 0: info.host = line; break;
|
||||
case 1: info.cc = line; break;
|
||||
case 2: info.country = line; break;
|
||||
case 3: info.prov = line; break;
|
||||
case 4: info.provname = line; break;
|
||||
case 5: info.city = line; break;
|
||||
case 6: info.timezone = line; break;
|
||||
case 7: info.localdate = line; break;
|
||||
}
|
||||
num++;
|
||||
}
|
||||
return info;
|
||||
}
|
||||
|
||||
string get_os_info(int fd) {
|
||||
struct tcp_info ti;
|
||||
socklen_t tisize = sizeof(ti);
|
||||
getsockopt(fd, IPPROTO_TCP, TCP_INFO, &ti, &tisize);
|
||||
|
||||
int ttl;
|
||||
socklen_t ttlsize = sizeof(ttl);
|
||||
getsockopt(fd, IPPROTO_IP, IP_TTL, &ttl, &ttlsize);
|
||||
|
||||
return "win_size=" + to_string(ti.tcpi_rcv_space) + ", ttl=" + to_string(ttl);
|
||||
}
|
||||
|
||||
string getETag(string filename) {
|
||||
ifstream etags = ifstream("/var/necronda/ETags");
|
||||
|
||||
ifstream a = ifstream();
|
||||
string line;
|
||||
int index = 0;
|
||||
int i = 0;
|
||||
string timestamp = getTimestamp(filename);
|
||||
long size = getFileSize(filename);
|
||||
while (getline(etags, line)) {
|
||||
i++;
|
||||
if (line == filename) {
|
||||
index = i;
|
||||
break;
|
||||
}
|
||||
long p1 = line.find(':');
|
||||
if (p1 == string::npos) continue;
|
||||
long p2 = line.find(':', (unsigned) p1 + 1);
|
||||
if (p2 == string::npos) continue;
|
||||
long p3 = line.find(':', (unsigned) p2 + 1);
|
||||
if (p3 == string::npos) continue;
|
||||
string FILENAME = line.substr(0, (unsigned) p1);
|
||||
string HASH = line.substr((unsigned) p1 + 1, (unsigned) (p2 - p1));
|
||||
string TIMESTAMP = line.substr((unsigned) p2 + 1, (unsigned) (p3 - p2));
|
||||
long SIZE = strtol(line.substr((unsigned) p3 + 1, line.length() - p3).c_str(), nullptr, 10);
|
||||
if (FILENAME == filename) {
|
||||
index = i;
|
||||
if (timestamp != TIMESTAMP || size != SIZE) {
|
||||
break;
|
||||
} else {
|
||||
etags.close();
|
||||
return HASH;
|
||||
}
|
||||
}
|
||||
}
|
||||
etags.close();
|
||||
|
||||
MD5_CTX mdContext;
|
||||
MD5_Init(&mdContext);
|
||||
size_t bytes;
|
||||
char buffer[4096];
|
||||
FILE *file = fopen(filename.c_str(), "rb");
|
||||
if (file == nullptr) {
|
||||
throw (char *) "Invalid file";
|
||||
}
|
||||
while ((bytes = fread(buffer, 1, 4096, file)) != 0) {
|
||||
MD5_Update(&mdContext, buffer, bytes);
|
||||
}
|
||||
fclose(file);
|
||||
unsigned char md[16];
|
||||
MD5_Final(md, &mdContext);
|
||||
char md5buff[32];
|
||||
for (int i = 0; i < 16; i++) {
|
||||
sprintf(md5buff + i * 2, "%02x", md[i]);
|
||||
}
|
||||
string md5 = string(md5buff);
|
||||
|
||||
if (index == 0) {
|
||||
char buff[256];
|
||||
sprintf(buff, "%s:%s:%s:%ld\n", filename.c_str(), md5.c_str(), timestamp.c_str(), size);
|
||||
FILE *f = fopen("/var/necronda/ETags", "a");
|
||||
if (f == nullptr) {
|
||||
throw (char *) strerror(errno);
|
||||
}
|
||||
fseek(f, 0, SEEK_END);
|
||||
fwrite(buff, 1, strlen(buff), f);
|
||||
fflush(f);
|
||||
fclose(f);
|
||||
} else {
|
||||
|
||||
}
|
||||
|
||||
return md5;
|
||||
}
|
||||
|
||||
#include <iostream>
|
||||
#include <wait.h>
|
||||
#include <thread>
|
||||
|
||||
|
||||
long getPosition(std::string str, char c, int occurence) {
|
||||
int tempOccur = 0;
|
||||
int num = 0;
|
||||
for (auto it : str) {
|
||||
num++;
|
||||
if (it == c) {
|
||||
if (++tempOccur == occurence) {
|
||||
return num;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
int websocket_handler(Socket *socket, stds *pipes) {
|
||||
fd_set readfd;
|
||||
int maxfd = (socket->getFd() > pipes->stdout->_fileno) ? socket->getFd() : pipes->stdout->_fileno;
|
||||
FD_ZERO(&readfd);
|
||||
FD_SET(socket->getFd(), &readfd);
|
||||
FD_SET(pipes->stdout->_fileno, &readfd);
|
||||
|
||||
/*while (true) {
|
||||
int ret = ::select(maxfd + 1, &readfd, nullptr, nullptr, nullptr);
|
||||
if (ret < 0) {
|
||||
throw (char *) strerror(errno);
|
||||
}
|
||||
|
||||
int c = fgetc(pipes->stdout);
|
||||
if (c == -1) {
|
||||
long rec = socket->receive(pipes->stdin);
|
||||
} else {
|
||||
ungetc(c, pipes->stdout);
|
||||
socket->send(pipes->stdout);
|
||||
}
|
||||
}*/
|
||||
}
|
||||
|
||||
/**
|
||||
* Handles (keep-alive) HTTP connections
|
||||
* @param prefix The connection prefix
|
||||
* @param socket The socket
|
||||
* @param id The client ID
|
||||
* @param num The Connection Number in the client
|
||||
* @return Should the server wait for another header?
|
||||
*/
|
||||
bool connection_handler(const char *preprefix, const char *col1, const char *col2, Socket *socket, long id, long num, IpAddressInfo *info) {
|
||||
bool error = false;
|
||||
char buffer[1024];
|
||||
char *prefix = (char *) preprefix;
|
||||
|
||||
HttpConnection req;
|
||||
try {
|
||||
req = HttpConnection(socket);
|
||||
} catch (char *msg) {
|
||||
try {
|
||||
if (msg == "Malformed header") {
|
||||
log(prefix, "Unable to parse header: Malformed header");
|
||||
socket->send("HTTP/1.1 400 Bad Request\r\nConnection: close\r\n\r\n");
|
||||
return false;
|
||||
} else if (msg == "timeout") {
|
||||
log(prefix, "Timeout!");
|
||||
socket->send("HTTP/1.1 408 Request Timeout\r\nConnection: close\r\n\r\n");
|
||||
return false;
|
||||
} else {
|
||||
log(prefix, (string) "Unable to receive from socket: " + msg);
|
||||
return false;
|
||||
}
|
||||
} catch (char *msg2) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
bool noRedirect, redir, invalidMethod, etag, compress, wantsWebsocket, websocket = false;
|
||||
URI path;
|
||||
pid_t childpid;
|
||||
FILE *file;
|
||||
int statuscode;
|
||||
string hash, type, host;
|
||||
thread *t;
|
||||
long pos;
|
||||
stds pipes;
|
||||
|
||||
if (req.isExistingField("Connection") && req.getField("Connection") == "keep-alive") {
|
||||
req.setField("Connection", "keep-alive");
|
||||
req.setField("Keep-Alive", "timeout=3600, max=100");
|
||||
} else {
|
||||
req.setField("Connection", "close");
|
||||
error = true;
|
||||
}
|
||||
|
||||
host = "";
|
||||
|
||||
if (!req.isExistingField("Host")) {
|
||||
req.respond(400);
|
||||
goto respond;
|
||||
}
|
||||
host = req.getField("Host");
|
||||
pos = host.find(':');
|
||||
if (pos != string::npos) {
|
||||
host.erase(pos, host.length() - pos);
|
||||
}
|
||||
|
||||
/*
|
||||
FILE *name = popen(("dig @8.8.8.8 +time=1 -x " + socket->getPeerAddress()->toString() +
|
||||
" | grep -oP \"^[^;].*\\t\\K([^ ]*)\\w\"").c_str(), "r");
|
||||
char hostbuffer[1024];
|
||||
memset(hostbuffer, 0, 1024);
|
||||
size_t size = fread(hostbuffer, 1, 1024, name);
|
||||
hostbuffer[size - 1] = 0; // remove \n
|
||||
if (size <= 1) {
|
||||
sprintf(hostbuffer, "%s", socket->getPeerAddress()->toString().c_str());
|
||||
}
|
||||
*/
|
||||
|
||||
sprintf(buffer, "[\x1B[1m%s\x1B[0m][%i]%s[%s][%i]%s ", host.c_str(), socket->getSocketPort(), col1,
|
||||
info->host.c_str(), socket->getPeerPort(), col2);
|
||||
prefix = buffer;
|
||||
|
||||
log(prefix, "\x1B[1m" + req.getMethod() + " " + req.getPath() + "\x1B[0m");
|
||||
log_to_file(prefix, "\x1B[1m" + req.getMethod() + " " + req.getPath() + "\x1B[0m", host);
|
||||
|
||||
noRedirect = req.getPath().find("/.well-known/") == 0 || (req.getPath().find("/files/") == 0);
|
||||
redir = true;
|
||||
if (!noRedirect) {
|
||||
if (getWebRoot(host).empty()) {
|
||||
req.redirect(303, "https://www.necronda.net" + req.getPath());
|
||||
} else if (socket->getSocketPort() != 443) {
|
||||
req.redirect(302, "https://" + host + req.getPath());
|
||||
} else {
|
||||
redir = false;
|
||||
}
|
||||
} else {
|
||||
redir = false;
|
||||
}
|
||||
|
||||
path = URI(getWebRoot(host), req.getPath());
|
||||
childpid = 0;
|
||||
|
||||
if (redir) {
|
||||
goto respond;
|
||||
} else if (!path.getNewPath().empty() && req.getMethod() != "POST") {
|
||||
req.redirect(303, path.getNewPath());
|
||||
goto respond;
|
||||
}
|
||||
|
||||
file = path.openFile();
|
||||
if (file == nullptr) {
|
||||
req.setField("Cache-Control", "public, max-age=60");
|
||||
req.respond(404);
|
||||
goto respond;
|
||||
}
|
||||
type = path.getFileType();
|
||||
|
||||
if (type.find("inode/") == 0 || (path.getRelativeFilePath().find("/.") != string::npos && !noRedirect)) {
|
||||
req.respond(403);
|
||||
goto respond;
|
||||
}
|
||||
|
||||
req.setField("Content-Type", type);
|
||||
req.setField("Last-Modified", getHttpDate(path.getFilePath()));
|
||||
|
||||
invalidMethod = false;
|
||||
etag = false;
|
||||
|
||||
if (path.isStatic()) {
|
||||
hash = getETag(path.getFilePath());
|
||||
req.setField("ETag", hash);
|
||||
req.setField("Accept-Ranges", "bytes");
|
||||
if (type.find("text/") == 0) {
|
||||
req.setField("Cache-Control", "public, max-age=3600");
|
||||
} else {
|
||||
req.setField("Cache-Control", "public, max-age=86400");
|
||||
}
|
||||
req.setField("Allow", "GET");
|
||||
if (req.getMethod() != "GET") {
|
||||
invalidMethod = true;
|
||||
}
|
||||
if (req.isExistingField("If-None-Match") && req.getField("If-None-Match") == hash) {
|
||||
etag = true;
|
||||
}
|
||||
} else {
|
||||
req.setField("Accept-Ranges", "none");
|
||||
req.setField("Cache-Control", "private, no-cache");
|
||||
req.setField("Allow", "GET, POST, PUT");
|
||||
if (req.getMethod() != "GET" && req.getMethod() != "POST" && req.getMethod() != "PUT") {
|
||||
invalidMethod = true;
|
||||
}
|
||||
}
|
||||
|
||||
if (invalidMethod) {
|
||||
req.respond(405);
|
||||
goto respond;
|
||||
} else if (etag) {
|
||||
req.respond(304);
|
||||
goto respond;
|
||||
}
|
||||
|
||||
statuscode = 0;
|
||||
if (!path.isStatic()) {
|
||||
string cmd = (string) "env -i" +
|
||||
" REDIRECT_STATUS=" + cli_encode("CGI") +
|
||||
" DOCUMENT_ROOT=" + cli_encode(getWebRoot(host)) +
|
||||
" " + req.cgiExport() +
|
||||
(req.isExistingField("Content-Length") ? " CONTENT_LENGTH=" +
|
||||
cli_encode(req.getField(
|
||||
"Content-Length"))
|
||||
: "") +
|
||||
(req.isExistingField("Content-Type") ? " CONTENT_TYPE=" + cli_encode(
|
||||
req.getField("Content-Type")) : "") +
|
||||
((socket->isSecured()) ? " HTTPS=on" : "") +
|
||||
" PATH_INFO=" + cli_encode(path.getFilePathInfo()) +
|
||||
" PATH_TRANSLATED=" + cli_encode(path.getAbsolutePath()) +
|
||||
" QUERY_STRING=" + cli_encode(path.getQuery()) +
|
||||
" REMOTE_ADDR=" + cli_encode(socket->getPeerAddress()->toString()) +
|
||||
" REMOTE_HOST=" + cli_encode(info->host) +
|
||||
" REMOTE_PORT=" + cli_encode(to_string(socket->getPeerPort())) +
|
||||
" REQUEST_METHOD=" + cli_encode(req.getMethod()) +
|
||||
" REQUEST_URI=" + cli_encode(req.getPath()) +
|
||||
" SCRIPT_FILENAME=" + cli_encode(path.getFilePath()) +
|
||||
" SCRIPT_NAME=" + cli_encode(path.getRelativePath()) +
|
||||
" SERVER_ADMIN=" + cli_encode("lorenz.stechauner@gmail.com") +
|
||||
" SERVER_NAME=" + cli_encode(host) +
|
||||
" SERVER_PORT=" + cli_encode(to_string(socket->getSocketPort())) +
|
||||
" SERVER_SOFTWARE=" + cli_encode("Necronda 3.0") +
|
||||
" SERVER_PROTOCOL=" + cli_encode("HTTP/1.1") +
|
||||
" GATEWAY_INTERFACE=" + cli_encode("CGI/1.1") +
|
||||
" /usr/bin/php-cgi";
|
||||
|
||||
pipes = procopen(cmd.c_str());
|
||||
childpid = pipes.pid;
|
||||
|
||||
long len = req.isExistingField("Content-Length")
|
||||
? strtol(req.getField("Content-Length").c_str(), nullptr, 10)
|
||||
: ((req.getMethod() == "POST" || req.getMethod() == "PUT") ? -1 : 0);
|
||||
|
||||
socket->receive(pipes.stdin, len);
|
||||
wantsWebsocket = req.getMethod() == "GET" &&
|
||||
req.isExistingResponseField("Connection") && req.getField("Connection") == "Upgrade" &&
|
||||
req.isExistingResponseField("Upgrade") && req.getField("Upgrade") == "websocket";
|
||||
if (!wantsWebsocket) {
|
||||
// Close only if no Websocket upgrade is possible
|
||||
fclose(pipes.stdin);
|
||||
}
|
||||
t = new thread(php_error_handler, prefix, pipes.stderr);
|
||||
|
||||
string line;
|
||||
while (!(line = read_line(pipes.stdout)).empty()) {
|
||||
pos = line.find(':');
|
||||
string index = line.substr(0, pos);
|
||||
string data = line.substr(pos + 1, line.length() - pos);
|
||||
|
||||
while (index[0] == ' ') index.erase(index.begin() + 0);
|
||||
while (index[index.length() - 1] == ' ') index.erase(index.end() - 1);
|
||||
while (data[0] == ' ') data.erase(data.begin() + 0);
|
||||
while (data[data.length() - 1] == ' ') data.erase(data.end() - 1);
|
||||
|
||||
if (index == "Status") {
|
||||
statuscode = (int) strtol(data.substr(0, 3).c_str(), nullptr, 10);
|
||||
} else {
|
||||
if (index == "Location" && statuscode == 0) {
|
||||
statuscode = 303;
|
||||
} else if (index == "Content-Type") {
|
||||
type = data;
|
||||
}
|
||||
req.setField(index, data);
|
||||
}
|
||||
}
|
||||
|
||||
websocket = statuscode == 101 &&
|
||||
req.isExistingResponseField("Connection") && req.getResponseField("Connection") == "Upgrade" &&
|
||||
req.isExistingResponseField("Upgrade") && req.getResponseField("Upgrade") == "websocket";
|
||||
|
||||
fclose(file);
|
||||
file = pipes.stdout;
|
||||
if (websocket) {
|
||||
log(prefix, "Upgrade to WebSocket!");
|
||||
req.respond(statuscode);
|
||||
goto respond;
|
||||
} else {
|
||||
int c = fgetc(pipes.stdout);
|
||||
if (c == -1) {
|
||||
// No Data -> Error
|
||||
req.respond((statuscode == 0) ? 500 : statuscode);
|
||||
goto respond;
|
||||
} else {
|
||||
ungetc(c, pipes.stdout);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
statuscode = (statuscode == 0) ? 200 : statuscode;
|
||||
|
||||
compress = (type.find("text/") == 0 ||
|
||||
(type.find("application/") == 0 && type.find("+xml") != string::npos) ||
|
||||
type == "application/json" ||
|
||||
type == "application/javascript") &&
|
||||
req.isExistingField("Accept-Encoding") &&
|
||||
req.getField("Accept-Encoding").find("deflate") != string::npos;
|
||||
|
||||
if (compress) {
|
||||
req.setField("Accept-Ranges", "none");
|
||||
}
|
||||
|
||||
if (compress && req.isExistingField("Range")) {
|
||||
req.respond(416);
|
||||
} else if (req.isExistingField("Range")) {
|
||||
string range = req.getField("Range");
|
||||
if (range.find("bytes=") != 0 || !path.isStatic()) {
|
||||
req.respond(416);
|
||||
} else {
|
||||
fseek(file, 0L, SEEK_END);
|
||||
long len = ftell(file);
|
||||
fseek(file, 0L, SEEK_SET);
|
||||
long p = range.find('-');
|
||||
if (p == string::npos) {
|
||||
req.respond(416);
|
||||
} else {
|
||||
string part1 = range.substr(6, (unsigned long) (p - 6));
|
||||
string part2 = range.substr((unsigned long) (p + 1),
|
||||
range.length() - p - 1);
|
||||
long num1 = stol(part1, nullptr, 10);
|
||||
long num2 = len - 1;
|
||||
if (!part2.empty()) {
|
||||
num2 = stol(part2, nullptr, 10);
|
||||
}
|
||||
if (num1 < 0 || num1 >= len || num2 < 0 || num2 >= len) {
|
||||
req.respond(416);
|
||||
} else {
|
||||
req.setField("Content-Range",
|
||||
(string) "bytes " + to_string(num1) + "-" +
|
||||
to_string(num2) +
|
||||
"/" + to_string(len));
|
||||
req.respond(206, file, compress, num1, num2);
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
req.respond(statuscode, file, compress);
|
||||
}
|
||||
|
||||
fclose(file);
|
||||
if (childpid > 0) {
|
||||
waitpid(childpid, nullptr, 0);
|
||||
}
|
||||
|
||||
respond:
|
||||
|
||||
HttpStatusCode status = req.getStatusCode();
|
||||
int code = status.code;
|
||||
string color;
|
||||
string comment;
|
||||
if ((code >= 200 && code < 300) || code == 304 || code == 101) {
|
||||
color = "\x1B[1;32m"; // Success (Cached, Switching Protocols): Green
|
||||
} else if (code >= 100 && code < 200) {
|
||||
color = "\x1B[1;93m"; // Continue: Yellow
|
||||
} else if (code >= 300 && code < 400) {
|
||||
color = "\x1B[1;93m"; // Redirect: Yellow
|
||||
comment = " -> " +
|
||||
(req.isExistingResponseField("Location") ? req.getResponseField("Location") : "<invalid>");
|
||||
} else if (code >= 400 && code < 500) {
|
||||
color = "\x1B[1;31m"; // Client Error: Red
|
||||
//comment = " -> " + req.getPath();
|
||||
} else if (code >= 500 & code < 600) {
|
||||
color = "\x1B[1;31m"; // Server Error: Red
|
||||
//comment = " -> " + req.getPath();
|
||||
}
|
||||
string msg = color + to_string(status.code) + " " + status.message + comment + " (" + formatTime(req.getDuration()) + ")\x1B[0m";
|
||||
log(prefix, msg);
|
||||
if (!host.empty()) {
|
||||
log_to_file(prefix, msg, host);
|
||||
}
|
||||
|
||||
if (websocket) {
|
||||
websocket_handler(socket, &pipes);
|
||||
log(prefix, "\x1B[1mClosing WebSocket (" + formatTime(req.getDuration()) + ")\x1B[0m");
|
||||
}
|
||||
} catch (char *msg) {
|
||||
HttpStatusCode status = req.getStatusCode();
|
||||
log(prefix, to_string(status.code) + " " + status.message + " (" + formatTime(req.getDuration()) + ")");
|
||||
try {
|
||||
if (strncmp(msg, "timeout", strlen(msg)) == 0) {
|
||||
log(prefix, "Timeout!");
|
||||
req.setField("Connection", "close");
|
||||
req.respond(408);
|
||||
error = true;
|
||||
} else if (strncmp(msg, "Invalid path", strlen(msg)) == 0) {
|
||||
log(prefix, "Timeout!");
|
||||
req.setField("Connection", "close");
|
||||
req.respond(400);
|
||||
} else {
|
||||
log(prefix, (string) "Unable to receive from socket: " + msg);
|
||||
error = true;
|
||||
}
|
||||
} catch (char *msg2) {
|
||||
|
||||
}
|
||||
}
|
||||
return !error;
|
||||
}
|
||||
|
||||
/**
|
||||
* Handles HTTP clients
|
||||
* @param socket The socket
|
||||
* @param id The client ID
|
||||
*/
|
||||
void client_handler(Socket *socket, long id, bool ssl) {
|
||||
const char *prefix;
|
||||
char const *col1;
|
||||
char const *col2 = "\x1B[0m";
|
||||
IpAddressInfo info = get_ip_address_info(socket->getPeerAddress());
|
||||
auto os = get_os_info(socket->getFd());
|
||||
{
|
||||
auto group = (int) (id % 6);
|
||||
if (group == 0) {
|
||||
col1 = "\x1B[0;31m"; // Red
|
||||
} else if (group == 1) {
|
||||
col1 = "\x1B[0;32m"; // Green
|
||||
} else if (group == 2) {
|
||||
col1 = "\x1B[0;34m"; // Blue
|
||||
} else if (group == 3) {
|
||||
col1 = "\x1B[0;33m"; // Yellow
|
||||
} else if (group == 4) {
|
||||
col1 = "\x1B[0;35m"; // Magenta
|
||||
} else {
|
||||
col1 = "\x1B[0;36m"; // Cyan
|
||||
}
|
||||
|
||||
string *a = new string("[" + socket->getSocketAddress()->toString() + "][" +
|
||||
to_string(socket->getSocketPort()) + "]" + col1 +
|
||||
"[" + info.host + "][" + to_string(socket->getPeerPort()) +
|
||||
"]" + col2 + " ");
|
||||
prefix = a->c_str();
|
||||
}
|
||||
|
||||
log(prefix, "Connection established");
|
||||
log(prefix, string("Host: ") + info.host + " (" + socket->getPeerAddress()->toString() + ")");
|
||||
log(prefix, string("OS: ") + os);
|
||||
log(prefix, string("Location: ") + info.cc + "/" + info.country + ", " + info.prov + "/" + info.provname + ", " + info.city);
|
||||
log(prefix, string("Local Date: ") + info.localdate + " (" + info.timezone + ")");
|
||||
|
||||
|
||||
bool err = false;
|
||||
try {
|
||||
socket->setReceiveTimeout(3600000);
|
||||
socket->setSendTimeout(36000000);
|
||||
} catch (char *msg) {
|
||||
log(prefix, (string) "Unable to set timeout on socket: " + msg);
|
||||
err = true;
|
||||
}
|
||||
|
||||
try {
|
||||
if (ssl) {
|
||||
//socket->sslHandshake("/home/lorenz/Documents/Projects/Necronda-Server/necronda-server-3.0/privkey.pem",
|
||||
// "/home/lorenz/Documents/Projects/Necronda-Server/necronda-server-3.0/fullchain.pem");
|
||||
socket->sslHandshake("/cert/necronda.net/privkey.pem",
|
||||
"/cert/necronda.net/fullchain.pem");
|
||||
}
|
||||
} catch (char *msg) {
|
||||
log(prefix, (string) "Unable to perform handshake: " + msg);
|
||||
err = true;
|
||||
}
|
||||
|
||||
long reqnum = 0;
|
||||
if (!err) {
|
||||
while (connection_handler(prefix, col1, col2, socket, id, ++reqnum, &info));
|
||||
reqnum--;
|
||||
}
|
||||
|
||||
log(prefix,
|
||||
"Connection terminated (#:" + to_string(reqnum) + ", R: " + formatSize(socket->getBytesReceived()) + ", S: " +
|
||||
formatSize(socket->getBytesSent()) + ", T: " + formatTime(socket->getDuration()) + ")");
|
||||
socket->close();
|
||||
}
|
||||
|
||||
|
27
src/defs.h
Normal file
27
src/defs.h
Normal file
@ -0,0 +1,27 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Definitions
|
||||
* @file src/defs.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-05-04
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_DEF_H
|
||||
#define SESIMOS_DEF_H
|
||||
|
||||
#define SERVER_VERSION "5.0-wip"
|
||||
#define SERVER_STR "sesimos/" SERVER_VERSION
|
||||
#define SERVER_STR_HTML "sesimos web server " SERVER_VERSION
|
||||
|
||||
#define CHUNK_SIZE 8192
|
||||
#define MAX_PROXY_CNX_PER_HOST 16
|
||||
|
||||
#ifndef DEFAULT_HOST
|
||||
# define DEFAULT_HOST "www.necronda.net"
|
||||
#endif
|
||||
|
||||
#ifndef SERVER_NAME
|
||||
# define SERVER_NAME DEFAULT_HOST
|
||||
#endif
|
||||
|
||||
#endif //SESIMOS_DEF_H
|
74
src/lib/compress.c
Normal file
74
src/lib/compress.c
Normal file
@ -0,0 +1,74 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Compression interface
|
||||
* @file src/lib/compress.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-05-05
|
||||
*/
|
||||
|
||||
#include "compress.h"
|
||||
|
||||
#include <errno.h>
|
||||
|
||||
|
||||
int compress_init(compress_ctx *ctx, int mode) {
|
||||
ctx->brotli = NULL;
|
||||
ctx->mode = 0;
|
||||
int ret;
|
||||
if (mode & COMPRESS_GZ) {
|
||||
ctx->mode |= COMPRESS_GZ;
|
||||
ctx->gzip.zalloc = Z_NULL;
|
||||
ctx->gzip.zfree = Z_NULL;
|
||||
ctx->gzip.opaque = Z_NULL;
|
||||
ret = deflateInit2(&ctx->gzip, COMPRESS_LEVEL_GZIP, Z_DEFLATED, 15 + 16, 9, Z_DEFAULT_STRATEGY);
|
||||
if (ret != Z_OK) return -1;
|
||||
}
|
||||
if (mode & COMPRESS_BR) {
|
||||
ctx->mode |= COMPRESS_BR;
|
||||
ctx->brotli = BrotliEncoderCreateInstance(NULL, NULL, NULL);
|
||||
if (ctx->brotli == NULL) return -1;
|
||||
BrotliEncoderSetParameter(ctx->brotli, BROTLI_PARAM_MODE, BROTLI_MODE_GENERIC);
|
||||
BrotliEncoderSetParameter(ctx->brotli, BROTLI_PARAM_QUALITY, COMPRESS_LEVEL_BROTLI);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
int compress_compress(compress_ctx *ctx, const char *in, unsigned long *in_len, char *out, unsigned long *out_len, int finish) {
|
||||
if ((ctx->mode & COMPRESS_GZ) && (ctx->mode & COMPRESS_BR)) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
}
|
||||
return compress_compress_mode(ctx, ctx->mode, in, in_len, out, out_len, finish);
|
||||
}
|
||||
|
||||
int compress_compress_mode(compress_ctx *ctx, int mode, const char *in, unsigned long *in_len, char *out, unsigned long *out_len, int finish) {
|
||||
if ((mode & COMPRESS_GZ) && (mode & COMPRESS_BR)) {
|
||||
errno = EINVAL;
|
||||
return -1;
|
||||
} else if (mode & COMPRESS_GZ) {
|
||||
ctx->gzip.next_in = (unsigned char*) in;
|
||||
ctx->gzip.avail_in = *in_len;
|
||||
ctx->gzip.next_out = (unsigned char*) out;
|
||||
ctx->gzip.avail_out = *out_len;
|
||||
int ret = deflate(&ctx->gzip, finish ? Z_FINISH : Z_NO_FLUSH);
|
||||
*in_len = ctx->gzip.avail_in;
|
||||
*out_len = ctx->gzip.avail_out;
|
||||
return ret;
|
||||
} else if (mode & COMPRESS_BR) {
|
||||
int ret = BrotliEncoderCompressStream(ctx->brotli, finish ? BROTLI_OPERATION_FINISH : BROTLI_OPERATION_PROCESS,
|
||||
in_len, (const unsigned char**) &in, out_len, (unsigned char **) &out, NULL);
|
||||
return (ret == BROTLI_TRUE) ? 0 : -1;
|
||||
} else {
|
||||
errno = EINVAL;
|
||||
return -2;
|
||||
}
|
||||
}
|
||||
|
||||
int compress_free(compress_ctx *ctx) {
|
||||
if (ctx->brotli != NULL) {
|
||||
BrotliEncoderDestroyInstance(ctx->brotli);
|
||||
ctx->brotli = NULL;
|
||||
}
|
||||
ctx->mode = 0;
|
||||
return 0;
|
||||
}
|
36
src/lib/compress.h
Normal file
36
src/lib/compress.h
Normal file
@ -0,0 +1,36 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Compression interface (header file)
|
||||
* @file src/lib/compress.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-05-05
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_COMPRESS_H
|
||||
#define SESIMOS_COMPRESS_H
|
||||
|
||||
#include <zlib.h>
|
||||
#include <brotli/encode.h>
|
||||
|
||||
#define COMPRESS_LEVEL_GZIP 9
|
||||
#define COMPRESS_LEVEL_BROTLI BROTLI_MAX_QUALITY
|
||||
|
||||
#define COMPRESS_GZ 1
|
||||
#define COMPRESS_BR 2
|
||||
#define COMPRESS 3
|
||||
|
||||
typedef struct {
|
||||
int mode;
|
||||
z_stream gzip;
|
||||
BrotliEncoderState *brotli;
|
||||
} compress_ctx;
|
||||
|
||||
int compress_init(compress_ctx *ctx, int mode);
|
||||
|
||||
int compress_compress(compress_ctx *ctx, const char *in, unsigned long *in_len, char *out, unsigned long *out_len, int finish);
|
||||
|
||||
int compress_compress_mode(compress_ctx *ctx, int mode, const char *in, unsigned long *in_len, char *out, unsigned long *out_len, int finish);
|
||||
|
||||
int compress_free(compress_ctx *ctx);
|
||||
|
||||
#endif //SESIMOS_COMPRESS_H
|
223
src/lib/config.c
Normal file
223
src/lib/config.c
Normal file
@ -0,0 +1,223 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Configuration file loader
|
||||
* @file src/lib/config.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-01-05
|
||||
*/
|
||||
|
||||
#include "../logger.h"
|
||||
#include "config.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
config_t config;
|
||||
|
||||
int config_load(const char *filename) {
|
||||
FILE *file = fopen(filename, "r");
|
||||
if (file == NULL) {
|
||||
critical("Unable to open config file");
|
||||
return -1;
|
||||
}
|
||||
|
||||
memset(&config, 0, sizeof(config));
|
||||
|
||||
int i = 0;
|
||||
int j = 0;
|
||||
int line_num = 0;
|
||||
int mode = 0;
|
||||
char section = 0;
|
||||
char *source, *target;
|
||||
|
||||
char *line = NULL;
|
||||
ssize_t read;
|
||||
size_t line_len = 0;
|
||||
while ((read = getline(&line, &line_len, file)) != -1) {
|
||||
line_num++;
|
||||
char *ptr = line;
|
||||
char *comment = strpbrk(ptr, "#\r\n");
|
||||
if (comment != NULL) comment[0] = 0;
|
||||
|
||||
unsigned long len = strlen(ptr);
|
||||
char *end_ptr = ptr + len - 1;
|
||||
while (end_ptr[0] == ' ' || end_ptr[0] == '\t') {
|
||||
end_ptr[0] = 0;
|
||||
end_ptr--;
|
||||
}
|
||||
len = strlen(ptr);
|
||||
if (len == 0) continue;
|
||||
|
||||
if (ptr[0] == '[') {
|
||||
if (ptr[len - 1] != ']') goto err;
|
||||
ptr++;
|
||||
int l = 0;
|
||||
if (strncmp(ptr, "host", 4) == 0 && (ptr[4] == ' ' || ptr[4] == '\t')) {
|
||||
ptr += 4;
|
||||
while (ptr[0] == ' ' || ptr[0] == '\t' || ptr[0] == ']') ptr++;
|
||||
while (ptr[l] != ' ' && ptr[l] != '\t' && ptr[l] != ']') l++;
|
||||
if (l == 0) goto err;
|
||||
snprintf(config.hosts[i].name, sizeof(config.hosts[i].name), "%.*s", l, ptr);
|
||||
i++;
|
||||
section = 'h';
|
||||
} else if (strncmp(ptr, "cert", 4) == 0 && (ptr[4] == ' ' || ptr[4] == '\t')) {
|
||||
ptr += 4;
|
||||
while (ptr[0] == ' ' || ptr[0] == '\t' || ptr[0] == ']') ptr++;
|
||||
while (ptr[l] != ' ' && ptr[l] != '\t' && ptr[l] != ']') l++;
|
||||
if (l == 0) goto err;
|
||||
snprintf(config.certs[j].name, sizeof(config.certs[j].name), "%.*s", l, ptr);
|
||||
j++;
|
||||
section = 'c';
|
||||
} else {
|
||||
goto err;
|
||||
}
|
||||
continue;
|
||||
} else if (section == 0) {
|
||||
if (len > 10 && strncmp(ptr, "geoip_dir", 9) == 0 && (ptr[9] == ' ' || ptr[9] == '\t')) {
|
||||
source = ptr + 9;
|
||||
target = config.geoip_dir;
|
||||
} else if (len > 11 && strncmp(ptr, "dns_server", 10) == 0 && (ptr[10] == ' ' || ptr[10] == '\t')) {
|
||||
source = ptr + 10;
|
||||
target = config.dns_server;
|
||||
} else {
|
||||
goto err;
|
||||
}
|
||||
} else if (section == 'c') {
|
||||
cert_config_t *cc = &config.certs[j - 1];
|
||||
if (len > 12 && strncmp(ptr, "certificate", 11) == 0 && (ptr[11] == ' ' || ptr[11] == '\t')) {
|
||||
source = ptr + 11;
|
||||
target = cc->full_chain;
|
||||
} else if (len > 12 && strncmp(ptr, "private_key", 11) == 0 && (ptr[11] == ' ' || ptr[11] == '\t')) {
|
||||
source = ptr + 11;
|
||||
target = cc->priv_key;
|
||||
} else {
|
||||
goto err;
|
||||
}
|
||||
} else if (section == 'h') {
|
||||
host_config_t *hc = &config.hosts[i - 1];
|
||||
if (len > 8 && strncmp(ptr, "webroot", 7) == 0 && (ptr[7] == ' ' || ptr[7] == '\t')) {
|
||||
source = ptr + 7;
|
||||
target = hc->local.webroot;
|
||||
if (hc->type != 0 && hc->type != CONFIG_TYPE_LOCAL) {
|
||||
goto err;
|
||||
} else {
|
||||
hc->type = CONFIG_TYPE_LOCAL;
|
||||
}
|
||||
} else if (len > 5 && strncmp(ptr, "cert", 4) == 0 && (ptr[4] == ' ' || ptr[4] == '\t')) {
|
||||
source = ptr + 4;
|
||||
target = hc->cert_name;
|
||||
} else if (len > 9 && strncmp(ptr, "dir_mode", 8) == 0 && (ptr[8] == ' ' || ptr[8] == '\t')) {
|
||||
source = ptr + 8;
|
||||
target = NULL;
|
||||
mode = 1;
|
||||
if (hc->type != 0 && hc->type != CONFIG_TYPE_LOCAL) {
|
||||
goto err;
|
||||
} else {
|
||||
hc->type = CONFIG_TYPE_LOCAL;
|
||||
}
|
||||
} else if (len > 9 && strncmp(ptr, "hostname", 8) == 0 && (ptr[8] == ' ' || ptr[8] == '\t')) {
|
||||
source = ptr + 8;
|
||||
target = hc->proxy.hostname;
|
||||
if (hc->type != 0 && hc->type != CONFIG_TYPE_REVERSE_PROXY) {
|
||||
goto err;
|
||||
} else {
|
||||
hc->type = CONFIG_TYPE_REVERSE_PROXY;
|
||||
}
|
||||
} else if (len > 5 && strncmp(ptr, "port", 4) == 0 && (ptr[4] == ' ' || ptr[4] == '\t')) {
|
||||
source = ptr + 4;
|
||||
target = NULL;
|
||||
mode = 2;
|
||||
if (hc->type != 0 && hc->type != CONFIG_TYPE_REVERSE_PROXY) {
|
||||
goto err;
|
||||
} else {
|
||||
hc->type = CONFIG_TYPE_REVERSE_PROXY;
|
||||
}
|
||||
} else if (strcmp(ptr, "http") == 0) {
|
||||
if (hc->type != 0 && hc->type != CONFIG_TYPE_REVERSE_PROXY) {
|
||||
goto err;
|
||||
} else {
|
||||
hc->type = CONFIG_TYPE_REVERSE_PROXY;
|
||||
hc->proxy.enc = 0;
|
||||
}
|
||||
continue;
|
||||
} else if (strcmp(ptr, "https") == 0) {
|
||||
if (hc->type != 0 && hc->type != CONFIG_TYPE_REVERSE_PROXY) {
|
||||
goto err;
|
||||
} else {
|
||||
hc->type = CONFIG_TYPE_REVERSE_PROXY;
|
||||
hc->proxy.enc = 1;
|
||||
}
|
||||
continue;
|
||||
} else {
|
||||
goto err;
|
||||
}
|
||||
} else {
|
||||
goto err;
|
||||
}
|
||||
|
||||
while (source[0] == ' ' || source[0] == '\t') source++;
|
||||
if (strlen(source) == 0) {
|
||||
err:
|
||||
critical("Unable to parse config file (line %i)", line);
|
||||
return -2;
|
||||
}
|
||||
|
||||
if (target != NULL) {
|
||||
strcpy(target, source);
|
||||
} else if (mode == 1) {
|
||||
if (strcmp(source, "forbidden") == 0) {
|
||||
config.hosts[i - 1].local.dir_mode = URI_DIR_MODE_FORBIDDEN;
|
||||
} else if (strcmp(source, "info") == 0) {
|
||||
config.hosts[i - 1].local.dir_mode = URI_DIR_MODE_INFO;
|
||||
} else if (strcmp(source, "list") == 0) {
|
||||
config.hosts[i - 1].local.dir_mode = URI_DIR_MODE_LIST;
|
||||
} else {
|
||||
goto err;
|
||||
}
|
||||
} else if (mode == 2) {
|
||||
config.hosts[i - 1].proxy.port = (unsigned short) strtoul(source, NULL, 10);
|
||||
}
|
||||
}
|
||||
|
||||
free(line);
|
||||
|
||||
for (int k = 0; k < i; k++) {
|
||||
host_config_t *hc = &config.hosts[k];
|
||||
if (hc->type == CONFIG_TYPE_LOCAL) {
|
||||
char *webroot = config.hosts[k].local.webroot;
|
||||
if (webroot[strlen(webroot) - 1] == '/') {
|
||||
webroot[strlen(webroot) - 1] = 0;
|
||||
}
|
||||
}
|
||||
if (hc->cert_name[0] == 0) goto err2;
|
||||
int found = 0;
|
||||
for (int m = 0; m < j; m++) {
|
||||
if (strcmp(config.certs[m].name, hc->cert_name) == 0) {
|
||||
hc->cert = m;
|
||||
found = 1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!found) {
|
||||
err2:
|
||||
critical("Unable to parse config file");
|
||||
return -2;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
host_config_t *get_host_config(const char *host) {
|
||||
for (int i = 0; i < CONFIG_MAX_HOST_CONFIG; i++) {
|
||||
host_config_t *hc = &config.hosts[i];
|
||||
if (hc->type == CONFIG_TYPE_UNSET) break;
|
||||
if (strcmp(hc->name, host) == 0) return hc;
|
||||
if (hc->name[0] == '*' && hc->name[1] == '.') {
|
||||
const char *pos = strstr(host, hc->name + 1);
|
||||
if (pos != NULL && strlen(pos) == strlen(hc->name + 1)) return hc;
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
65
src/lib/config.h
Normal file
65
src/lib/config.h
Normal file
@ -0,0 +1,65 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Configuration file loader (header file)
|
||||
* @file src/lib/config.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-01-05
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_CONFIG_H
|
||||
#define SESIMOS_CONFIG_H
|
||||
|
||||
#include "uri.h"
|
||||
#include "../cache_handler.h"
|
||||
|
||||
#define CONFIG_MAX_HOST_CONFIG 64
|
||||
#define CONFIG_MAX_CERT_CONFIG 64
|
||||
|
||||
#define CONFIG_TYPE_UNSET 0
|
||||
#define CONFIG_TYPE_LOCAL 1
|
||||
#define CONFIG_TYPE_REVERSE_PROXY 2
|
||||
|
||||
#ifndef DEFAULT_CONFIG_FILE
|
||||
# define DEFAULT_CONFIG_FILE "/etc/sesimos/server.conf"
|
||||
#endif
|
||||
|
||||
|
||||
typedef struct {
|
||||
int type;
|
||||
char name[256];
|
||||
char cert_name[256];
|
||||
int cert;
|
||||
cache_t *cache;
|
||||
union {
|
||||
struct {
|
||||
char hostname[256];
|
||||
unsigned short port;
|
||||
unsigned char enc:1;
|
||||
} proxy;
|
||||
struct {
|
||||
char webroot[256];
|
||||
unsigned char dir_mode:2;
|
||||
} local;
|
||||
};
|
||||
} host_config_t;
|
||||
|
||||
typedef struct {
|
||||
char name[256];
|
||||
char full_chain[256];
|
||||
char priv_key[256];
|
||||
} cert_config_t;
|
||||
|
||||
typedef struct {
|
||||
host_config_t hosts[CONFIG_MAX_HOST_CONFIG];
|
||||
cert_config_t certs[CONFIG_MAX_CERT_CONFIG];
|
||||
char geoip_dir[256];
|
||||
char dns_server[256];
|
||||
} config_t;
|
||||
|
||||
extern config_t config;
|
||||
|
||||
int config_load(const char *filename);
|
||||
|
||||
host_config_t *get_host_config(const char *host);
|
||||
|
||||
#endif //SESIMOS_CONFIG_H
|
134
src/lib/error.c
Normal file
134
src/lib/error.c
Normal file
@ -0,0 +1,134 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Error interface
|
||||
* @file src/lib/error.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2023-01-08
|
||||
*/
|
||||
|
||||
#include "error.h"
|
||||
#include "http.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <openssl/ssl.h>
|
||||
#include <openssl/err.h>
|
||||
#include <maxminddb.h>
|
||||
|
||||
static const char *error_ssl_strerror(int err) {
|
||||
switch (err) {
|
||||
case SSL_ERROR_ZERO_RETURN:
|
||||
return "closed";
|
||||
case SSL_ERROR_WANT_READ:
|
||||
return "want read";
|
||||
case SSL_ERROR_WANT_WRITE:
|
||||
return "want write";
|
||||
case SSL_ERROR_WANT_CONNECT:
|
||||
return "want connect";
|
||||
case SSL_ERROR_WANT_ACCEPT:
|
||||
return "want accept";
|
||||
case SSL_ERROR_WANT_X509_LOOKUP:
|
||||
return "want x509 lookup";
|
||||
case SSL_ERROR_WANT_ASYNC:
|
||||
return "want async";
|
||||
case SSL_ERROR_WANT_ASYNC_JOB:
|
||||
return "want async job";
|
||||
case SSL_ERROR_WANT_CLIENT_HELLO_CB:
|
||||
return "want client hello callback";
|
||||
case SSL_ERROR_WANT_RETRY_VERIFY:
|
||||
return "want retry verify";
|
||||
case SSL_ERROR_SSL:
|
||||
return ERR_reason_error_string(ERR_get_error());
|
||||
default:
|
||||
return "unknown error";
|
||||
}
|
||||
}
|
||||
|
||||
static const char *error_http_strerror(int err) {
|
||||
switch (err) {
|
||||
case HTTP_ERROR_TOO_MANY_HEADER_FIELDS:
|
||||
return "too many header fields";
|
||||
case HTTP_ERROR_EOH_NOT_FOUND:
|
||||
return "end of http header not found";
|
||||
case HTTP_ERROR_HEADER_MALFORMED:
|
||||
return "http header malformed";
|
||||
case HTTP_ERROR_INVALID_VERSION:
|
||||
return "invalid http version";
|
||||
case HTTP_ERROR_URI_TOO_LONG:
|
||||
return "uri too long";
|
||||
case HTTP_ERROR_GENERAL:
|
||||
default:
|
||||
return "unknown error";
|
||||
}
|
||||
}
|
||||
|
||||
const char *error_str(int err_no, char *buf, int buf_len) {
|
||||
buf[0] = 0;
|
||||
unsigned char mode = (unsigned char) (err_no >> 24);
|
||||
int e = err_no & 0x00FFFFFF;
|
||||
if (mode == 0x00) {
|
||||
// normal
|
||||
strerror_r(e, buf, buf_len);
|
||||
return buf;
|
||||
} else if (mode == 0x01) {
|
||||
// ssl
|
||||
return error_ssl_strerror(e);
|
||||
} else if (mode == 0x02) {
|
||||
// mmdb
|
||||
return MMDB_strerror(e);
|
||||
} else if (mode == 0x03) {
|
||||
// http
|
||||
return error_http_strerror(e);
|
||||
}
|
||||
return buf;
|
||||
}
|
||||
|
||||
void error_ssl(int err) {
|
||||
if (err == SSL_ERROR_NONE) {
|
||||
errno = 0;
|
||||
} else if (err == SSL_ERROR_SYSCALL) {
|
||||
// errno already set
|
||||
} else {
|
||||
errno = 0x01000000 | err;
|
||||
}
|
||||
}
|
||||
|
||||
void error_mmdb(int err) {
|
||||
if (err == MMDB_SUCCESS) {
|
||||
errno = 0;
|
||||
} else if (err == MMDB_IO_ERROR) {
|
||||
// errno already set
|
||||
} else {
|
||||
errno = 0x02000000 | err;
|
||||
}
|
||||
}
|
||||
|
||||
int error_http(int err) {
|
||||
if (err == 0) {
|
||||
errno = 0;
|
||||
} else if (err == HTTP_ERROR_SYSCALL) {
|
||||
// errno already set
|
||||
} else {
|
||||
errno = 0x03000000 | err;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
static int error_get(unsigned char prefix) {
|
||||
return (errno >> 24 != prefix) ? 0 : errno & 0x00FFFFFF;
|
||||
}
|
||||
|
||||
int error_get_sys() {
|
||||
return error_get(0x00);
|
||||
}
|
||||
|
||||
int error_get_ssl() {
|
||||
return error_get(0x01);
|
||||
}
|
||||
|
||||
int error_get_mmdb() {
|
||||
return error_get(0x02);
|
||||
}
|
||||
|
||||
int error_get_http() {
|
||||
return error_get(0x03);
|
||||
}
|
28
src/lib/error.h
Normal file
28
src/lib/error.h
Normal file
@ -0,0 +1,28 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Error interface (header fie)
|
||||
* @file src/lib/error.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2023-01-08
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_ERROR_H
|
||||
#define SESIMOS_ERROR_H
|
||||
|
||||
const char *error_str(int err_no, char *buf, int buf_len);
|
||||
|
||||
void error_ssl(int err);
|
||||
|
||||
void error_mmdb(int err);
|
||||
|
||||
int error_http(int err);
|
||||
|
||||
int error_get_sys();
|
||||
|
||||
int error_get_ssl();
|
||||
|
||||
int error_get_mmdb();
|
||||
|
||||
int error_get_http();
|
||||
|
||||
#endif //SESIMOS_ERROR_H
|
630
src/lib/fastcgi.c
Normal file
630
src/lib/fastcgi.c
Normal file
@ -0,0 +1,630 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief FastCGI interface implementation
|
||||
* @file src/lib/fastcgi.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-26
|
||||
*/
|
||||
|
||||
#include "fastcgi.h"
|
||||
#include "utils.h"
|
||||
#include "compress.h"
|
||||
#include "../logger.h"
|
||||
#include "list.h"
|
||||
|
||||
#include <sys/un.h>
|
||||
#include <sys/socket.h>
|
||||
#include <string.h>
|
||||
|
||||
|
||||
char *fastcgi_add_param(char *buf, const char *key, const char *value) {
|
||||
char *ptr = buf;
|
||||
unsigned long key_len = strlen(key);
|
||||
unsigned long val_len = strlen(value);
|
||||
|
||||
|
||||
if (key_len <= 127) {
|
||||
ptr[0] = (char) (key_len & 0x7F);
|
||||
ptr++;
|
||||
} else {
|
||||
ptr[0] = (char) (0x80 | (key_len >> 24));
|
||||
ptr[1] = (char) ((key_len >> 16) & 0xFF);
|
||||
ptr[2] = (char) ((key_len >> 8) & 0xFF);
|
||||
ptr[3] = (char) (key_len & 0xFF);
|
||||
ptr += 4;
|
||||
}
|
||||
if (val_len <= 127) {
|
||||
ptr[0] = (char) (val_len & 0x7F);
|
||||
ptr++;
|
||||
} else {
|
||||
ptr[0] = (char) (0x80 | (val_len >> 24));
|
||||
ptr[1] = (char) ((val_len >> 16) & 0xFF);
|
||||
ptr[2] = (char) ((val_len >> 8) & 0xFF);
|
||||
ptr[3] = (char) (val_len & 0xFF);
|
||||
ptr += 4;
|
||||
}
|
||||
|
||||
strcpy(ptr, key);
|
||||
ptr += key_len;
|
||||
strcpy(ptr, value);
|
||||
ptr += val_len;
|
||||
|
||||
return ptr;
|
||||
}
|
||||
|
||||
int fastcgi_init(fastcgi_cnx_t *conn, int mode, unsigned int req_num, const sock *client, const http_req *req, const http_uri *uri) {
|
||||
conn->mode = mode;
|
||||
conn->req_id = (req_num + 1) & 0xFFFF;
|
||||
conn->out_buf = NULL;
|
||||
conn->out_off = 0;
|
||||
conn->webroot = uri->webroot;
|
||||
|
||||
int fcgi_sock = socket(AF_UNIX, SOCK_STREAM, 0);
|
||||
if (fcgi_sock < 0) {
|
||||
error("Unable to create unix socket");
|
||||
return -1;
|
||||
}
|
||||
conn->socket = fcgi_sock;
|
||||
|
||||
struct sockaddr_un sock_addr = {AF_UNIX};
|
||||
if (conn->mode == FASTCGI_SESIMOS) {
|
||||
snprintf(sock_addr.sun_path, sizeof(sock_addr.sun_path) - 1, "%s", SESIMOS_BACKEND_SOCKET);
|
||||
} else if (conn->mode == FASTCGI_PHP) {
|
||||
snprintf(sock_addr.sun_path, sizeof(sock_addr.sun_path) - 1, "%s", PHP_FPM_SOCKET);
|
||||
}
|
||||
|
||||
if (connect(conn->socket, (struct sockaddr *) &sock_addr, sizeof(sock_addr)) < 0) {
|
||||
error("Unable to connect to unix socket of FastCGI socket");
|
||||
return -1;
|
||||
}
|
||||
|
||||
FCGI_Header header = {
|
||||
.version = FCGI_VERSION_1,
|
||||
.requestIdB1 = conn->req_id >> 8,
|
||||
.requestIdB0 = conn->req_id & 0xFF,
|
||||
.paddingLength = 0,
|
||||
.reserved = 0
|
||||
};
|
||||
|
||||
header.type = FCGI_BEGIN_REQUEST;
|
||||
header.contentLengthB1 = 0;
|
||||
header.contentLengthB0 = sizeof(FCGI_BeginRequestBody);
|
||||
FCGI_BeginRequestRecord begin = {
|
||||
header,
|
||||
{.roleB1 = (FCGI_RESPONDER >> 8) & 0xFF, .roleB0 = FCGI_RESPONDER & 0xFF, .flags = 0}
|
||||
};
|
||||
if (send(conn->socket, &begin, sizeof(begin), 0) != sizeof(begin)) {
|
||||
error("Unable to send to FastCGI socket");
|
||||
return -2;
|
||||
}
|
||||
|
||||
char param_buf[4096];
|
||||
char buf0[256];
|
||||
char *param_ptr = param_buf + sizeof(header);
|
||||
|
||||
param_ptr = fastcgi_add_param(param_ptr, "REDIRECT_STATUS", "CGI");
|
||||
param_ptr = fastcgi_add_param(param_ptr, "DOCUMENT_ROOT", uri->webroot);
|
||||
param_ptr = fastcgi_add_param(param_ptr, "GATEWAY_INTERFACE", "CGI/1.1");
|
||||
param_ptr = fastcgi_add_param(param_ptr, "SERVER_SOFTWARE", SERVER_STR);
|
||||
param_ptr = fastcgi_add_param(param_ptr, "SERVER_PROTOCOL", "HTTP/1.1");
|
||||
param_ptr = fastcgi_add_param(param_ptr, "SERVER_NAME", http_get_header_field(&req->hdr, "Host"));
|
||||
if (client->enc) {
|
||||
param_ptr = fastcgi_add_param(param_ptr, "HTTPS", "on");
|
||||
}
|
||||
|
||||
struct sockaddr_storage addr_storage;
|
||||
struct sockaddr_in6 *addr;
|
||||
socklen_t len = sizeof(addr_storage);
|
||||
getsockname(client->socket, (struct sockaddr *) &addr_storage, &len);
|
||||
addr = (struct sockaddr_in6 *) &addr_storage;
|
||||
sprintf(buf0, "%i", addr->sin6_port);
|
||||
param_ptr = fastcgi_add_param(param_ptr, "SERVER_PORT", buf0);
|
||||
|
||||
len = sizeof(addr_storage);
|
||||
getpeername(client->socket, (struct sockaddr *) &addr_storage, &len);
|
||||
addr = (struct sockaddr_in6 *) &addr_storage;
|
||||
sprintf(buf0, "%i", addr->sin6_port);
|
||||
param_ptr = fastcgi_add_param(param_ptr, "REMOTE_PORT", buf0);
|
||||
param_ptr = fastcgi_add_param(param_ptr, "REMOTE_ADDR", conn->r_addr);
|
||||
param_ptr = fastcgi_add_param(param_ptr, "REMOTE_HOST", conn->r_host != NULL ? conn->r_host : conn->r_addr);
|
||||
//param_ptr = fastcgi_add_param(param_ptr, "REMOTE_IDENT", "");
|
||||
//param_ptr = fastcgi_add_param(param_ptr, "REMOTE_USER", "");
|
||||
|
||||
param_ptr = fastcgi_add_param(param_ptr, "REQUEST_METHOD", req->method);
|
||||
param_ptr = fastcgi_add_param(param_ptr, "REQUEST_URI", req->uri);
|
||||
param_ptr = fastcgi_add_param(param_ptr, "SCRIPT_NAME", uri->filename + strlen(uri->webroot));
|
||||
param_ptr = fastcgi_add_param(param_ptr, "SCRIPT_FILENAME", uri->filename);
|
||||
//param_ptr = fastcgi_add_param(param_ptr, "PATH_TRANSLATED", uri->filename);
|
||||
|
||||
param_ptr = fastcgi_add_param(param_ptr, "QUERY_STRING", uri->query != NULL ? uri->query : "");
|
||||
if (uri->pathinfo != NULL && strlen(uri->pathinfo) > 0) {
|
||||
sprintf(buf0, "/%s", uri->pathinfo);
|
||||
} else {
|
||||
buf0[0] = 0;
|
||||
}
|
||||
param_ptr = fastcgi_add_param(param_ptr, "PATH_INFO", buf0);
|
||||
|
||||
//param_ptr = fastcgi_add_param(param_ptr, "AUTH_TYPE", "");
|
||||
const char *content_length = http_get_header_field(&req->hdr, "Content-Length");
|
||||
param_ptr = fastcgi_add_param(param_ptr, "CONTENT_LENGTH", content_length != NULL ? content_length : "");
|
||||
const char *content_type = http_get_header_field(&req->hdr, "Content-Type");
|
||||
param_ptr = fastcgi_add_param(param_ptr, "CONTENT_TYPE", content_type != NULL ? content_type : "");
|
||||
//if (conn->ctx->geoip[0] != 0) {
|
||||
// param_ptr = fastcgi_add_param(param_ptr, "REMOTE_INFO", conn->ctx->geoip);
|
||||
//}
|
||||
|
||||
for (int i = 0; i < list_size(&req->hdr); i++) {
|
||||
const http_field *f = &req->hdr.fields[i];
|
||||
const char *name = http_field_get_name(f);
|
||||
char *ptr = buf0;
|
||||
ptr += sprintf(ptr, "HTTP_");
|
||||
for (int j = 0; j < strlen(name); j++, ptr++) {
|
||||
char ch = name[j];
|
||||
if ((ch >= 'A' && ch <= 'Z') || (ch >= '0' && ch <= '9')) {
|
||||
ch = ch;
|
||||
} else if (ch >= 'a' && ch <= 'z') {
|
||||
ch &= 0x5F;
|
||||
} else {
|
||||
ch = '_';
|
||||
}
|
||||
ptr[0] = ch;
|
||||
ptr[1] = 0;
|
||||
}
|
||||
param_ptr = fastcgi_add_param(param_ptr, buf0, http_field_get_value(f));
|
||||
}
|
||||
|
||||
unsigned short param_len = param_ptr - param_buf - sizeof(header);
|
||||
header.type = FCGI_PARAMS;
|
||||
header.contentLengthB1 = param_len >> 8;
|
||||
header.contentLengthB0 = param_len & 0xFF;
|
||||
memcpy(param_buf, &header, sizeof(header));
|
||||
if (send(conn->socket, param_buf, param_len + sizeof(header), 0) != param_len + sizeof(header)) {
|
||||
error("Unable to send to FastCGI socket");
|
||||
return -2;
|
||||
}
|
||||
|
||||
header.type = FCGI_PARAMS;
|
||||
header.contentLengthB1 = 0;
|
||||
header.contentLengthB0 = 0;
|
||||
if (send(conn->socket, &header, sizeof(header), 0) != sizeof(header)) {
|
||||
error("Unable to send to FastCGI socket");
|
||||
return -2;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int fastcgi_close_stdin(fastcgi_cnx_t *conn) {
|
||||
FCGI_Header header = {
|
||||
.version = FCGI_VERSION_1,
|
||||
.type = FCGI_STDIN,
|
||||
.requestIdB1 = conn->req_id >> 8,
|
||||
.requestIdB0 = conn->req_id & 0xFF,
|
||||
.contentLengthB1 = 0,
|
||||
.contentLengthB0 = 0,
|
||||
.paddingLength = 0,
|
||||
.reserved = 0
|
||||
};
|
||||
|
||||
if (send(conn->socket, &header, sizeof(header), 0) != sizeof(header)) {
|
||||
error("Unable to send to FastCGI socket");
|
||||
return -2;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int fastcgi_php_error(const fastcgi_cnx_t *conn, const char *msg, int msg_len, char *err_msg) {
|
||||
char *msg_str = malloc(msg_len + 1);
|
||||
char *ptr0 = msg_str;
|
||||
memcpy(msg_str, msg, msg_len);
|
||||
msg_str[msg_len] = 0;
|
||||
char *ptr1 = NULL;
|
||||
int len;
|
||||
int err = 0;
|
||||
// FIXME *msg is part of a stream, handle fragmented lines
|
||||
while (1) {
|
||||
log_lvl_t msg_type = LOG_INFO;
|
||||
int msg_pre_len = 0;
|
||||
ptr1 = strstr(ptr0, "PHP message: ");
|
||||
if (ptr1 == NULL) {
|
||||
len = (int) (msg_len - (ptr0 - msg_str));
|
||||
if (ptr0 == msg_str) msg_type = 2;
|
||||
} else {
|
||||
len = (int) (ptr1 - ptr0);
|
||||
}
|
||||
if (len == 0) {
|
||||
goto next;
|
||||
}
|
||||
|
||||
if (len >= 14 && strncmp(ptr0, "PHP Warning: ", 14) == 0) {
|
||||
msg_type = LOG_WARNING;
|
||||
msg_pre_len = 14;
|
||||
} else if (len >= 18 && strncmp(ptr0, "PHP Fatal error: ", 18) == 0) {
|
||||
msg_type = LOG_ERROR;
|
||||
msg_pre_len = 18;
|
||||
} else if (len >= 18 && strncmp(ptr0, "PHP Parse error: ", 18) == 0) {
|
||||
msg_type = LOG_ERROR;
|
||||
msg_pre_len = 18;
|
||||
} else if (len >= 18 && strncmp(ptr0, "PHP Notice: ", 13) == 0) {
|
||||
msg_type = LOG_NOTICE;
|
||||
msg_pre_len = 13;
|
||||
}
|
||||
|
||||
char *ptr2 = ptr0;
|
||||
char *ptr3;
|
||||
int len2;
|
||||
while (ptr2 - ptr0 < len) {
|
||||
ptr3 = strchr(ptr2, '\n');
|
||||
len2 = (int) (len - (ptr2 - ptr0));
|
||||
if (ptr3 != NULL && (ptr3 - ptr2) < len2) {
|
||||
len2 = (int) (ptr3 - ptr2);
|
||||
}
|
||||
logmsgf(msg_type, "%.*s", len2, ptr2);
|
||||
if (msg_type == 2 && ptr2 == ptr0) {
|
||||
strcpy_rem_webroot(err_msg, ptr2, len2, conn->webroot);
|
||||
err = 1;
|
||||
}
|
||||
if (ptr3 == NULL) {
|
||||
break;
|
||||
}
|
||||
ptr2 = ptr3 + 1;
|
||||
}
|
||||
|
||||
next:
|
||||
if (ptr1 == NULL) {
|
||||
break;
|
||||
}
|
||||
ptr0 = ptr1 + 13;
|
||||
}
|
||||
free(msg_str);
|
||||
return err;
|
||||
}
|
||||
|
||||
int fastcgi_header(fastcgi_cnx_t *conn, http_res *res, char *err_msg) {
|
||||
FCGI_Header header;
|
||||
char *content;
|
||||
unsigned short content_len, req_id;
|
||||
long ret;
|
||||
int err = 0;
|
||||
|
||||
while (1) {
|
||||
ret = recv(conn->socket, &header, sizeof(header), 0);
|
||||
if (ret < 0) {
|
||||
res->status = http_get_status(500);
|
||||
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
||||
error("Unable to receive from FastCGI socket");
|
||||
return 1;
|
||||
} else if (ret != sizeof(header)) {
|
||||
res->status = http_get_status(500);
|
||||
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
||||
error("Unable to receive from FastCGI socket");
|
||||
return 1;
|
||||
}
|
||||
req_id = (header.requestIdB1 << 8) | header.requestIdB0;
|
||||
content_len = (header.contentLengthB1 << 8) | header.contentLengthB0;
|
||||
content = malloc(content_len + header.paddingLength);
|
||||
ret = recv(conn->socket, content, content_len + header.paddingLength, 0);
|
||||
if (ret < 0) {
|
||||
res->status = http_get_status(500);
|
||||
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
||||
error("Unable to receive from FastCGI socket");
|
||||
free(content);
|
||||
return 1;
|
||||
} else if (ret != (content_len + header.paddingLength)) {
|
||||
res->status = http_get_status(500);
|
||||
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
||||
error("Unable to receive from FastCGI socket");
|
||||
free(content);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (req_id != conn->req_id) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (header.type == FCGI_END_REQUEST) {
|
||||
FCGI_EndRequestBody *body = (FCGI_EndRequestBody *) content;
|
||||
int app_status = (body->appStatusB3 << 24) | (body->appStatusB2 << 16) | (body->appStatusB1 << 8) |
|
||||
body->appStatusB0;
|
||||
if (body->protocolStatus != FCGI_REQUEST_COMPLETE) {
|
||||
error("FastCGI protocol error: %i", body->protocolStatus);
|
||||
}
|
||||
if (app_status != 0) {
|
||||
error("FastCGI app terminated with exit code %i", app_status);
|
||||
}
|
||||
close(conn->socket);
|
||||
conn->socket = 0;
|
||||
free(content);
|
||||
return 1;
|
||||
} else if (header.type == FCGI_STDERR) {
|
||||
// TODO implement Sesimos backend error handling
|
||||
if (conn->mode == FASTCGI_PHP) {
|
||||
err = err || fastcgi_php_error(conn, content, content_len, err_msg);
|
||||
}
|
||||
} else if (header.type == FCGI_STDOUT) {
|
||||
break;
|
||||
} else {
|
||||
error("Unknown FastCGI type: %i", header.type);
|
||||
}
|
||||
|
||||
free(content);
|
||||
}
|
||||
if (err) {
|
||||
res->status = http_get_status(500);
|
||||
return 2;
|
||||
}
|
||||
|
||||
conn->out_buf = content;
|
||||
conn->out_len = content_len;
|
||||
conn->out_off = (unsigned short) (strstr(content, "\r\n\r\n") - content + 4);
|
||||
|
||||
char *buf = content;
|
||||
unsigned short header_len = conn->out_off;
|
||||
if (header_len <= 0) {
|
||||
error("Unable to parse header: End of header not found");
|
||||
return 1;
|
||||
}
|
||||
|
||||
for (int i = 0; i < header_len; i++) {
|
||||
if ((buf[i] >= 0x00 && buf[i] <= 0x1F && buf[i] != '\r' && buf[i] != '\n') || buf[i] == 0x7F) {
|
||||
error("Unable to parse header: Header contains illegal characters");
|
||||
return 2;
|
||||
}
|
||||
}
|
||||
|
||||
char *ptr = buf;
|
||||
while (header_len != (ptr - buf)) {
|
||||
char *pos0 = strstr(ptr, "\r\n");
|
||||
if (pos0 == NULL) {
|
||||
error("Unable to parse header: Invalid header format");
|
||||
return 1;
|
||||
}
|
||||
|
||||
ret = http_parse_header_field(&res->hdr, ptr, pos0, 0);
|
||||
if (ret != 0) return (int) ret;
|
||||
|
||||
if (pos0[2] == '\r' && pos0[3] == '\n') {
|
||||
return 0;
|
||||
}
|
||||
ptr = pos0 + 2;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int fastcgi_send(fastcgi_cnx_t *conn, sock *client, int flags) {
|
||||
FCGI_Header header;
|
||||
long ret;
|
||||
char buf0[256];
|
||||
int len;
|
||||
char *content, *ptr;
|
||||
unsigned short req_id, content_len;
|
||||
char comp_out[4096];
|
||||
int finish_comp = 0;
|
||||
|
||||
compress_ctx comp_ctx;
|
||||
if (flags & FASTCGI_COMPRESS_BR) {
|
||||
flags &= ~FASTCGI_COMPRESS_GZ;
|
||||
if (compress_init(&comp_ctx, COMPRESS_BR) != 0) {
|
||||
error("Unable to init brotli");
|
||||
flags &= ~FASTCGI_COMPRESS_BR;
|
||||
}
|
||||
} else if (flags & FASTCGI_COMPRESS_GZ) {
|
||||
flags &= ~FASTCGI_COMPRESS_BR;
|
||||
if (compress_init(&comp_ctx, COMPRESS_GZ) != 0) {
|
||||
error("Unable to init gzip");
|
||||
flags &= ~FASTCGI_COMPRESS_GZ;
|
||||
}
|
||||
}
|
||||
|
||||
if (conn->out_buf != NULL && conn->out_len > conn->out_off) {
|
||||
content = conn->out_buf;
|
||||
ptr = content + conn->out_off;
|
||||
content_len = conn->out_len - conn->out_off;
|
||||
goto out;
|
||||
}
|
||||
|
||||
while (1) {
|
||||
ret = recv(conn->socket, &header, sizeof(header), 0);
|
||||
if (ret < 0) {
|
||||
error("Unable to receive from FastCGI socket");
|
||||
return -1;
|
||||
} else if (ret != sizeof(header)) {
|
||||
error("Unable to receive from FastCGI socket: received len (%li) != header len (%li)", ret, sizeof(header));
|
||||
return -1;
|
||||
}
|
||||
|
||||
req_id = (header.requestIdB1 << 8) | header.requestIdB0;
|
||||
content_len = (header.contentLengthB1 << 8) | header.contentLengthB0;
|
||||
content = malloc(content_len + header.paddingLength);
|
||||
ptr = content;
|
||||
|
||||
long rcv_len = 0;
|
||||
while (rcv_len < content_len + header.paddingLength) {
|
||||
ret = recv(conn->socket, content + rcv_len, content_len + header.paddingLength - rcv_len, 0);
|
||||
if (ret < 0) {
|
||||
error("Unable to receive from FastCGI socket");
|
||||
free(content);
|
||||
return -1;
|
||||
}
|
||||
rcv_len += ret;
|
||||
}
|
||||
|
||||
if (header.type == FCGI_END_REQUEST) {
|
||||
FCGI_EndRequestBody *body = (FCGI_EndRequestBody *) content;
|
||||
int app_status = (body->appStatusB3 << 24) | (body->appStatusB2 << 16) | (body->appStatusB1 << 8) |
|
||||
body->appStatusB0;
|
||||
if (body->protocolStatus != FCGI_REQUEST_COMPLETE) {
|
||||
error("FastCGI protocol error: %i", body->protocolStatus);
|
||||
}
|
||||
if (app_status != 0) {
|
||||
error("FastCGI app terminated with exit code %i", app_status);
|
||||
}
|
||||
close(conn->socket);
|
||||
conn->socket = 0;
|
||||
free(content);
|
||||
|
||||
if (flags & FASTCGI_COMPRESS) {
|
||||
finish_comp = 1;
|
||||
content_len = 0;
|
||||
goto out;
|
||||
finish:
|
||||
compress_free(&comp_ctx);
|
||||
}
|
||||
|
||||
if (flags & FASTCGI_CHUNKED) {
|
||||
sock_send(client, "0\r\n\r\n", 5, 0);
|
||||
}
|
||||
|
||||
return 0;
|
||||
} else if (header.type == FCGI_STDERR) {
|
||||
// TODO implement Sesimos backend error handling
|
||||
if (conn->mode == FASTCGI_PHP) {
|
||||
fastcgi_php_error(conn, content, content_len, buf0);
|
||||
}
|
||||
} else if (header.type == FCGI_STDOUT) {
|
||||
unsigned long avail_in, avail_out;
|
||||
out:
|
||||
avail_in = content_len;
|
||||
char *next_in = ptr;
|
||||
do {
|
||||
int buf_len = content_len;
|
||||
if (flags & FASTCGI_COMPRESS) {
|
||||
avail_out = sizeof(comp_out);
|
||||
compress_compress(&comp_ctx, next_in + content_len - avail_in, &avail_in, comp_out, &avail_out, finish_comp);
|
||||
ptr = comp_out;
|
||||
buf_len = (int) (sizeof(comp_out) - avail_out);
|
||||
}
|
||||
if (buf_len != 0) {
|
||||
len = sprintf(buf0, "%X\r\n", buf_len);
|
||||
if (flags & FASTCGI_CHUNKED) sock_send(client, buf0, len, 0);
|
||||
sock_send(client, ptr, buf_len, 0);
|
||||
if (flags & FASTCGI_CHUNKED) sock_send(client, "\r\n", 2, 0);
|
||||
}
|
||||
} while ((flags & FASTCGI_COMPRESS) && (avail_in != 0 || avail_out != sizeof(comp_out)));
|
||||
if (finish_comp) goto finish;
|
||||
} else {
|
||||
error("Unknown FastCGI type: %i", header.type);
|
||||
}
|
||||
free(content);
|
||||
}
|
||||
}
|
||||
|
||||
int fastcgi_dump(fastcgi_cnx_t *conn, char *buf, long len) {
|
||||
FCGI_Header header;
|
||||
long ret;
|
||||
char buf0[256];
|
||||
char *content, *ptr = buf;
|
||||
unsigned short req_id, content_len;
|
||||
|
||||
if (conn->out_buf != NULL && conn->out_len > conn->out_off) {
|
||||
ptr += snprintf(ptr, len, "%.*s", conn->out_len - conn->out_off, conn->out_buf + conn->out_off);
|
||||
}
|
||||
|
||||
while (1) {
|
||||
ret = recv(conn->socket, &header, sizeof(header), 0);
|
||||
if (ret < 0) {
|
||||
error("Unable to receive from FastCGI socket");
|
||||
return -1;
|
||||
} else if (ret != sizeof(header)) {
|
||||
error("Unable to receive from FastCGI socket: received len (%li) != header len (%li)", ret, sizeof(header));
|
||||
return -1;
|
||||
}
|
||||
|
||||
req_id = (header.requestIdB1 << 8) | header.requestIdB0;
|
||||
content_len = (header.contentLengthB1 << 8) | header.contentLengthB0;
|
||||
content = malloc(content_len + header.paddingLength);
|
||||
|
||||
long rcv_len = 0;
|
||||
while (rcv_len < content_len + header.paddingLength) {
|
||||
ret = recv(conn->socket, content + rcv_len, content_len + header.paddingLength - rcv_len, 0);
|
||||
if (ret < 0) {
|
||||
error("Unable to receive from FastCGI socket");
|
||||
free(content);
|
||||
return -1;
|
||||
}
|
||||
rcv_len += ret;
|
||||
}
|
||||
|
||||
if (header.type == FCGI_END_REQUEST) {
|
||||
FCGI_EndRequestBody *body = (FCGI_EndRequestBody *) content;
|
||||
int app_status = (body->appStatusB3 << 24) | (body->appStatusB2 << 16) | (body->appStatusB1 << 8) |
|
||||
body->appStatusB0;
|
||||
if (body->protocolStatus != FCGI_REQUEST_COMPLETE) {
|
||||
error("FastCGI protocol error: %i", body->protocolStatus);
|
||||
}
|
||||
if (app_status != 0) {
|
||||
error("FastCGI app terminated with exit code %i", app_status);
|
||||
}
|
||||
close(conn->socket);
|
||||
conn->socket = 0;
|
||||
free(content);
|
||||
|
||||
return 0;
|
||||
} else if (header.type == FCGI_STDERR) {
|
||||
// TODO implement Sesimos backend error handling
|
||||
if (conn->mode == FASTCGI_PHP) {
|
||||
fastcgi_php_error(conn, content, content_len, buf0);
|
||||
}
|
||||
} else if (header.type == FCGI_STDOUT) {
|
||||
ptr += snprintf(ptr, len - (ptr - buf), "%.*s", content_len, content);
|
||||
} else {
|
||||
error("Unknown FastCGI type: %i", header.type);
|
||||
}
|
||||
free(content);
|
||||
}
|
||||
}
|
||||
|
||||
int fastcgi_receive(fastcgi_cnx_t *conn, sock *client, unsigned long len) {
|
||||
unsigned long rcv_len = 0;
|
||||
char *buf[16384];
|
||||
long ret;
|
||||
FCGI_Header header = {
|
||||
.version = FCGI_VERSION_1,
|
||||
.type = FCGI_STDIN,
|
||||
.requestIdB1 = conn->req_id >> 8,
|
||||
.requestIdB0 = conn->req_id & 0xFF,
|
||||
.contentLengthB1 = 0,
|
||||
.contentLengthB0 = 0,
|
||||
.paddingLength = 0,
|
||||
.reserved = 0
|
||||
};
|
||||
|
||||
while (rcv_len < len) {
|
||||
ret = sock_recv(client, buf, sizeof(buf), 0);
|
||||
if (ret <= 0) {
|
||||
error("Unable to receive");
|
||||
return -1;
|
||||
}
|
||||
|
||||
rcv_len += ret;
|
||||
header.contentLengthB1 = (ret >> 8) & 0xFF;
|
||||
header.contentLengthB0 = ret & 0xFF;
|
||||
if (send(conn->socket, &header, sizeof(header), 0) != sizeof(header)) goto err;
|
||||
if (send(conn->socket, buf, ret, 0) != ret) {
|
||||
err:
|
||||
error("Unable to send to FastCGI socket");
|
||||
return -2;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
int fastcgi_receive_chunked(fastcgi_cnx_t *conn, sock *client) {
|
||||
long ret;
|
||||
unsigned long next_len;
|
||||
|
||||
while (1) {
|
||||
ret = sock_get_chunk_header(client);
|
||||
if (ret < 0) return (int) ret;
|
||||
|
||||
next_len = ret;
|
||||
if (next_len <= 0) break;
|
||||
|
||||
ret = fastcgi_receive(conn, client, next_len);
|
||||
if (ret < 0) return (int) ret;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
61
src/lib/fastcgi.h
Normal file
61
src/lib/fastcgi.h
Normal file
@ -0,0 +1,61 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief FastCGI interface implementation (header file)
|
||||
* @file src/lib/fastcgi.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-26
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_FASTCGI_H
|
||||
#define SESIMOS_FASTCGI_H
|
||||
|
||||
#include "include/fastcgi.h"
|
||||
#include "http.h"
|
||||
#include "uri.h"
|
||||
|
||||
#define FASTCGI_CHUNKED 1
|
||||
#define FASTCGI_COMPRESS_GZ 2
|
||||
#define FASTCGI_COMPRESS_BR 4
|
||||
#define FASTCGI_COMPRESS 6
|
||||
#define FASTCGI_COMPRESS_HOLD 8
|
||||
|
||||
#define FASTCGI_PHP 1
|
||||
#define FASTCGI_SESIMOS 2
|
||||
|
||||
#ifndef PHP_FPM_SOCKET
|
||||
# define PHP_FPM_SOCKET "/var/run/php-fpm/php-fpm.sock"
|
||||
#endif
|
||||
|
||||
#define SESIMOS_BACKEND_SOCKET "/var/run/sesimos/backend.sock"
|
||||
|
||||
typedef struct {
|
||||
int mode;
|
||||
int socket;
|
||||
unsigned short req_id;
|
||||
char *out_buf;
|
||||
const char *webroot;
|
||||
unsigned short out_len;
|
||||
unsigned short out_off;
|
||||
char *r_addr;
|
||||
char *r_host;
|
||||
} fastcgi_cnx_t;
|
||||
|
||||
char *fastcgi_add_param(char *buf, const char *key, const char *value);
|
||||
|
||||
int fastcgi_init(fastcgi_cnx_t *conn, int mode, unsigned int req_num, const sock *client, const http_req *req, const http_uri *uri);
|
||||
|
||||
int fastcgi_close_stdin(fastcgi_cnx_t *conn);
|
||||
|
||||
int fastcgi_php_error(const fastcgi_cnx_t *conn, const char *msg, int msg_len, char *err_msg);
|
||||
|
||||
int fastcgi_header(fastcgi_cnx_t *conn, http_res *res, char *err_msg);
|
||||
|
||||
int fastcgi_send(fastcgi_cnx_t *conn, sock *client, int flags);
|
||||
|
||||
int fastcgi_dump(fastcgi_cnx_t *conn, char *buf, long len);
|
||||
|
||||
int fastcgi_receive(fastcgi_cnx_t *conn, sock *client, unsigned long len);
|
||||
|
||||
int fastcgi_receive_chunked(fastcgi_cnx_t *conn, sock *client);
|
||||
|
||||
#endif //SESIMOS_FASTCGI_H
|
196
src/lib/geoip.c
Normal file
196
src/lib/geoip.c
Normal file
@ -0,0 +1,196 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief MaxMind GeoIP Database interface
|
||||
* @file src/lib/geoip.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-05-04
|
||||
*/
|
||||
|
||||
#include "geoip.h"
|
||||
#include "../logger.h"
|
||||
#include "error.h"
|
||||
#include <memory.h>
|
||||
#include <dirent.h>
|
||||
|
||||
|
||||
static MMDB_s mmdbs[GEOIP_MAX_MMDB];
|
||||
|
||||
static MMDB_entry_data_list_s *geoip_json(MMDB_entry_data_list_s *list, char *str, long *str_off, long str_len) {
|
||||
switch (list->entry_data.type) {
|
||||
case MMDB_DATA_TYPE_MAP:
|
||||
*str_off += sprintf(str + *str_off, "{");
|
||||
break;
|
||||
case MMDB_DATA_TYPE_ARRAY:
|
||||
*str_off += sprintf(str + *str_off, "[");
|
||||
break;
|
||||
case MMDB_DATA_TYPE_UTF8_STRING:
|
||||
*str_off += sprintf(str + *str_off, "\"%.*s\"", list->entry_data.data_size, list->entry_data.utf8_string);
|
||||
break;
|
||||
case MMDB_DATA_TYPE_UINT16:
|
||||
*str_off += sprintf(str + *str_off, "%u", list->entry_data.uint16);
|
||||
break;
|
||||
case MMDB_DATA_TYPE_UINT32:
|
||||
*str_off += sprintf(str + *str_off, "%u", list->entry_data.uint32);
|
||||
break;
|
||||
case MMDB_DATA_TYPE_UINT64:
|
||||
*str_off += sprintf(str + *str_off, "%lu", list->entry_data.uint64);
|
||||
break;
|
||||
case MMDB_DATA_TYPE_UINT128:
|
||||
*str_off += sprintf(str + *str_off, "%llu", (unsigned long long) list->entry_data.uint128);
|
||||
break;
|
||||
case MMDB_DATA_TYPE_INT32:
|
||||
*str_off += sprintf(str + *str_off, "%i", list->entry_data.int32);
|
||||
break;
|
||||
case MMDB_DATA_TYPE_BOOLEAN:
|
||||
*str_off += sprintf(str + *str_off, "%s", list->entry_data.boolean ? "true" : "false");
|
||||
break;
|
||||
case MMDB_DATA_TYPE_FLOAT:
|
||||
*str_off += sprintf(str + *str_off, "%f", list->entry_data.float_value);
|
||||
break;
|
||||
case MMDB_DATA_TYPE_DOUBLE:
|
||||
*str_off += sprintf(str + *str_off, "%f", list->entry_data.double_value);
|
||||
break;
|
||||
}
|
||||
|
||||
if (list->entry_data.type != MMDB_DATA_TYPE_MAP && list->entry_data.type != MMDB_DATA_TYPE_ARRAY)
|
||||
return list->next;
|
||||
|
||||
MMDB_entry_data_list_s *next = list->next;
|
||||
int stat = 0;
|
||||
for (int i = 0; i < list->entry_data.data_size; i++) {
|
||||
next = geoip_json(next, str, str_off, str_len);
|
||||
if (list->entry_data.type == MMDB_DATA_TYPE_MAP) {
|
||||
stat = !stat;
|
||||
if (stat) {
|
||||
i--;
|
||||
*str_off += sprintf(str + *str_off, ":");
|
||||
continue;
|
||||
}
|
||||
}
|
||||
if (i != list->entry_data.data_size - 1)
|
||||
*str_off += sprintf(str + *str_off, ",");
|
||||
}
|
||||
|
||||
*str_off += sprintf(str + *str_off, (list->entry_data.type == MMDB_DATA_TYPE_MAP) ? "}" : "]");
|
||||
|
||||
return next;
|
||||
}
|
||||
|
||||
int geoip_init(const char *directory) {
|
||||
char buf[512];
|
||||
|
||||
memset(mmdbs, 0, sizeof(mmdbs));
|
||||
|
||||
if (directory[0] == 0)
|
||||
return 0;
|
||||
|
||||
DIR *geoip_dir;
|
||||
if ((geoip_dir = opendir(directory)) == NULL)
|
||||
return -1;
|
||||
|
||||
struct dirent *entry;
|
||||
int i = 0, status;
|
||||
while ((entry = readdir(geoip_dir)) != NULL) {
|
||||
if (strcmp(entry->d_name + strlen(entry->d_name) - 5, ".mmdb") != 0)
|
||||
continue;
|
||||
|
||||
if (i >= GEOIP_MAX_MMDB) {
|
||||
critical("Unable to initialize geoip: Too many .mmdb files");
|
||||
closedir(geoip_dir);
|
||||
return 1;
|
||||
}
|
||||
|
||||
sprintf(buf, "%s/%s", directory, entry->d_name);
|
||||
if ((status = MMDB_open(buf, 0, &mmdbs[i])) != MMDB_SUCCESS) {
|
||||
error_mmdb(status);
|
||||
critical("Unable to initialize geoip: Unable to open .mmdb file");
|
||||
closedir(geoip_dir);
|
||||
return 1;
|
||||
}
|
||||
i++;
|
||||
}
|
||||
|
||||
closedir(geoip_dir);
|
||||
|
||||
if (i == 0) {
|
||||
critical("Unable to initialize geoip: No .mmdb files found in %s", directory);
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void geoip_free() {
|
||||
for (int i = 0; i < GEOIP_MAX_MMDB && mmdbs[i].file_content != NULL; i++) {
|
||||
MMDB_close(&mmdbs[i]);
|
||||
}
|
||||
}
|
||||
|
||||
int geoip_lookup_country(struct sockaddr *addr, char *str) {
|
||||
for (int i = 0; i < GEOIP_MAX_MMDB && mmdbs[i].file_content != NULL; i++) {
|
||||
// lookup
|
||||
int mmdb_res;
|
||||
MMDB_lookup_result_s result = MMDB_lookup_sockaddr(&mmdbs[i], addr, &mmdb_res);
|
||||
if (mmdb_res != MMDB_SUCCESS) {
|
||||
return -1;
|
||||
} else if (!result.found_entry) {
|
||||
continue;
|
||||
}
|
||||
|
||||
// get country iso code
|
||||
MMDB_entry_data_s data;
|
||||
int status;
|
||||
if ((status = MMDB_get_value(&result.entry, &data, "country", "iso_code", NULL)) != MMDB_SUCCESS) {
|
||||
if (status == MMDB_IO_ERROR) {
|
||||
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
// no, or invalid data
|
||||
if (!data.has_data || data.type != MMDB_DATA_TYPE_UTF8_STRING)
|
||||
continue;
|
||||
|
||||
// return country code
|
||||
sprintf(str, "%.2s", data.utf8_string);
|
||||
return 0;
|
||||
}
|
||||
|
||||
// not found
|
||||
return 1;
|
||||
}
|
||||
|
||||
int geoip_lookup_json(struct sockaddr *addr, char *json, long len) {
|
||||
long str_off = 0;
|
||||
for (int i = 0; i < GEOIP_MAX_MMDB && mmdbs[i].filename != NULL; i++) {
|
||||
int mmdb_res;
|
||||
MMDB_lookup_result_s result = MMDB_lookup_sockaddr(&mmdbs[i], addr, &mmdb_res);
|
||||
if (mmdb_res != MMDB_SUCCESS) {
|
||||
error_mmdb(mmdb_res);
|
||||
error("Unable to lookup geoip info");
|
||||
continue;
|
||||
} else if (!result.found_entry) {
|
||||
continue;
|
||||
}
|
||||
|
||||
MMDB_entry_data_list_s *list;
|
||||
if ((mmdb_res = MMDB_get_entry_data_list(&result.entry, &list)) != MMDB_SUCCESS) {
|
||||
error_mmdb(mmdb_res);
|
||||
error("Unable to lookup geoip info");
|
||||
continue;
|
||||
}
|
||||
|
||||
long prev = str_off;
|
||||
if (str_off != 0) {
|
||||
str_off--;
|
||||
}
|
||||
geoip_json(list, json, &str_off, len);
|
||||
if (prev != 0) {
|
||||
json[prev - 1] = ',';
|
||||
}
|
||||
|
||||
MMDB_free_entry_data_list(list);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
24
src/lib/geoip.h
Normal file
24
src/lib/geoip.h
Normal file
@ -0,0 +1,24 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief MaxMind GeoIP Database interface (header file)
|
||||
* @file src/lib/geoip.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-05-04
|
||||
*/
|
||||
|
||||
|
||||
#ifndef SESIMOS_GEOIP_H
|
||||
#define SESIMOS_GEOIP_H
|
||||
|
||||
#include <maxminddb.h>
|
||||
|
||||
#define GEOIP_MAX_JSON_SIZE 8192
|
||||
#define GEOIP_MAX_MMDB 3
|
||||
|
||||
int geoip_init(const char *directory);
|
||||
|
||||
void geoip_free();
|
||||
|
||||
int geoip_lookup_country(struct sockaddr *addr, char *str);
|
||||
|
||||
#endif //SESIMOS_GEOIP_H
|
436
src/lib/http.c
Normal file
436
src/lib/http.c
Normal file
@ -0,0 +1,436 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief HTTP implementation
|
||||
* @file src/lib/http.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-09
|
||||
*/
|
||||
|
||||
#include "http.h"
|
||||
#include "utils.h"
|
||||
#include "compress.h"
|
||||
#include "list.h"
|
||||
#include "error.h"
|
||||
|
||||
#include <string.h>
|
||||
|
||||
|
||||
void http_to_camel_case(char *str, int mode) {
|
||||
if (mode == HTTP_PRESERVE)
|
||||
return;
|
||||
|
||||
char ch, last = '-';
|
||||
for (int i = 0; i < strlen(str); i++) {
|
||||
ch = str[i];
|
||||
if (mode == HTTP_CAMEL && last == '-' && ch >= 'a' && ch <= 'z') {
|
||||
str[i] = (char) ((int) ch & 0x5F);
|
||||
} else if (mode == HTTP_LOWER && ch >= 'A' && ch <= 'Z') {
|
||||
str[i] = (char) ((int) ch | 0x20);
|
||||
}
|
||||
last = str[i];
|
||||
}
|
||||
}
|
||||
|
||||
const char *http_field_get_name(const http_field *field) {
|
||||
if (field->type == HTTP_FIELD_NORMAL) {
|
||||
return field->normal.name;
|
||||
} else if (field->type == HTTP_FIELD_EX_VALUE) {
|
||||
return field->ex_value.name;
|
||||
} else if (field->type == HTTP_FIELD_EX_NAME) {
|
||||
return field->ex_name.name;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
const char *http_field_get_value(const http_field *field) {
|
||||
if (field->type == HTTP_FIELD_NORMAL) {
|
||||
return field->normal.value;
|
||||
} else if (field->type == HTTP_FIELD_EX_VALUE) {
|
||||
return field->ex_value.value;
|
||||
} else if (field->type == HTTP_FIELD_EX_NAME) {
|
||||
return field->ex_name.value;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void http_free_field(http_field *f) {
|
||||
if (f->type == HTTP_FIELD_NORMAL) {
|
||||
f->normal.name[0] = 0;
|
||||
f->normal.value[0] = 0;
|
||||
} else if (f->type == HTTP_FIELD_EX_VALUE) {
|
||||
f->ex_value.name[0] = 0;
|
||||
free(f->ex_value.value);
|
||||
f->ex_value.value = NULL;
|
||||
} else if (f->type == HTTP_FIELD_EX_NAME) {
|
||||
free(f->ex_name.name);
|
||||
free(f->ex_name.value);
|
||||
f->ex_name.name = NULL;
|
||||
f->ex_name.value = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
void http_free_hdr(http_hdr *hdr) {
|
||||
for (int i = 0; i < list_size(hdr->fields); i++) {
|
||||
http_free_field(&hdr->fields[i]);
|
||||
}
|
||||
list_free(hdr->fields);
|
||||
hdr->last_field_num = -1;
|
||||
}
|
||||
|
||||
void http_free_req(http_req *req) {
|
||||
if (req->uri != NULL) free(req->uri);
|
||||
req->uri = NULL;
|
||||
http_free_hdr(&req->hdr);
|
||||
}
|
||||
|
||||
void http_free_res(http_res *res) {
|
||||
http_free_hdr(&res->hdr);
|
||||
}
|
||||
|
||||
int http_init_hdr(http_hdr *hdr) {
|
||||
hdr->fields = list_create(sizeof(http_field), HTTP_INIT_HEADER_FIELD_NUM);
|
||||
if (hdr->fields == NULL)
|
||||
return error_http(HTTP_ERROR_SYSCALL);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int http_parse_header_field(http_hdr *hdr, const char *buf, const char *end_ptr, int flags) {
|
||||
if (hdr->last_field_num > list_size(hdr->fields))
|
||||
return error_http(HTTP_ERROR_GENERAL);
|
||||
|
||||
char *pos1 = (char *) buf, *pos2 = (char *) end_ptr;
|
||||
if (buf[0] == ' ' || buf[0] == '\t') {
|
||||
if (hdr->last_field_num == -1)
|
||||
return error_http(HTTP_ERROR_GENERAL);
|
||||
|
||||
http_field *f = &hdr->fields[(int) hdr->last_field_num];
|
||||
|
||||
str_trim_lws(&pos1, &pos2);
|
||||
http_append_to_header_field(f, pos1, pos2 - pos1);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
pos1 = memchr(buf, ':', end_ptr - buf);
|
||||
if (pos1 == NULL)
|
||||
return error_http(HTTP_ERROR_GENERAL);
|
||||
|
||||
long len1 = pos1 - buf;
|
||||
|
||||
pos1++;
|
||||
str_trim_lws(&pos1, &pos2);
|
||||
long len2 = pos2 - pos1;
|
||||
|
||||
int field_num = list_size(hdr->fields);
|
||||
int found = http_get_header_field_num_len(hdr, buf, len1);
|
||||
if (!(flags & HTTP_MERGE_FIELDS) || found == -1) {
|
||||
if (http_add_header_field_len(hdr, buf, len1, pos1, len2 < 0 ? 0 : len2) != 0)
|
||||
return error_http(HTTP_ERROR_TOO_MANY_HEADER_FIELDS);
|
||||
} else {
|
||||
field_num = found;
|
||||
http_append_to_header_field(&hdr->fields[found], ", ", 2);
|
||||
http_append_to_header_field(&hdr->fields[found], pos1, len2);
|
||||
}
|
||||
|
||||
hdr->last_field_num = field_num;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int http_parse_request(char *buf, http_req *req) {
|
||||
char *ptr, *pos0 = buf, *pos1, *pos2;
|
||||
long len;
|
||||
|
||||
unsigned long header_len = strstr(buf, "\r\n\r\n") - buf + 4;
|
||||
if (header_len <= 0)
|
||||
return error_http(HTTP_ERROR_EOH_NOT_FOUND);
|
||||
|
||||
for (int i = 0; i < header_len; i++) {
|
||||
if ((buf[i] >= 0x00 && buf[i] <= 0x1F && buf[i] != '\r' && buf[i] != '\n') || buf[i] == 0x7F)
|
||||
return error_http(HTTP_ERROR_HEADER_MALFORMED);
|
||||
}
|
||||
|
||||
ptr = buf;
|
||||
while (header_len > (ptr - buf + 2)) {
|
||||
pos0 = strstr(ptr, "\r\n");
|
||||
if (pos0 == NULL)
|
||||
return error_http(HTTP_ERROR_HEADER_MALFORMED);
|
||||
|
||||
if (req->version[0] == 0) {
|
||||
pos1 = (char *) strchr(ptr, ' ') + 1;
|
||||
if (pos1 == NULL) goto err_hdr_fmt;
|
||||
|
||||
if (pos1 - ptr - 1 >= sizeof(req->method))
|
||||
return error_http(HTTP_ERROR_HEADER_MALFORMED);
|
||||
|
||||
for (int i = 0; i < (pos1 - ptr - 1); i++) {
|
||||
if (ptr[i] < 'A' || ptr[i] > 'Z')
|
||||
return error_http(HTTP_ERROR_HEADER_MALFORMED);
|
||||
}
|
||||
snprintf(req->method, sizeof(req->method), "%.*s", (int) (pos1 - ptr - 1), ptr);
|
||||
|
||||
pos2 = (char *) strchr(pos1, ' ') + 1;
|
||||
if (pos2 == NULL) {
|
||||
err_hdr_fmt:
|
||||
return error_http(HTTP_ERROR_HEADER_MALFORMED);
|
||||
}
|
||||
|
||||
if (memcmp(pos2, "HTTP/", 5) != 0 || memcmp(pos2 + 8, "\r\n", 2) != 0)
|
||||
return error_http(HTTP_ERROR_INVALID_VERSION);
|
||||
|
||||
len = pos2 - pos1 - 1;
|
||||
if (len >= 2048)
|
||||
return error_http(HTTP_ERROR_URI_TOO_LONG);
|
||||
|
||||
req->uri = malloc(len + 1);
|
||||
sprintf(req->uri, "%.*s", (int) len, pos1);
|
||||
sprintf(req->version, "%.3s", pos2 + 5);
|
||||
} else {
|
||||
if (http_parse_header_field(&req->hdr, ptr, pos0, HTTP_MERGE_FIELDS) != 0)
|
||||
return -1;
|
||||
}
|
||||
ptr = pos0 + 2;
|
||||
}
|
||||
|
||||
if (pos0[2] == '\r' && pos0[3] == '\n') {
|
||||
return (int) header_len;
|
||||
}
|
||||
|
||||
return error_http(HTTP_ERROR_GENERAL);
|
||||
}
|
||||
|
||||
int http_receive_request(sock *client, http_req *req) {
|
||||
long rcv_len;
|
||||
char buf[CLIENT_MAX_HEADER_SIZE];
|
||||
memset(buf, 0, sizeof(buf));
|
||||
memset(req->method, 0, sizeof(req->method));
|
||||
memset(req->version, 0, sizeof(req->version));
|
||||
req->uri = NULL;
|
||||
req->hdr.last_field_num = -1;
|
||||
http_init_hdr(&req->hdr);
|
||||
|
||||
rcv_len = sock_recv(client, buf, CLIENT_MAX_HEADER_SIZE - 1, MSG_PEEK);
|
||||
if (rcv_len <= 0)
|
||||
return -1;
|
||||
|
||||
buf[rcv_len] = 0;
|
||||
|
||||
long header_len = http_parse_request(buf, req);
|
||||
if (header_len < 0)
|
||||
return (int) -header_len;
|
||||
|
||||
rcv_len = sock_recv(client, buf, header_len, 0);
|
||||
if (rcv_len != header_len)
|
||||
return -1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
const char *http_get_header_field(const http_hdr *hdr, const char *field_name) {
|
||||
return http_get_header_field_len(hdr, field_name, strlen(field_name));
|
||||
}
|
||||
|
||||
const char *http_get_header_field_len(const http_hdr *hdr, const char *field_name, unsigned long len) {
|
||||
int num = http_get_header_field_num_len(hdr, field_name, len);
|
||||
return (num >= 0 && num < list_size(hdr->fields)) ? http_field_get_value(&hdr->fields[num]) : NULL;
|
||||
}
|
||||
|
||||
int http_get_header_field_num(const http_hdr *hdr, const char *field_name) {
|
||||
return http_get_header_field_num_len(hdr, field_name, strlen(field_name));
|
||||
}
|
||||
|
||||
int http_get_header_field_num_len(const http_hdr *hdr, const char *field_name, unsigned long len) {
|
||||
for (int i = 0; i < list_size(hdr->fields); i++) {
|
||||
if (strncasecmp(field_name, http_field_get_name(&hdr->fields[i]), len) == 0)
|
||||
return i;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
int http_add_header_field(http_hdr *hdr, const char *field_name, const char *field_value) {
|
||||
return http_add_header_field_len(hdr, field_name, strlen(field_name), field_value, strlen(field_value));
|
||||
}
|
||||
|
||||
int http_add_header_field_len(http_hdr *hdr, const char *name, unsigned long name_len, const char *value, unsigned long value_len) {
|
||||
http_field *f;
|
||||
hdr->fields = list_append_ptr(hdr->fields, (void **) &f);
|
||||
|
||||
if (name_len < sizeof(f->normal.name) && value_len < sizeof(f->normal.value)) {
|
||||
f->type = HTTP_FIELD_NORMAL;
|
||||
memcpy(f->normal.name, name, name_len);
|
||||
memcpy(f->normal.value, value, value_len);
|
||||
f->normal.name[name_len] = 0;
|
||||
f->normal.value[value_len] = 0;
|
||||
http_to_camel_case(f->normal.name, HTTP_PRESERVE);
|
||||
} else if (name_len < sizeof(f->ex_value.name)) {
|
||||
f->type = HTTP_FIELD_EX_VALUE;
|
||||
f->ex_value.value = malloc(value_len + 1);
|
||||
memcpy(f->ex_value.name, name, name_len);
|
||||
memcpy(f->ex_value.value, value, value_len);
|
||||
f->ex_value.name[name_len] = 0;
|
||||
f->ex_value.value[value_len] = 0;
|
||||
http_to_camel_case(f->ex_value.name, HTTP_PRESERVE);
|
||||
} else {
|
||||
f->type = HTTP_FIELD_EX_NAME;
|
||||
f->ex_name.name = malloc(name_len + 1);
|
||||
f->ex_name.value = malloc(value_len + 1);
|
||||
memcpy(f->ex_name.name, name, name_len);
|
||||
memcpy(f->ex_name.value, value, value_len);
|
||||
f->ex_name.name[name_len] = 0;
|
||||
f->ex_name.value[value_len] = 0;
|
||||
http_to_camel_case(f->ex_name.name, HTTP_PRESERVE);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void http_append_to_header_field(http_field *field, const char *value, unsigned long len) {
|
||||
if (field->type == HTTP_FIELD_NORMAL) {
|
||||
unsigned long total_len = strlen(field->normal.value) + len + 1;
|
||||
if (total_len < sizeof(field->normal.value)) {
|
||||
strncat(field->normal.value, value, len);
|
||||
} else {
|
||||
field->type = HTTP_FIELD_EX_VALUE;
|
||||
char *new = malloc(total_len);
|
||||
strcpy(new, field->normal.value);
|
||||
strncat(new, value, len);
|
||||
field->ex_value.value = new;
|
||||
}
|
||||
} else if (field->type == HTTP_FIELD_EX_VALUE) {
|
||||
field->ex_value.value = realloc(field->ex_value.value, strlen(field->ex_value.value) + len + 1);
|
||||
strncat(field->ex_value.value, value, len);
|
||||
} else if (field->type == HTTP_FIELD_EX_NAME) {
|
||||
field->ex_name.value = realloc(field->ex_name.value, strlen(field->ex_name.value) + len + 1);
|
||||
strncat(field->ex_name.value, value, len);
|
||||
}
|
||||
}
|
||||
|
||||
void http_remove_header_field(http_hdr *hdr, const char *field_name, int mode) {
|
||||
int i = 0;
|
||||
int diff = 1;
|
||||
if (mode == HTTP_REMOVE_LAST) {
|
||||
i = list_size(hdr->fields) - 1;
|
||||
diff = -1;
|
||||
}
|
||||
for (; i < list_size(hdr->fields) && i >= 0; i += diff) {
|
||||
if (strcasecmp(field_name, http_field_get_name(&hdr->fields[i])) == 0) {
|
||||
http_free_field(&hdr->fields[i]);
|
||||
list_remove(hdr->fields, i);
|
||||
if (mode == HTTP_REMOVE_ALL) {
|
||||
i -= diff;
|
||||
} else {
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
int http_send_response(sock *client, http_res *res) {
|
||||
char buf[CLIENT_MAX_HEADER_SIZE];
|
||||
long off = sprintf(buf, "HTTP/%s %03i %s\r\n", res->version, res->status->code, res->status->msg);
|
||||
for (int i = 0; i < list_size(res->hdr.fields); i++) {
|
||||
const http_field *f = &res->hdr.fields[i];
|
||||
off += sprintf(buf + off, "%s: %s\r\n", http_field_get_name(f), http_field_get_value(f));
|
||||
}
|
||||
off += sprintf(buf + off, "\r\n");
|
||||
if (sock_send(client, buf, off, 0) != off)
|
||||
return -1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int http_send_request(sock *server, http_req *req) {
|
||||
char buf[CLIENT_MAX_HEADER_SIZE];
|
||||
long off = sprintf(buf, "%s %s HTTP/%s\r\n", req->method, req->uri, req->version);
|
||||
for (int i = 0; i < list_size(req->hdr.fields); i++) {
|
||||
const http_field *f = &req->hdr.fields[i];
|
||||
off += sprintf(buf + off, "%s: %s\r\n", http_field_get_name(f), http_field_get_value(f));
|
||||
}
|
||||
off += sprintf(buf + off, "\r\n");
|
||||
if (sock_send(server, buf, off, 0) != off)
|
||||
return -1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
const http_status *http_get_status(unsigned short status_code) {
|
||||
for (int i = 0; i < http_statuses_size; i++) {
|
||||
if (http_statuses[i].code == status_code) {
|
||||
return &http_statuses[i];
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
const http_status_msg *http_get_error_msg(const http_status *status) {
|
||||
unsigned short code = status->code;
|
||||
for (int i = 0; i < http_status_messages_size; i++) {
|
||||
if (http_status_messages[i].code == code) {
|
||||
return &http_status_messages[i];
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
const char *http_get_status_color(const http_status *status) {
|
||||
unsigned short code = status->code;
|
||||
if (code >= 100 && code < 200) {
|
||||
return HTTP_1XX_STR;
|
||||
} else if ((code >= 200 && code < 300) || code == 304) {
|
||||
return HTTP_2XX_STR;
|
||||
} else if (code >= 300 && code < 400) {
|
||||
return HTTP_3XX_STR;
|
||||
} else if (code >= 400 && code < 500) {
|
||||
return HTTP_4XX_STR;
|
||||
} else if (code >= 500 && code < 600) {
|
||||
return HTTP_5XX_STR;
|
||||
}
|
||||
return "";
|
||||
}
|
||||
|
||||
char *http_format_date(time_t time, char *buf, size_t size) {
|
||||
struct tm timeinfo;
|
||||
strftime(buf, size, "%a, %d %b %Y %H:%M:%S GMT", gmtime_r(&time, &timeinfo));
|
||||
return buf;
|
||||
}
|
||||
|
||||
char *http_get_date(char *buf, size_t size) {
|
||||
time_t rawtime;
|
||||
time(&rawtime);
|
||||
return http_format_date(rawtime, buf, size);
|
||||
}
|
||||
|
||||
const http_doc_info *http_get_status_info(const http_status *status) {
|
||||
unsigned short code = status->code;
|
||||
static http_doc_info info[] = {
|
||||
{"info", HTTP_COLOR_INFO, "/.sesimos/res/icon-info.svg", http_info_doc},
|
||||
{"success", HTTP_COLOR_SUCCESS, "/.sesimos/res/icon-success.svg", http_success_doc},
|
||||
{"warning", HTTP_COLOR_WARNING, "/.sesimos/res/icon-warning.svg", http_warning_doc},
|
||||
{"error", HTTP_COLOR_ERROR, "/.sesimos/res/icon-error.svg", http_error_doc}
|
||||
};
|
||||
if (code >= 100 && code < 200) {
|
||||
return &info[0];
|
||||
} else if ((code >= 200 && code < 300) || code == 304) {
|
||||
return &info[1];
|
||||
} else if (code >= 300 && code < 400) {
|
||||
return &info[2];
|
||||
} else if (code >= 400 && code < 600) {
|
||||
return &info[3];
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
int http_get_compression(const http_req *req, const http_res *res) {
|
||||
const char *accept_encoding = http_get_header_field(&req->hdr, "Accept-Encoding");
|
||||
const char *content_type = http_get_header_field(&res->hdr, "Content-Type");
|
||||
const char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding");
|
||||
if (mime_is_compressible(content_type) && content_encoding == NULL && accept_encoding != NULL) {
|
||||
if (strstr(accept_encoding, "br") != NULL) {
|
||||
return COMPRESS_BR;
|
||||
} else if (strstr(accept_encoding, "gzip") != NULL) {
|
||||
return COMPRESS_GZ;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
192
src/lib/http.h
Normal file
192
src/lib/http.h
Normal file
@ -0,0 +1,192 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief HTTP implementation (header file)
|
||||
* @file src/lib/http.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-09
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_HTTP_H
|
||||
#define SESIMOS_HTTP_H
|
||||
|
||||
#include "sock.h"
|
||||
|
||||
#define HTTP_PRESERVE 0
|
||||
#define HTTP_LOWER 1
|
||||
#define HTTP_CAMEL 2
|
||||
|
||||
#define HTTP_REMOVE_ONE 0
|
||||
#define HTTP_REMOVE_ALL 1
|
||||
#define HTTP_REMOVE_LAST 2
|
||||
|
||||
#define HTTP_FIELD_NORMAL 0
|
||||
#define HTTP_FIELD_EX_VALUE 1
|
||||
#define HTTP_FIELD_EX_NAME 2
|
||||
|
||||
#define HTTP_MERGE_FIELDS 1
|
||||
|
||||
#define HTTP_1XX_STR "\x1B[1;32m"
|
||||
#define HTTP_2XX_STR "\x1B[1;32m"
|
||||
#define HTTP_3XX_STR "\x1B[1;33m"
|
||||
#define HTTP_4XX_STR "\x1B[1;31m"
|
||||
#define HTTP_5XX_STR "\x1B[1;31m"
|
||||
|
||||
#define HTTP_COLOR_SUCCESS "#008000"
|
||||
#define HTTP_COLOR_INFO "#606060"
|
||||
#define HTTP_COLOR_WARNING "#E0C000"
|
||||
#define HTTP_COLOR_ERROR "#C00000"
|
||||
|
||||
#define CLIENT_MAX_HEADER_SIZE 8192
|
||||
#define HTTP_INIT_HEADER_FIELD_NUM 16
|
||||
|
||||
#define HTTP_TYPE_INFORMATIONAL 1
|
||||
#define HTTP_TYPE_SUCCESS 2
|
||||
#define HTTP_TYPE_REDIRECTION 3
|
||||
#define HTTP_TYPE_CLIENT_ERROR 4
|
||||
#define HTTP_TYPE_SERVER_ERROR 5
|
||||
|
||||
#define HTTP_ERROR_GENERAL 1
|
||||
#define HTTP_ERROR_SYSCALL 2
|
||||
#define HTTP_ERROR_TOO_MANY_HEADER_FIELDS 3
|
||||
#define HTTP_ERROR_EOH_NOT_FOUND 4
|
||||
#define HTTP_ERROR_HEADER_MALFORMED 5
|
||||
#define HTTP_ERROR_INVALID_VERSION 6
|
||||
#define HTTP_ERROR_URI_TOO_LONG 7
|
||||
#define HTTP_ERROR_
|
||||
|
||||
#ifndef SERVER_STR
|
||||
# define SERVER_STR "sesimos"
|
||||
#endif
|
||||
|
||||
#ifndef SERVER_STR_HTML
|
||||
# define SERVER_STR_HTML "sesimos web server"
|
||||
#endif
|
||||
|
||||
typedef struct {
|
||||
unsigned short code:10;
|
||||
unsigned char type:3;
|
||||
char msg[64];
|
||||
} http_status;
|
||||
|
||||
typedef struct {
|
||||
unsigned short code:10;
|
||||
const char *msg;
|
||||
} http_status_msg;
|
||||
|
||||
typedef struct {
|
||||
char mode[8];
|
||||
char color[8];
|
||||
const char *icon;
|
||||
const char *doc;
|
||||
} http_doc_info;
|
||||
|
||||
typedef struct {
|
||||
char type;
|
||||
union {
|
||||
struct {
|
||||
char name[32];
|
||||
char value[32];
|
||||
} normal;
|
||||
struct {
|
||||
char name[64 - sizeof(char *)];
|
||||
char *value;
|
||||
} ex_value;
|
||||
struct {
|
||||
char *name;
|
||||
char *value;
|
||||
} ex_name;
|
||||
};
|
||||
} http_field;
|
||||
|
||||
typedef struct {
|
||||
int last_field_num;
|
||||
http_field *fields;
|
||||
} http_hdr;
|
||||
|
||||
typedef struct {
|
||||
char method[16];
|
||||
char *uri;
|
||||
char version[4];
|
||||
http_hdr hdr;
|
||||
} http_req;
|
||||
|
||||
typedef struct {
|
||||
const http_status *status;
|
||||
char version[4];
|
||||
http_hdr hdr;
|
||||
} http_res;
|
||||
|
||||
typedef enum {
|
||||
NONE, INTERNAL, CLIENT_REQ, SERVER_REQ, SERVER, SERVER_RES, CLIENT_RES
|
||||
} http_error_origin;
|
||||
|
||||
typedef struct {
|
||||
unsigned short status;
|
||||
http_error_origin origin;
|
||||
const char* ws_key;
|
||||
} http_status_ctx;
|
||||
|
||||
extern const http_status http_statuses[];
|
||||
extern const http_status_msg http_status_messages[];
|
||||
extern const int http_statuses_size;
|
||||
extern const int http_status_messages_size;
|
||||
|
||||
extern const char http_error_doc[], http_warning_doc[], http_success_doc[], http_info_doc[];
|
||||
|
||||
void http_to_camel_case(char *str, int mode);
|
||||
|
||||
const char *http_field_get_name(const http_field *field);
|
||||
|
||||
const char *http_field_get_value(const http_field *field);
|
||||
|
||||
int http_init_hdr(http_hdr *hdr);
|
||||
|
||||
void http_free_field(http_field *f);
|
||||
|
||||
void http_free_hdr(http_hdr *hdr);
|
||||
|
||||
void http_free_req(http_req *req);
|
||||
|
||||
void http_free_res(http_res *res);
|
||||
|
||||
int http_parse_request(char *buf, http_req *req);
|
||||
|
||||
int http_receive_request(sock *client, http_req *req);
|
||||
|
||||
int http_parse_header_field(http_hdr *hdr, const char *buf, const char *end_ptr, int flags);
|
||||
|
||||
const char *http_get_header_field(const http_hdr *hdr, const char *field_name);
|
||||
|
||||
const char *http_get_header_field_len(const http_hdr *hdr, const char *field_name, unsigned long len);
|
||||
|
||||
int http_get_header_field_num(const http_hdr *hdr, const char *field_name);
|
||||
|
||||
int http_get_header_field_num_len(const http_hdr *hdr, const char *field_name, unsigned long len);
|
||||
|
||||
int http_add_header_field(http_hdr *hdr, const char *field_name, const char *field_value);
|
||||
|
||||
int http_add_header_field_len(http_hdr *hdr, const char *name, unsigned long name_len, const char *value, unsigned long value_len);
|
||||
|
||||
void http_append_to_header_field(http_field *field, const char *value, unsigned long len);
|
||||
|
||||
void http_remove_header_field(http_hdr *hdr, const char *field_name, int mode);
|
||||
|
||||
int http_send_response(sock *client, http_res *res);
|
||||
|
||||
int http_send_request(sock *server, http_req *req);
|
||||
|
||||
const http_status *http_get_status(unsigned short status_code);
|
||||
|
||||
const http_status_msg *http_get_error_msg(const http_status *status);
|
||||
|
||||
const char *http_get_status_color(const http_status *status);
|
||||
|
||||
char *http_format_date(time_t time, char *buf, size_t size);
|
||||
|
||||
char *http_get_date(char *buf, size_t size);
|
||||
|
||||
const http_doc_info *http_get_status_info(const http_status *status);
|
||||
|
||||
int http_get_compression(const http_req *req, const http_res *res);
|
||||
|
||||
#endif //SESIMOS_HTTP_H
|
212
src/lib/http_static.c
Normal file
212
src/lib/http_static.c
Normal file
@ -0,0 +1,212 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief HTTP static implementation
|
||||
* @file src/lib/http_static.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-05-03
|
||||
* @details https://www.iana.org/assignments/http-status-codes/http-status-codes.xhtml
|
||||
*/
|
||||
|
||||
#include "http.h"
|
||||
|
||||
const http_status http_statuses[] = {
|
||||
{100, HTTP_TYPE_INFORMATIONAL, "Continue"},
|
||||
{101, HTTP_TYPE_INFORMATIONAL, "Switching Protocols"},
|
||||
{102, HTTP_TYPE_INFORMATIONAL, "Processing"},
|
||||
{103, HTTP_TYPE_INFORMATIONAL, "Early Hints"},
|
||||
|
||||
{200, HTTP_TYPE_SUCCESS, "OK"},
|
||||
{201, HTTP_TYPE_SUCCESS, "Created"},
|
||||
{202, HTTP_TYPE_SUCCESS, "Accepted"},
|
||||
{203, HTTP_TYPE_SUCCESS, "Non-Authoritative Information"},
|
||||
{204, HTTP_TYPE_SUCCESS, "No Content"},
|
||||
{205, HTTP_TYPE_SUCCESS, "Reset Content"},
|
||||
{206, HTTP_TYPE_SUCCESS, "Partial Content"},
|
||||
{207, HTTP_TYPE_SUCCESS, "Multi-Status"},
|
||||
{208, HTTP_TYPE_SUCCESS, "Already Reported"},
|
||||
{226, HTTP_TYPE_SUCCESS, "Instance Manipulation Used"},
|
||||
|
||||
{300, HTTP_TYPE_REDIRECTION, "Multiple Choices"},
|
||||
{301, HTTP_TYPE_REDIRECTION, "Moved Permanently"},
|
||||
{302, HTTP_TYPE_REDIRECTION, "Found"},
|
||||
{303, HTTP_TYPE_REDIRECTION, "See Other"},
|
||||
{304, HTTP_TYPE_SUCCESS, "Not Modified"},
|
||||
{305, HTTP_TYPE_REDIRECTION, "Use Proxy"},
|
||||
{307, HTTP_TYPE_REDIRECTION, "Temporary Redirect"},
|
||||
{308, HTTP_TYPE_REDIRECTION, "Permanent Redirect"},
|
||||
|
||||
{400, HTTP_TYPE_CLIENT_ERROR, "Bad Request"},
|
||||
{401, HTTP_TYPE_CLIENT_ERROR, "Unauthorized"},
|
||||
{402, HTTP_TYPE_CLIENT_ERROR, "Payment Required"},
|
||||
{403, HTTP_TYPE_CLIENT_ERROR, "Forbidden"},
|
||||
{404, HTTP_TYPE_CLIENT_ERROR, "Not Found"},
|
||||
{405, HTTP_TYPE_CLIENT_ERROR, "Method Not Allowed"},
|
||||
{406, HTTP_TYPE_CLIENT_ERROR, "Not Acceptable"},
|
||||
{407, HTTP_TYPE_CLIENT_ERROR, "Proxy Authentication Required"},
|
||||
{408, HTTP_TYPE_CLIENT_ERROR, "Request Timeout"},
|
||||
{409, HTTP_TYPE_CLIENT_ERROR, "Conflict"},
|
||||
{410, HTTP_TYPE_CLIENT_ERROR, "Gone"},
|
||||
{411, HTTP_TYPE_CLIENT_ERROR, "Length Required"},
|
||||
{412, HTTP_TYPE_CLIENT_ERROR, "Precondition Failed"},
|
||||
{413, HTTP_TYPE_CLIENT_ERROR, "Request Entity Too Large"},
|
||||
{414, HTTP_TYPE_CLIENT_ERROR, "Request-URI Too Long"},
|
||||
{415, HTTP_TYPE_CLIENT_ERROR, "Unsupported Media Type"},
|
||||
{416, HTTP_TYPE_CLIENT_ERROR, "Range Not Satisfiable"},
|
||||
{417, HTTP_TYPE_CLIENT_ERROR, "Expectation Failed"},
|
||||
{421, HTTP_TYPE_CLIENT_ERROR, "Misdirected Request"},
|
||||
{422, HTTP_TYPE_CLIENT_ERROR, "Unprocessable Content"},
|
||||
{423, HTTP_TYPE_CLIENT_ERROR, "Locked"},
|
||||
{424, HTTP_TYPE_CLIENT_ERROR, "Failed Dependency"},
|
||||
{425, HTTP_TYPE_CLIENT_ERROR, "Too Early"},
|
||||
{426, HTTP_TYPE_CLIENT_ERROR, "Upgrade Required"},
|
||||
{428, HTTP_TYPE_CLIENT_ERROR, "Precondition Required"},
|
||||
{429, HTTP_TYPE_CLIENT_ERROR, "Too Many Requests"},
|
||||
{431, HTTP_TYPE_CLIENT_ERROR, "Request Header Fields Too Large"},
|
||||
{451, HTTP_TYPE_CLIENT_ERROR, "Unavailable For Legal Reasons"},
|
||||
|
||||
{500, HTTP_TYPE_SERVER_ERROR, "Internal Server Error"},
|
||||
{501, HTTP_TYPE_SERVER_ERROR, "Not Implemented"},
|
||||
{502, HTTP_TYPE_SERVER_ERROR, "Bad Gateway"},
|
||||
{503, HTTP_TYPE_SERVER_ERROR, "Service Unavailable"},
|
||||
{504, HTTP_TYPE_SERVER_ERROR, "Gateway Timeout"},
|
||||
{505, HTTP_TYPE_SERVER_ERROR, "HTTP Version Not Supported"},
|
||||
{506, HTTP_TYPE_SERVER_ERROR, "Variant Also Negotiates"},
|
||||
{507, HTTP_TYPE_SERVER_ERROR, "Insufficient Storage"},
|
||||
{508, HTTP_TYPE_SERVER_ERROR, "Loop Detected"},
|
||||
{511, HTTP_TYPE_SERVER_ERROR, "Network Authentication Required"},
|
||||
};
|
||||
|
||||
const http_status_msg http_status_messages[] = {
|
||||
{100, "The client SHOULD continue with its request. The server MUST send a final response after the request "
|
||||
"has been completed."},
|
||||
{101, "The server understands and is willing to comply with the clients request, via the Upgrade message "
|
||||
"header field, for a change in the application protocol being used on this connection."},
|
||||
{102, "The server has a reasonable expectation that the request will take significant time to complete. The "
|
||||
"server MUST send a final response after the request has been completed."},
|
||||
{103, "The client can speculatively evaluate the header fields included in the response while waiting for the "
|
||||
"final response. The server MUST send a final response after the request has been completed."},
|
||||
|
||||
{200, "The request has succeeded."},
|
||||
{201, "The request has been fulfilled and resulted in a new resource being created."},
|
||||
{202, "The request has been accepted for processing, but the processing has not been completed."},
|
||||
{203, "The returned meta information in the entity-header is not the definitive set as available from the "
|
||||
"origin server, but is gathered from a local or a third-party copy."},
|
||||
{204, "The server has fulfilled the request but does not need to return an entity-body, and might want to "
|
||||
"return updated meta information."},
|
||||
{205, "The server has fulfilled the request and the user agent SHOULD reset the document view which caused the "
|
||||
"request to be sent."},
|
||||
{206, "The server has fulfilled the partial GET request for the resource."},
|
||||
{207, "The response provides status for multiple independent operations."},
|
||||
{208, "The response is used to avoid enumerating the internal members of multiple bindings to the same "
|
||||
"collection repeatedly."},
|
||||
{226, "The server has fulfilled a GET request for the resource, and the response is a representation of the "
|
||||
"result of one or more instance-manipulations applied to the current instance."},
|
||||
|
||||
{300, "The requested resource corresponds to any one of a set of representations, each with its own specific "
|
||||
"location, and agent-driven negotiation information is being provided so that the user (or user agent) "
|
||||
"can select a preferred representation and redirect its request to that location."},
|
||||
{301, "The requested resource has been assigned a new permanent URI and any future references to this resource "
|
||||
"SHOULD use one of the returned URIs."},
|
||||
{302, "The requested resource resides temporarily under a different URI."},
|
||||
{303, "The response to the request can be found under a different URI and SHOULD be retrieved using a GET "
|
||||
"method on that resource."},
|
||||
{304, "The request has been fulfilled and the requested resource has not been modified."},
|
||||
{305, "The requested resource MUST be accessed through the proxy given by the Location field."},
|
||||
{307, "The requested resource resides temporarily under a different URI."},
|
||||
{308, "The requested resource has been assigned a new permanent URI and any future references to this resource "
|
||||
"ought to use one of the enclosed URIs."},
|
||||
|
||||
{400, "The request could not be understood by the server due to malformed syntax."},
|
||||
{401, "The request requires user authentication."},
|
||||
{403, "The server understood the request, but is refusing to fulfill it."},
|
||||
{404, "The server has not found anything matching the Request-URI."},
|
||||
{405, "The method specified in the Request-Line is not allowed for the resource identified by the "
|
||||
"Request-URI."},
|
||||
{406, "The resource identified by the request is only capable of generating response entities which have "
|
||||
"content characteristics not acceptable according to the accept headers sent in the request."},
|
||||
{407, "The request requires user authentication on the proxy."},
|
||||
{408, "The client did not produce a request within the time that the server was prepared to wait."},
|
||||
{409, "The request could not be completed due to a conflict with the current state of the resource."},
|
||||
{410, "The requested resource is no longer available at the server and no forwarding address is known."},
|
||||
{411, "The server refuses to accept the request without a defined Content-Length."},
|
||||
{412, "The precondition given in one or more of the request-header fields evaluated to false when it was "
|
||||
"tested on the server."},
|
||||
{413, "The server is refusing to process a request because the request entity is larger than the server is "
|
||||
"willing or able to process."},
|
||||
{414, "The server is refusing to service the request because the Request-URI is longer than the server is "
|
||||
"willing to interpret."},
|
||||
{415, "The server is refusing to service the request because the entity of the request is in a format not "
|
||||
"supported by the requested resource for the requested method."},
|
||||
{416, "None of the ranges in the requests Range header field overlap the current extent of the selected "
|
||||
"resource or that the set of ranges requested has been rejected due to invalid ranges or an excessive "
|
||||
"request of small or overlapping ranges."},
|
||||
{417, "The expectation given in an Expect request-header field could not be met by this server, or, if the "
|
||||
"server is a proxy, the server has unambiguous evidence that the request could not be met by the "
|
||||
"next-hop server."},
|
||||
{421, "The server is not able to produce a response. The client MAY retry the request over a different "
|
||||
"connection."},
|
||||
{422, "The server understands the content type of the request content, and the syntax of the request content "
|
||||
"is correct, but the server was unable to process the contained information."},
|
||||
{423, "The source or destination resource of a method is locked."},
|
||||
{424, "The method could not be performed on the resource because the requested action depended on another "
|
||||
"action and that action failed."},
|
||||
{425, "The server is unwilling to risk processing a request that might be replayed."},
|
||||
{426, "The server refuses to perform the request using the current protocol but might be willing to do so "
|
||||
"after the client upgrades to a different protocol. The server MUST send an Upgrade header field to"
|
||||
"indicate the required protocol(s)."},
|
||||
{428, "The origin server requires the request to be conditional. By requiring requests to be conditional, the "
|
||||
"server can assure that clients are working with the correct copies and thus avoiding a lost update."},
|
||||
{429, "The client has sent too many requests in a given amount of time."},
|
||||
{431, "The server is unwilling to process the request because its header fields are too large. The request MAY "
|
||||
"be resubmitted after reducing the size of the request header fields."},
|
||||
{451, "The server is denying access to the resource as a consequence of a legal demand."},
|
||||
|
||||
{500, "The server encountered an unexpected condition which prevented it from fulfilling the request."},
|
||||
{501, "The server does not support the functionality required to fulfill the request."},
|
||||
{502, "The server, while acting as a gateway or proxy, received an invalid response from the upstream server "
|
||||
"it accessed in attempting to fulfill the request."},
|
||||
{503, "The server is currently unable to handle the request due to a temporary overloading or maintenance of "
|
||||
"the server."},
|
||||
{504, "The server, while acting as a gateway or proxy, did not receive a timely response from the upstream "
|
||||
"server specified by the URI or some other auxiliary server it needed to access in attempting to "
|
||||
"complete the request."},
|
||||
{505, "The server does not support, or refuses to support, the HTTP protocol version that was used in the "
|
||||
"request message."},
|
||||
{506, "The server has an internal configuration error: the chosen variant resource is configured to engage in "
|
||||
"transparent content negotiation itself, and is therefore not a proper end point in the negotiation "
|
||||
"process."},
|
||||
{507, "The method could not be performed on the resource because the server is unable to store the "
|
||||
"representation needed to successfully complete the request. This condition is considered to be "
|
||||
"temporary."},
|
||||
{508, "The server terminated an operation because it encountered an infinite loop while processing the "
|
||||
"request."},
|
||||
{511, "The client needs to authenticate to gain network access. The response representation SHOULD contain a "
|
||||
"link to a resource that allows the user to submit credentials."},
|
||||
};
|
||||
|
||||
const int http_statuses_size = sizeof(http_statuses) / sizeof(http_status);
|
||||
const int http_status_messages_size = sizeof(http_status_messages) / sizeof(http_status_msg);
|
||||
|
||||
const char http_error_doc[] =
|
||||
" <h1>%1$i</h1>\n"
|
||||
" <h2>%2$s :(</h2>\n"
|
||||
" <p>%3$s</p>\n"
|
||||
" <p>%4$s</p>\n";
|
||||
|
||||
const char http_warning_doc[] =
|
||||
" <h1>%1$i</h1>\n"
|
||||
" <h2>%2$s :)</h2>\n"
|
||||
" <p>%3$s</p>\n"
|
||||
" <p>%4$s</p>\n";
|
||||
|
||||
const char http_success_doc[] =
|
||||
" <h1>%1$i</h1>\n"
|
||||
" <h2>%2$s :)</h2>\n"
|
||||
" <p>%3$s</p>\n"
|
||||
" <p>%4$s</p>\n";
|
||||
|
||||
const char http_info_doc[] =
|
||||
" <h1>%1$i</h1>\n"
|
||||
" <h2>%2$s :)</h2>\n"
|
||||
" <p>%3$s</p>\n"
|
||||
" <p>%4$s</p>\n";
|
121
src/lib/include/fastcgi.h
Normal file
121
src/lib/include/fastcgi.h
Normal file
@ -0,0 +1,121 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief FastCGI header file
|
||||
* @file src/lib/include/fastcgi.h
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_EXTERN_FASTCGI_H
|
||||
#define SESIMOS_EXTERN_FASTCGI_H
|
||||
|
||||
/*
|
||||
* Listening socket file number
|
||||
*/
|
||||
#define FCGI_LISTENSOCK_FILENO 0
|
||||
|
||||
typedef struct {
|
||||
unsigned char version;
|
||||
unsigned char type;
|
||||
unsigned char requestIdB1;
|
||||
unsigned char requestIdB0;
|
||||
unsigned char contentLengthB1;
|
||||
unsigned char contentLengthB0;
|
||||
unsigned char paddingLength;
|
||||
unsigned char reserved;
|
||||
} FCGI_Header;
|
||||
|
||||
/*
|
||||
* Number of bytes in a FCGI_Header. Future versions of the protocol
|
||||
* will not reduce this number.
|
||||
*/
|
||||
#define FCGI_HEADER_LEN 8
|
||||
|
||||
/*
|
||||
* Value for version component of FCGI_Header
|
||||
*/
|
||||
#define FCGI_VERSION_1 1
|
||||
|
||||
/*
|
||||
* Values for type component of FCGI_Header
|
||||
*/
|
||||
#define FCGI_BEGIN_REQUEST 1
|
||||
#define FCGI_ABORT_REQUEST 2
|
||||
#define FCGI_END_REQUEST 3
|
||||
#define FCGI_PARAMS 4
|
||||
#define FCGI_STDIN 5
|
||||
#define FCGI_STDOUT 6
|
||||
#define FCGI_STDERR 7
|
||||
#define FCGI_DATA 8
|
||||
#define FCGI_GET_VALUES 9
|
||||
#define FCGI_GET_VALUES_RESULT 10
|
||||
#define FCGI_UNKNOWN_TYPE 11
|
||||
#define FCGI_MAXTYPE (FCGI_UNKNOWN_TYPE)
|
||||
|
||||
/*
|
||||
* Value for requestId component of FCGI_Header
|
||||
*/
|
||||
#define FCGI_NULL_REQUEST_ID 0
|
||||
|
||||
typedef struct {
|
||||
unsigned char roleB1;
|
||||
unsigned char roleB0;
|
||||
unsigned char flags;
|
||||
unsigned char reserved[5];
|
||||
} FCGI_BeginRequestBody;
|
||||
|
||||
typedef struct {
|
||||
FCGI_Header header;
|
||||
FCGI_BeginRequestBody body;
|
||||
} FCGI_BeginRequestRecord;
|
||||
|
||||
/*
|
||||
* Mask for flags component of FCGI_BeginRequestBody
|
||||
*/
|
||||
#define FCGI_KEEP_CONN 1
|
||||
|
||||
/*
|
||||
* Values for role component of FCGI_BeginRequestBody
|
||||
*/
|
||||
#define FCGI_RESPONDER 1
|
||||
#define FCGI_AUTHORIZER 2
|
||||
#define FCGI_FILTER 3
|
||||
|
||||
typedef struct {
|
||||
unsigned char appStatusB3;
|
||||
unsigned char appStatusB2;
|
||||
unsigned char appStatusB1;
|
||||
unsigned char appStatusB0;
|
||||
unsigned char protocolStatus;
|
||||
unsigned char reserved[3];
|
||||
} FCGI_EndRequestBody;
|
||||
|
||||
typedef struct {
|
||||
FCGI_Header header;
|
||||
FCGI_EndRequestBody body;
|
||||
} FCGI_EndRequestRecord;
|
||||
|
||||
/*
|
||||
* Values for protocolStatus component of FCGI_EndRequestBody
|
||||
*/
|
||||
#define FCGI_REQUEST_COMPLETE 0
|
||||
#define FCGI_CANT_MPX_CONN 1
|
||||
#define FCGI_OVERLOADED 2
|
||||
#define FCGI_UNKNOWN_ROLE 3
|
||||
|
||||
/*
|
||||
* Variable names for FCGI_GET_VALUES / FCGI_GET_VALUES_RESULT records
|
||||
*/
|
||||
#define FCGI_MAX_CONNS "FCGI_MAX_CONNS"
|
||||
#define FCGI_MAX_REQS "FCGI_MAX_REQS"
|
||||
#define FCGI_MPXS_CONNS "FCGI_MPXS_CONNS"
|
||||
|
||||
typedef struct {
|
||||
unsigned char type;
|
||||
unsigned char reserved[7];
|
||||
} FCGI_UnknownTypeBody;
|
||||
|
||||
typedef struct {
|
||||
FCGI_Header header;
|
||||
FCGI_UnknownTypeBody body;
|
||||
} FCGI_UnknownTypeRecord;
|
||||
|
||||
#endif //SESIMOS_EXTERN_FASTCGI_H
|
139
src/lib/list.c
Normal file
139
src/lib/list.c
Normal file
@ -0,0 +1,139 @@
|
||||
|
||||
#include "list.h"
|
||||
|
||||
#include <malloc.h>
|
||||
#include <memory.h>
|
||||
#include <errno.h>
|
||||
|
||||
#define FACTOR 4
|
||||
|
||||
typedef struct {
|
||||
int init_size, elem_size, max_size, size;
|
||||
} list_meta_t;
|
||||
|
||||
static void *list_resize(list_meta_t *list, int new_size) {
|
||||
if (new_size <= 0)
|
||||
return NULL;
|
||||
|
||||
list_meta_t *new_ptr = realloc(list, sizeof(list_meta_t) + list->elem_size * new_size);
|
||||
if (new_ptr == NULL)
|
||||
return NULL;
|
||||
|
||||
new_ptr->max_size = new_size;
|
||||
return new_ptr;
|
||||
}
|
||||
|
||||
void *list_create(int elem_size, int init_elem_n) {
|
||||
if (elem_size <= 0 || init_elem_n <= 0) {
|
||||
errno = EINVAL;
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void *list_ptr = malloc(sizeof(list_meta_t) + elem_size * init_elem_n);
|
||||
list_meta_t *list = list_ptr;
|
||||
list->init_size = init_elem_n;
|
||||
list->elem_size = elem_size;
|
||||
list->max_size = init_elem_n;
|
||||
list->size = 0;
|
||||
return (unsigned char *) list_ptr + sizeof(list_meta_t);
|
||||
}
|
||||
|
||||
int list_size(const void *list_ptr) {
|
||||
list_meta_t *list = (void *) ((unsigned char *) list_ptr - sizeof(list_meta_t));
|
||||
return list->size;
|
||||
}
|
||||
|
||||
int list_find(void *list_ptr, void *elem) {
|
||||
list_meta_t *list = (void *) ((unsigned char *) list_ptr - sizeof(list_meta_t));
|
||||
unsigned char *array = list_ptr;
|
||||
|
||||
for (int i = 0; i < list->size; i++) {
|
||||
if (memcmp(array + i * list->elem_size, elem, list->elem_size) == 0) {
|
||||
return i;
|
||||
}
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
void *list_insert(void *list_ptr, void *elem, int n) {
|
||||
void *ptr = NULL;
|
||||
list_ptr = list_insert_ptr(list_ptr, &ptr, n);
|
||||
if (list_ptr != NULL && ptr != NULL) {
|
||||
list_meta_t *list = (void *) ((unsigned char *) list_ptr - sizeof(list_meta_t));
|
||||
memcpy(ptr, elem, list->elem_size);
|
||||
}
|
||||
|
||||
return list_ptr;
|
||||
}
|
||||
|
||||
void *list_insert_ptr(void *list_ptr, void **elem, int n) {
|
||||
list_meta_t *list = (void *) ((unsigned char *) list_ptr - sizeof(list_meta_t));
|
||||
if (n < 0)
|
||||
n = list->size + n + 1;
|
||||
|
||||
if (list->size >= list->max_size) {
|
||||
if ((list = list_resize(list, list->max_size * FACTOR)) == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
unsigned char *array = (unsigned char *) list + sizeof(list_meta_t);
|
||||
|
||||
if (n < list->size)
|
||||
memmove(array + (n + 1) * list->elem_size, array + n * list->elem_size, (list->size - n) * list->elem_size);
|
||||
*elem = array + n * list->elem_size;
|
||||
|
||||
list->size++;
|
||||
return (unsigned char *) list + sizeof(list_meta_t);
|
||||
}
|
||||
|
||||
void *list_append(void *list_ptr, void *elem) {
|
||||
return list_insert(list_ptr, elem, -1);
|
||||
}
|
||||
|
||||
void *list_append_ptr(void *list_ptr, void **elem) {
|
||||
return list_insert_ptr(list_ptr, elem, -1);
|
||||
}
|
||||
|
||||
void *list_remove(void *list_ptr, int n) {
|
||||
list_meta_t *list = (void *) ((unsigned char *) list_ptr - sizeof(list_meta_t));
|
||||
if (n < 0)
|
||||
n = list->size + n;
|
||||
|
||||
unsigned char *array = list_ptr;
|
||||
|
||||
if (list->size > 1 && n < list->size)
|
||||
memmove(array + n * list->elem_size, array + (n + 1) * list->elem_size, (list->size - n - 1) * list->elem_size);
|
||||
|
||||
list->size--;
|
||||
if (list->size < list->max_size / FACTOR / 2 && list->max_size / FACTOR >= list->init_size) {
|
||||
if ((list = list_resize(list, list->max_size / FACTOR)) == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
return (unsigned char *) list + sizeof(list_meta_t);
|
||||
}
|
||||
|
||||
void *list_delete(void *list_ptr, void *elem) {
|
||||
int idx = list_find(list_ptr, elem);
|
||||
if (idx == -1) {
|
||||
return list_ptr;
|
||||
} else {
|
||||
return list_remove(list_ptr, idx);
|
||||
}
|
||||
}
|
||||
|
||||
void *list_clear(void *list_ptr) {
|
||||
list_meta_t *list = (void *) ((unsigned char *) list_ptr - sizeof(list_meta_t));
|
||||
list->size = 0;
|
||||
memset(list_ptr, 0, list->max_size * list->elem_size);
|
||||
list->max_size = list->init_size;
|
||||
return (unsigned char *) list_resize(list, list->max_size * list->elem_size) + sizeof(list_meta_t);
|
||||
}
|
||||
|
||||
void list_free(void *list_ptr) {
|
||||
list_meta_t *list = (void *) ((unsigned char *) list_ptr - sizeof(list_meta_t));
|
||||
free(list);
|
||||
}
|
27
src/lib/list.h
Normal file
27
src/lib/list.h
Normal file
@ -0,0 +1,27 @@
|
||||
|
||||
#ifndef SESIMOS_LIST_H
|
||||
#define SESIMOS_LIST_H
|
||||
|
||||
void *list_create(int elem_size, int init_elem_n);
|
||||
|
||||
int list_size(const void *list_ptr);
|
||||
|
||||
int list_find(void *list_ptr, void *elem);
|
||||
|
||||
void *list_insert(void *list_ptr, void *elem, int n);
|
||||
|
||||
void *list_insert_ptr(void *list_ptr, void **elem, int n);
|
||||
|
||||
void *list_append(void *list_ptr, void *elem);
|
||||
|
||||
void *list_append_ptr(void *list_ptr, void **elem);
|
||||
|
||||
void *list_remove(void *list_ptr, int n);
|
||||
|
||||
void *list_delete(void *list_ptr, void *elem);
|
||||
|
||||
void *list_clear(void *list_ptr);
|
||||
|
||||
void list_free(void *list_ptr);
|
||||
|
||||
#endif //SESIMOS_LIST_H
|
168
src/lib/mpmc.c
Normal file
168
src/lib/mpmc.c
Normal file
@ -0,0 +1,168 @@
|
||||
|
||||
#include "mpmc.h"
|
||||
#include "../logger.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <malloc.h>
|
||||
#include <memory.h>
|
||||
#include <pthread.h>
|
||||
#include <signal.h>
|
||||
|
||||
static void *mpmc_worker(void *arg);
|
||||
|
||||
int mpmc_init(mpmc_t *ctx, int n_workers, int buf_size, void (*consumer)(void *obj), const char *name) {
|
||||
ctx->alive = 1;
|
||||
ctx->n_workers = n_workers;
|
||||
ctx->size = buf_size, ctx->max_size = buf_size;
|
||||
ctx->rd = 0, ctx->wr = 0;
|
||||
ctx->buffer = NULL, ctx->workers = NULL;
|
||||
ctx->consumer = consumer;
|
||||
ctx->name = name;
|
||||
|
||||
if (sem_init(&ctx->free, 0, ctx->size) != 0 ||
|
||||
sem_init(&ctx->used, 0, 0) != 0 ||
|
||||
sem_init(&ctx->lck_rd, 0, 1) != 0 ||
|
||||
sem_init(&ctx->lck_wr, 0, 1) != 0)
|
||||
{
|
||||
mpmc_destroy(ctx);
|
||||
return -1;
|
||||
}
|
||||
|
||||
if ((ctx->buffer = malloc(ctx->size * sizeof(void *))) == NULL ||
|
||||
(ctx->workers = malloc(ctx->n_workers * sizeof(pthread_t))) == NULL)
|
||||
{
|
||||
mpmc_destroy(ctx);
|
||||
return -1;
|
||||
}
|
||||
|
||||
memset(ctx->buffer, 0, ctx->size * sizeof(void *));
|
||||
memset(ctx->workers, -1, ctx->n_workers * sizeof(pthread_t));
|
||||
|
||||
for (int i = 0; i < ctx->n_workers; i++) {
|
||||
int ret;
|
||||
if ((ret = pthread_create(&ctx->workers[i], NULL, mpmc_worker, ctx)) != 0) {
|
||||
mpmc_destroy(ctx);
|
||||
errno = ret;
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int mpmc_queue(mpmc_t *ctx, void *obj) {
|
||||
// wait for buffer to be emptied
|
||||
try_again_1:
|
||||
if (sem_wait(&ctx->free) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
goto try_again_1;
|
||||
} else {
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
// lock wr field
|
||||
try_again_2:
|
||||
if (sem_wait(&ctx->lck_wr) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
goto try_again_2;
|
||||
} else {
|
||||
sem_post(&ctx->free);
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
int p = ctx->wr;
|
||||
ctx->wr = (ctx->wr + 1) % ctx->size;
|
||||
|
||||
// unlock wr field
|
||||
sem_post(&ctx->lck_wr);
|
||||
|
||||
// fill buffer with object
|
||||
ctx->buffer[p] = obj;
|
||||
|
||||
// inform worker
|
||||
sem_post(&ctx->used);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void *mpmc_worker(void *arg) {
|
||||
mpmc_t *ctx = arg;
|
||||
|
||||
int id;
|
||||
for (id = 0; id < ctx->n_workers && ctx->workers[id] != pthread_self(); id++);
|
||||
logger_set_name("%s/%i", ctx->name, id);
|
||||
|
||||
while (ctx->alive) {
|
||||
// wait for buffer to be filled
|
||||
if (sem_wait(&ctx->used) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
continue;
|
||||
} else {
|
||||
critical("Unable to lock semaphore");
|
||||
errno = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// lock rd field
|
||||
if (sem_wait(&ctx->lck_rd) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
sem_post(&ctx->used);
|
||||
continue;
|
||||
} else {
|
||||
critical("Unable to lock semaphore");
|
||||
errno = 0;
|
||||
sem_post(&ctx->used);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
int p = ctx->rd;
|
||||
ctx->rd = (ctx->rd + 1) % ctx->size;
|
||||
|
||||
// unlock rd field
|
||||
sem_post(&ctx->lck_rd);
|
||||
|
||||
// consume object
|
||||
ctx->consumer(ctx->buffer[p]);
|
||||
logger_set_prefix("");
|
||||
|
||||
// unlock slot in buffer
|
||||
sem_post(&ctx->free);
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void mpmc_stop(mpmc_t *ctx) {
|
||||
ctx->alive = 0;
|
||||
}
|
||||
|
||||
void mpmc_destroy(mpmc_t *ctx) {
|
||||
int e = errno;
|
||||
|
||||
// stop threads, if running
|
||||
mpmc_stop(ctx);
|
||||
for (int i = 0; i < ctx->n_workers; i++) {
|
||||
if (ctx->workers[i] == -1) break;
|
||||
pthread_kill(ctx->workers[i], SIGUSR1);
|
||||
pthread_join(ctx->workers[i], NULL);
|
||||
}
|
||||
|
||||
sem_destroy(&ctx->free);
|
||||
sem_destroy(&ctx->used);
|
||||
sem_destroy(&ctx->lck_rd);
|
||||
sem_destroy(&ctx->lck_wr);
|
||||
free(ctx->buffer);
|
||||
free(ctx->workers);
|
||||
|
||||
// reset errno
|
||||
errno = e;
|
||||
}
|
||||
|
27
src/lib/mpmc.h
Normal file
27
src/lib/mpmc.h
Normal file
@ -0,0 +1,27 @@
|
||||
|
||||
#ifndef SESIMOS_MPMC_H
|
||||
#define SESIMOS_MPMC_H
|
||||
|
||||
#include <semaphore.h>
|
||||
|
||||
typedef struct {
|
||||
unsigned char alive;
|
||||
int n_workers;
|
||||
int rd, wr;
|
||||
sem_t free, used, lck_rd, lck_wr;
|
||||
int size, max_size;
|
||||
void **buffer;
|
||||
pthread_t *workers;
|
||||
void (*consumer)(void *obj);
|
||||
const char* name;
|
||||
} mpmc_t;
|
||||
|
||||
int mpmc_init(mpmc_t *ctx, int n_workers, int buf_size, void (*consumer)(void *obj), const char *name);
|
||||
|
||||
int mpmc_queue(mpmc_t *ctx, void *obj);
|
||||
|
||||
void mpmc_stop(mpmc_t *ctx);
|
||||
|
||||
void mpmc_destroy(mpmc_t *ctx);
|
||||
|
||||
#endif //SESIMOS_MPMC_H
|
673
src/lib/proxy.c
Normal file
673
src/lib/proxy.c
Normal file
@ -0,0 +1,673 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Reverse proxy
|
||||
* @file src/lib/proxy.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-01-07
|
||||
*/
|
||||
|
||||
#include "../defs.h"
|
||||
#include "../server.h"
|
||||
#include "../logger.h"
|
||||
#include "proxy.h"
|
||||
#include "utils.h"
|
||||
#include "compress.h"
|
||||
#include "config.h"
|
||||
#include "error.h"
|
||||
|
||||
#include <openssl/ssl.h>
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
#include <openssl/err.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <netdb.h>
|
||||
#include <semaphore.h>
|
||||
|
||||
static SSL_CTX *proxy_ctx = NULL;
|
||||
static proxy_ctx_t *proxies = NULL;
|
||||
static sem_t *available = NULL;
|
||||
static sem_t lock;
|
||||
static int num_proxy_hosts = -1;
|
||||
|
||||
int proxy_preload(void) {
|
||||
int n = 0;
|
||||
for (int i = 0; i < CONFIG_MAX_HOST_CONFIG; i++) {
|
||||
host_config_t *hc = &config.hosts[i];
|
||||
if (hc->type == CONFIG_TYPE_UNSET) break;
|
||||
if (hc->type != CONFIG_TYPE_REVERSE_PROXY) continue;
|
||||
n++;
|
||||
}
|
||||
|
||||
proxy_ctx = SSL_CTX_new(TLS_client_method());
|
||||
if (proxy_ctx == NULL) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
proxies = malloc(n * MAX_PROXY_CNX_PER_HOST * sizeof(proxy_ctx_t));
|
||||
if (proxies == NULL) {
|
||||
proxy_unload();
|
||||
return -1;
|
||||
}
|
||||
memset(proxies, 0, n * MAX_PROXY_CNX_PER_HOST * sizeof(proxy_ctx_t));
|
||||
|
||||
available = malloc(n * sizeof(*available));
|
||||
if (available == NULL) {
|
||||
proxy_unload();
|
||||
return -1;
|
||||
}
|
||||
for (int i = 0; i < n; i++) {
|
||||
if (sem_init(&available[i], 0, MAX_PROXY_CNX_PER_HOST) != 0) {
|
||||
proxy_unload();
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
if (sem_init(&lock, 0, 1) != 0) {
|
||||
proxy_unload();
|
||||
return -1;
|
||||
}
|
||||
|
||||
num_proxy_hosts = n;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void proxy_unload(void) {
|
||||
int e = errno;
|
||||
SSL_CTX_free(proxy_ctx);
|
||||
sem_destroy(&lock);
|
||||
if (num_proxy_hosts != -1) {
|
||||
for (int i = 0; i < num_proxy_hosts; i++) {
|
||||
sem_destroy(&available[i]);
|
||||
}
|
||||
}
|
||||
free(available);
|
||||
free(proxies);
|
||||
errno = e;
|
||||
}
|
||||
|
||||
void proxy_close_all(void) {
|
||||
proxy_ctx_t *ptr = proxies;
|
||||
for (int i = 0; i < MAX_PROXY_CNX_PER_HOST * num_proxy_hosts; i++, ptr++) {
|
||||
if (ptr->initialized)
|
||||
proxy_close(ptr);
|
||||
}
|
||||
}
|
||||
|
||||
proxy_ctx_t *proxy_get_by_conf(host_config_t *conf) {
|
||||
int n = 0;
|
||||
for (int i = 0; i < CONFIG_MAX_HOST_CONFIG; i++) {
|
||||
host_config_t *hc = &config.hosts[i];
|
||||
if (hc->type == CONFIG_TYPE_UNSET) break;
|
||||
if (hc->type != CONFIG_TYPE_REVERSE_PROXY) continue;
|
||||
if (hc == conf) break;
|
||||
n++;
|
||||
}
|
||||
|
||||
try_again_1:
|
||||
if (sem_wait(&available[n]) != 0) {
|
||||
if (errno == EINTR) {
|
||||
goto try_again_1;
|
||||
} else {
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
try_again_2:
|
||||
if (sem_wait(&lock) != 0) {
|
||||
if (errno == EINTR) {
|
||||
goto try_again_2;
|
||||
} else {
|
||||
sem_post(&available[n]);
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
proxy_ctx_t *ptr = proxies + n * MAX_PROXY_CNX_PER_HOST;
|
||||
for (int i = 0; i < MAX_PROXY_CNX_PER_HOST; i++, ptr++) {
|
||||
if (!ptr->in_use) {
|
||||
ptr->in_use = 1;
|
||||
sem_post(&lock);
|
||||
return ptr;
|
||||
}
|
||||
}
|
||||
|
||||
sem_post(&lock);
|
||||
sem_post(&available[n]);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void proxy_unlock_ctx(proxy_ctx_t *ctx) {
|
||||
int n = (int) ((ctx - proxies) / MAX_PROXY_CNX_PER_HOST);
|
||||
ctx->in_use = 0;
|
||||
sem_post(&available[n]);
|
||||
}
|
||||
|
||||
int proxy_request_header(http_req *req, sock *sock) {
|
||||
char buf1[256], buf2[256];
|
||||
int p_len;
|
||||
|
||||
const char *via = http_get_header_field(&req->hdr, "Via");
|
||||
sprintf(buf1, "HTTP/%s %s", req->version, SERVER_NAME);
|
||||
if (via == NULL) {
|
||||
http_add_header_field(&req->hdr, "Via", buf1);
|
||||
} else {
|
||||
p_len = snprintf(buf2, sizeof(buf2), "%s, %s", via, buf1);
|
||||
if (p_len < 0 || p_len >= sizeof(buf2)) {
|
||||
error("Header field 'Via' too long");
|
||||
return -1;
|
||||
}
|
||||
http_remove_header_field(&req->hdr, "Via", HTTP_REMOVE_ALL);
|
||||
http_add_header_field(&req->hdr, "Via", buf2);
|
||||
}
|
||||
|
||||
const char *host = http_get_header_field(&req->hdr, "Host");
|
||||
const char *forwarded = http_get_header_field(&req->hdr, "Forwarded");
|
||||
int client_ipv6 = strchr(sock->addr, ':') != NULL;
|
||||
int server_ipv6 = strchr(sock->s_addr, ':') != NULL;
|
||||
|
||||
p_len = snprintf(buf1, sizeof(buf1), "by=%s%s%s;for=%s%s%s;host=%s;proto=%s",
|
||||
server_ipv6 ? "\"[" : "", sock->s_addr, server_ipv6 ? "]\"" : "",
|
||||
client_ipv6 ? "\"[" : "", sock->addr, client_ipv6 ? "]\"" : "",
|
||||
host, sock->enc ? "https" : "http");
|
||||
if (p_len < 0 || p_len >= sizeof(buf1)) {
|
||||
error("Appended part of header field 'Forwarded' too long");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (forwarded == NULL) {
|
||||
http_add_header_field(&req->hdr, "Forwarded", buf1);
|
||||
} else {
|
||||
p_len = snprintf(buf2, sizeof(buf2), "%s, %s", forwarded, buf1);
|
||||
if (p_len < 0 || p_len >= sizeof(buf2)) {
|
||||
error("Header field 'Forwarded' too long");
|
||||
return -1;
|
||||
}
|
||||
http_remove_header_field(&req->hdr, "Forwarded", HTTP_REMOVE_ALL);
|
||||
http_add_header_field(&req->hdr, "Forwarded", buf2);
|
||||
}
|
||||
|
||||
const char *xff = http_get_header_field(&req->hdr, "X-Forwarded-For");
|
||||
if (xff == NULL) {
|
||||
http_add_header_field(&req->hdr, "X-Forwarded-For", sock->addr);
|
||||
} else {
|
||||
sprintf(buf1, "%s, %s", xff, sock->addr);
|
||||
http_remove_header_field(&req->hdr, "X-Forwarded-For", HTTP_REMOVE_ALL);
|
||||
http_add_header_field(&req->hdr, "X-Forwarded-For", buf1);
|
||||
}
|
||||
|
||||
const char *xfh = http_get_header_field(&req->hdr, "X-Forwarded-Host");
|
||||
if (xfh == NULL) {
|
||||
if (forwarded == NULL) {
|
||||
http_add_header_field(&req->hdr, "X-Forwarded-Host", host);
|
||||
} else {
|
||||
char *ptr = strchr(forwarded, ',');
|
||||
unsigned long len;
|
||||
if (ptr != NULL) len = ptr - forwarded;
|
||||
else len = strlen(forwarded);
|
||||
ptr = strstr(forwarded, "host=");
|
||||
if ((ptr - forwarded) < len) {
|
||||
char *end = strchr(ptr, ';');
|
||||
if (end == NULL) len -= (ptr - forwarded);
|
||||
else len = (end - ptr);
|
||||
len -= 5;
|
||||
sprintf(buf1, "%.*s", (int) len, ptr + 5);
|
||||
http_add_header_field(&req->hdr, "X-Forwarded-Host", buf1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
const char *xfp = http_get_header_field(&req->hdr, "X-Forwarded-Proto");
|
||||
if (xfp == NULL) {
|
||||
if (forwarded == NULL) {
|
||||
http_add_header_field(&req->hdr, "X-Forwarded-Proto", sock->enc ? "https" : "http");
|
||||
} else {
|
||||
char *ptr = strchr(forwarded, ',');
|
||||
unsigned long len;
|
||||
if (ptr != NULL) len = ptr - forwarded;
|
||||
else len = strlen(forwarded);
|
||||
ptr = strstr(forwarded, "proto=");
|
||||
if ((ptr - forwarded) < len) {
|
||||
char *end = strchr(ptr, ';');
|
||||
if (end == NULL) len -= (ptr - forwarded);
|
||||
else len = (end - ptr);
|
||||
len -= 6;
|
||||
sprintf(buf1, "%.*s", (int) len, ptr + 6);
|
||||
http_add_header_field(&req->hdr, "X-Forwarded-Proto", buf1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int proxy_response_header(http_req *req, http_res *res, host_config_t *conf) {
|
||||
char buf1[256], buf2[256];
|
||||
int p_len;
|
||||
|
||||
const char *via = http_get_header_field(&res->hdr, "Via");
|
||||
p_len = snprintf(buf1, sizeof(buf1), "HTTP/%s %s", req->version, SERVER_NAME);
|
||||
if (p_len < 0 || p_len >= sizeof(buf1)) {
|
||||
error("Appended part of header field 'Via' too long");
|
||||
return -1;
|
||||
}
|
||||
if (via == NULL) {
|
||||
http_add_header_field(&res->hdr, "Via", buf1);
|
||||
} else {
|
||||
p_len = snprintf(buf2, sizeof(buf2), "%s, %s", via, buf1);
|
||||
if (p_len < 0 || p_len >= sizeof(buf2)) {
|
||||
error("Header field 'Via' too long");
|
||||
return -1;
|
||||
}
|
||||
http_remove_header_field(&res->hdr, "Via", HTTP_REMOVE_ALL);
|
||||
http_add_header_field(&res->hdr, "Via", buf2);
|
||||
}
|
||||
|
||||
const char *location = http_get_header_field(&res->hdr, "Location");
|
||||
if (location != NULL) {
|
||||
char *hostnames[] = {conf->name, conf->proxy.hostname};
|
||||
for (int i = 0; i < sizeof(hostnames) / sizeof(hostnames[0]); i++) {
|
||||
char *hostname = hostnames[i];
|
||||
|
||||
p_len = snprintf(buf1, sizeof(buf1), "http://%s/", hostname);
|
||||
if (strncmp(location, buf1, p_len) == 0) goto match;
|
||||
|
||||
p_len = snprintf(buf1, sizeof(buf1), "https://%s/", hostname);
|
||||
if (strncmp(location, buf1, p_len) == 0) goto match;
|
||||
|
||||
p_len = snprintf(buf1, sizeof(buf1), "http://%s:%i/", hostname, conf->proxy.port);
|
||||
if (strncmp(location, buf1, p_len) == 0) goto match;
|
||||
|
||||
p_len = snprintf(buf1, sizeof(buf1), "https://%s:%i/", hostname, conf->proxy.port);
|
||||
if (strncmp(location, buf1, p_len) == 0) goto match;
|
||||
}
|
||||
|
||||
if (0) {
|
||||
match:
|
||||
strcpy(buf1, location + p_len - 1);
|
||||
http_remove_header_field(&res->hdr, "Location", HTTP_REMOVE_ALL);
|
||||
http_add_header_field(&res->hdr, "Location", buf1);
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int proxy_init(proxy_ctx_t **proxy_ptr, http_req *req, http_res *res, http_status_ctx *ctx, host_config_t *conf, sock *client, http_status *custom_status, char *err_msg) {
|
||||
char buffer[CHUNK_SIZE], err_buf[256];
|
||||
const char *connection, *upgrade, *ws_version;
|
||||
long ret;
|
||||
int tries = 0, retry = 0;
|
||||
|
||||
*proxy_ptr = proxy_get_by_conf(conf);
|
||||
proxy_ctx_t *proxy = *proxy_ptr;
|
||||
proxy->client = NULL;
|
||||
|
||||
if (proxy->initialized && sock_has_pending(&proxy->proxy) == 0)
|
||||
goto proxy;
|
||||
|
||||
retry:
|
||||
if (proxy->initialized) {
|
||||
info(BLUE_STR "Closing proxy connection");
|
||||
sock_close(&proxy->proxy);
|
||||
proxy->initialized = 0;
|
||||
}
|
||||
retry = 0;
|
||||
tries++;
|
||||
|
||||
proxy->proxy.socket = socket(AF_INET6, SOCK_STREAM, 0);
|
||||
if (proxy->proxy.socket < 0) {
|
||||
error("Unable to create socket");
|
||||
res->status = http_get_status(500);
|
||||
ctx->origin = INTERNAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (sock_set_timeout(&proxy->proxy, SERVER_TIMEOUT_INIT) != 0)
|
||||
goto proxy_timeout_err;
|
||||
|
||||
struct hostent *host_ent = gethostbyname2(conf->proxy.hostname, AF_INET6);
|
||||
if (host_ent == NULL) {
|
||||
host_ent = gethostbyname2(conf->proxy.hostname, AF_INET);
|
||||
if (host_ent == NULL) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_REQ;
|
||||
error("Unable to connect to server: Name or service not known");
|
||||
sprintf(err_msg, "Unable to connect to server: Name or service not known.");
|
||||
goto proxy_err;
|
||||
}
|
||||
}
|
||||
|
||||
struct sockaddr_in6 address = {.sin6_family = AF_INET6, .sin6_port = htons(conf->proxy.port)};
|
||||
if (host_ent->h_addrtype == AF_INET6) {
|
||||
memcpy(&address.sin6_addr, host_ent->h_addr_list[0], host_ent->h_length);
|
||||
} else if (host_ent->h_addrtype == AF_INET) {
|
||||
unsigned char addr[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xFF, 0xFF, 0, 0, 0, 0};
|
||||
memcpy(addr + 12, host_ent->h_addr_list[0], host_ent->h_length);
|
||||
memcpy(&address.sin6_addr, addr, 16);
|
||||
}
|
||||
|
||||
inet_ntop(address.sin6_family, (void *) &address.sin6_addr, buffer, sizeof(buffer));
|
||||
|
||||
info(BLUE_STR "Connecting to " BLD_STR "[%s]:%i" CLR_STR BLUE_STR "...", buffer, conf->proxy.port);
|
||||
if (connect(proxy->proxy.socket, (struct sockaddr *) &address, sizeof(address)) < 0) {
|
||||
if (errno == ETIMEDOUT || errno == EINPROGRESS) {
|
||||
res->status = http_get_status(504);
|
||||
ctx->origin = SERVER_REQ;
|
||||
} else if (errno == ECONNREFUSED) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_REQ;
|
||||
} else {
|
||||
res->status = http_get_status(500);
|
||||
ctx->origin = INTERNAL;
|
||||
}
|
||||
error("Unable to connect to [%s]:%i", buffer, conf->proxy.port);
|
||||
sprintf(err_msg, "Unable to connect to server: %s.", error_str(errno, err_buf, sizeof(err_buf)));
|
||||
goto proxy_err;
|
||||
}
|
||||
|
||||
if (sock_set_timeout(&proxy->proxy, SERVER_TIMEOUT) != 0) {
|
||||
proxy_timeout_err:
|
||||
res->status = http_get_status(500);
|
||||
ctx->origin = INTERNAL;
|
||||
error("Unable to set timeout for reverse proxy socket");
|
||||
sprintf(err_msg, "Unable to set timeout for reverse proxy socket: %s", error_str(errno, err_buf, sizeof(err_buf)));
|
||||
goto proxy_err;
|
||||
}
|
||||
|
||||
if (conf->proxy.enc) {
|
||||
proxy->proxy.ssl = SSL_new(proxy_ctx);
|
||||
SSL_set_fd(proxy->proxy.ssl, proxy->proxy.socket);
|
||||
SSL_set_connect_state(proxy->proxy.ssl);
|
||||
|
||||
ret = SSL_do_handshake(proxy->proxy.ssl);
|
||||
if (ret != 1) {
|
||||
error_ssl(SSL_get_error(proxy->proxy.ssl, (int) ret));
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_REQ;
|
||||
error("Unable to perform handshake");
|
||||
sprintf(err_msg, "Unable to perform handshake: %s.", error_str(errno, err_buf, sizeof(err_buf)));
|
||||
goto proxy_err;
|
||||
}
|
||||
proxy->proxy.enc = 1;
|
||||
}
|
||||
|
||||
proxy->initialized = 1;
|
||||
proxy->host = conf->name;
|
||||
info(BLUE_STR "Established new connection with " BLD_STR "[%s]:%i", buffer, conf->proxy.port);
|
||||
|
||||
proxy:
|
||||
connection = http_get_header_field(&req->hdr, "Connection");
|
||||
if (connection != NULL && (strstr(connection, "upgrade") != NULL || strstr(connection, "Upgrade") != NULL)) {
|
||||
upgrade = http_get_header_field(&req->hdr, "Upgrade");
|
||||
ws_version = http_get_header_field(&req->hdr, "Sec-WebSocket-Version");
|
||||
if (upgrade != NULL && ws_version != NULL && strcmp(upgrade, "websocket") == 0 && strcmp(ws_version, "13") == 0) {
|
||||
ctx->ws_key = http_get_header_field(&req->hdr, "Sec-WebSocket-Key");
|
||||
} else {
|
||||
res->status = http_get_status(501);
|
||||
ctx->origin = INTERNAL;
|
||||
return -1;
|
||||
}
|
||||
} else {
|
||||
http_remove_header_field(&req->hdr, "Connection", HTTP_REMOVE_ALL);
|
||||
http_add_header_field(&req->hdr, "Connection", "keep-alive");
|
||||
}
|
||||
|
||||
ret = proxy_request_header(req, client);
|
||||
if (ret != 0) {
|
||||
res->status = http_get_status(500);
|
||||
ctx->origin = INTERNAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
ret = http_send_request(&proxy->proxy, req);
|
||||
if (ret < 0) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_REQ;
|
||||
error("Unable to send request to server (1)");
|
||||
sprintf(err_msg, "Unable to send request to server: %s.", error_str(errno, err_buf, sizeof(err_buf)));
|
||||
retry = tries < 4;
|
||||
goto proxy_err;
|
||||
}
|
||||
|
||||
const char *content_length = http_get_header_field(&req->hdr, "Content-Length");
|
||||
unsigned long content_len = content_length != NULL ? strtoul(content_length, NULL, 10) : 0;
|
||||
const char *transfer_encoding = http_get_header_field(&req->hdr, "Transfer-Encoding");
|
||||
|
||||
ret = 0;
|
||||
if (content_len > 0) {
|
||||
ret = sock_splice(&proxy->proxy, client, buffer, sizeof(buffer), content_len);
|
||||
} else if (transfer_encoding != NULL && strstr(transfer_encoding, "chunked") != NULL) {
|
||||
ret = sock_splice_chunked(&proxy->proxy, client, buffer, sizeof(buffer));
|
||||
}
|
||||
|
||||
if (ret < 0 || (content_len != 0 && ret != content_len)) {
|
||||
if (ret == -1) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_REQ;
|
||||
error("Unable to send request to server (2)");
|
||||
sprintf(err_msg, "Unable to send request to server: %s.", error_str(errno, err_buf, sizeof(err_buf)));
|
||||
retry = tries < 4;
|
||||
goto proxy_err;
|
||||
} else if (ret == -2) {
|
||||
res->status = http_get_status(400);
|
||||
ctx->origin = CLIENT_REQ;
|
||||
error("Unable to receive request from client");
|
||||
sprintf(err_msg, "Unable to receive request from client: %s.", error_str(errno, err_buf, sizeof(err_buf)));
|
||||
return -1;
|
||||
}
|
||||
res->status = http_get_status(500);
|
||||
ctx->origin = INTERNAL;
|
||||
error("Unknown Error");
|
||||
return -1;
|
||||
}
|
||||
|
||||
ret = sock_recv(&proxy->proxy, buffer, sizeof(buffer), MSG_PEEK);
|
||||
if (ret <= 0) {
|
||||
int e_sys = error_get_sys(), e_ssl = error_get_ssl();
|
||||
if (e_sys == EAGAIN || e_sys == EINPROGRESS || e_ssl == SSL_ERROR_WANT_READ || e_ssl == SSL_ERROR_WANT_WRITE) {
|
||||
res->status = http_get_status(504);
|
||||
ctx->origin = SERVER_RES;
|
||||
} else {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_RES;
|
||||
}
|
||||
error("Unable to receive response from server");
|
||||
sprintf(err_msg, "Unable to receive response from server: %s.", error_str(errno, err_buf, sizeof(err_buf)));
|
||||
retry = tries < 4;
|
||||
goto proxy_err;
|
||||
}
|
||||
|
||||
char *buf = buffer;
|
||||
unsigned short header_len = (unsigned short) (strstr(buffer, "\r\n\r\n") - buffer + 4);
|
||||
|
||||
if (header_len <= 0) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_RES;
|
||||
error("Unable to parse header: End of header not found");
|
||||
sprintf(err_msg, "Unable to parser header: End of header not found.");
|
||||
goto proxy_err;
|
||||
}
|
||||
|
||||
for (int i = 0; i < header_len; i++) {
|
||||
if ((buf[i] >= 0x00 && buf[i] <= 0x1F && buf[i] != '\r' && buf[i] != '\n') || buf[i] == 0x7F) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_RES;
|
||||
error("Unable to parse header: Header contains illegal characters");
|
||||
sprintf(err_msg, "Unable to parse header: Header contains illegal characters.");
|
||||
goto proxy_err;
|
||||
}
|
||||
}
|
||||
|
||||
char *ptr = buf;
|
||||
while (header_len != (ptr - buf)) {
|
||||
char *pos0 = strstr(ptr, "\r\n");
|
||||
if (pos0 == NULL) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_RES;
|
||||
error("Unable to parse header: Invalid header format");
|
||||
sprintf(err_msg, "Unable to parse header: Invalid header format.");
|
||||
goto proxy_err;
|
||||
}
|
||||
if (ptr == buf) {
|
||||
if (strncmp(ptr, "HTTP/", 5) != 0) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_RES;
|
||||
error("Unable to parse header: Invalid header format");
|
||||
sprintf(err_msg, "Unable to parse header: Invalid header format.");
|
||||
goto proxy_err;
|
||||
}
|
||||
int status_code = (int) strtol(ptr + 9, NULL, 10);
|
||||
res->status = http_get_status(status_code);
|
||||
if (res->status == NULL && status_code >= 100 && status_code <= 999) {
|
||||
custom_status->code = status_code;
|
||||
custom_status->type = 0;
|
||||
snprintf(custom_status->msg, sizeof(custom_status->msg), "%.*s",
|
||||
(int) (strchr(ptr, '\r') - ptr - 13), ptr + 13);
|
||||
res->status = custom_status;
|
||||
} else if (res->status == NULL) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_RES;
|
||||
error("Unable to parse header: Invalid or unknown status code");
|
||||
sprintf(err_msg, "Unable to parse header: Invalid or unknown status code.");
|
||||
goto proxy_err;
|
||||
}
|
||||
} else {
|
||||
if (http_parse_header_field(&res->hdr, ptr, pos0, 0) != 0) {
|
||||
res->status = http_get_status(502);
|
||||
ctx->origin = SERVER_RES;
|
||||
error("Unable to parse header");
|
||||
sprintf(err_msg, "Unable to parse header.");
|
||||
goto proxy_err;
|
||||
}
|
||||
}
|
||||
if (pos0[2] == '\r' && pos0[3] == '\n') {
|
||||
break;
|
||||
}
|
||||
ptr = pos0 + 2;
|
||||
}
|
||||
sock_recv(&proxy->proxy, buffer, header_len, 0);
|
||||
|
||||
ret = proxy_response_header(req, res, conf);
|
||||
if (ret != 0) {
|
||||
res->status = http_get_status(500);
|
||||
ctx->origin = INTERNAL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
|
||||
proxy_err:
|
||||
errno = 0;
|
||||
if (retry) goto retry;
|
||||
return -1;
|
||||
}
|
||||
|
||||
int proxy_send(proxy_ctx_t *proxy, sock *client, unsigned long len_to_send, int flags) {
|
||||
char buffer[CHUNK_SIZE], comp_out[CHUNK_SIZE], buf[256], *ptr;
|
||||
long ret = 0, len, snd_len;
|
||||
int finish_comp = 0;
|
||||
|
||||
compress_ctx comp_ctx;
|
||||
if (flags & PROXY_COMPRESS_BR) {
|
||||
flags &= ~PROXY_COMPRESS_GZ;
|
||||
if (compress_init(&comp_ctx, COMPRESS_BR) != 0) {
|
||||
error("Unable to init brotli");
|
||||
flags &= ~PROXY_COMPRESS_BR;
|
||||
}
|
||||
} else if (flags & PROXY_COMPRESS_GZ) {
|
||||
flags &= ~PROXY_COMPRESS_BR;
|
||||
if (compress_init(&comp_ctx, COMPRESS_GZ) != 0) {
|
||||
error("Unable to init gzip");
|
||||
flags &= ~PROXY_COMPRESS_GZ;
|
||||
}
|
||||
}
|
||||
|
||||
do {
|
||||
snd_len = 0;
|
||||
if (flags & PROXY_CHUNKED) {
|
||||
ret = sock_get_chunk_header(&proxy->proxy);
|
||||
if (ret < 0) {
|
||||
if (ret == -1) {
|
||||
error("Unable to receive from server: Malformed chunk header");
|
||||
} else {
|
||||
error("Unable to receive from server");
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
len_to_send = ret;
|
||||
ret = 1;
|
||||
if (len_to_send == 0 && (flags & PROXY_COMPRESS)) {
|
||||
finish_comp = 1;
|
||||
len = 0;
|
||||
ptr = NULL;
|
||||
goto out;
|
||||
finish:
|
||||
compress_free(&comp_ctx);
|
||||
}
|
||||
}
|
||||
while (snd_len < len_to_send) {
|
||||
unsigned long avail_in, avail_out;
|
||||
ret = sock_recv(&proxy->proxy, buffer, CHUNK_SIZE < (len_to_send - snd_len) ? CHUNK_SIZE : len_to_send - snd_len, 0);
|
||||
if (ret <= 0) {
|
||||
error("Unable to receive from server");
|
||||
break;
|
||||
}
|
||||
len = ret;
|
||||
ptr = buffer;
|
||||
out:
|
||||
avail_in = len;
|
||||
char *next_in = ptr;
|
||||
do {
|
||||
long buf_len = len;
|
||||
if (flags & PROXY_COMPRESS) {
|
||||
avail_out = sizeof(comp_out);
|
||||
compress_compress(&comp_ctx, next_in + len - avail_in, &avail_in, comp_out, &avail_out, finish_comp);
|
||||
ptr = comp_out;
|
||||
buf_len = (int) (sizeof(comp_out) - avail_out);
|
||||
snd_len += (long) (len - avail_in);
|
||||
}
|
||||
if (buf_len != 0) {
|
||||
len = sprintf(buf, "%lX\r\n", buf_len);
|
||||
ret = 1;
|
||||
|
||||
if (flags & PROXY_CHUNKED) ret = sock_send(client, buf, len, 0);
|
||||
if (ret <= 0) goto err;
|
||||
|
||||
ret = sock_send(client, ptr, buf_len, 0);
|
||||
if (ret <= 0) goto err;
|
||||
if (!(flags & PROXY_COMPRESS)) snd_len += ret;
|
||||
|
||||
if (flags & PROXY_CHUNKED) ret = sock_send(client, "\r\n", 2, 0);
|
||||
if (ret <= 0) {
|
||||
err:
|
||||
error("Unable to send");
|
||||
break;
|
||||
}
|
||||
}
|
||||
} while ((flags & PROXY_COMPRESS) && (avail_in != 0 || avail_out != sizeof(comp_out)));
|
||||
if (ret <= 0) break;
|
||||
if (finish_comp) goto finish;
|
||||
}
|
||||
if (ret <= 0) break;
|
||||
if (flags & PROXY_CHUNKED) sock_recv(&proxy->proxy, buffer, 2, 0);
|
||||
} while ((flags & PROXY_CHUNKED) && len_to_send > 0);
|
||||
|
||||
if (ret <= 0) return -1;
|
||||
|
||||
if (flags & PROXY_CHUNKED) {
|
||||
ret = sock_send(client, "0\r\n\r\n", 5, 0);
|
||||
if (ret <= 0) {
|
||||
error("Unable to send");
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int proxy_dump(proxy_ctx_t *proxy, char *buf, long len) {
|
||||
sock_recv(&proxy->proxy, buf, len, 0);
|
||||
return 0;
|
||||
}
|
51
src/lib/proxy.h
Normal file
51
src/lib/proxy.h
Normal file
@ -0,0 +1,51 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Reverse proxy (header file)
|
||||
* @file src/lib/proxy.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-01-07
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_PROXY_H
|
||||
#define SESIMOS_PROXY_H
|
||||
|
||||
#define PROXY_CHUNKED 1
|
||||
#define PROXY_COMPRESS_GZ 2
|
||||
#define PROXY_COMPRESS_BR 4
|
||||
#define PROXY_COMPRESS 6
|
||||
|
||||
#ifndef SERVER_NAME
|
||||
# define SERVER_NAME "reverse proxy"
|
||||
#endif
|
||||
|
||||
#include "http.h"
|
||||
#include "config.h"
|
||||
|
||||
typedef struct {
|
||||
unsigned char initialized:1, in_use:1;
|
||||
sock proxy;
|
||||
char *host;
|
||||
void *client;
|
||||
} proxy_ctx_t;
|
||||
|
||||
int proxy_preload(void);
|
||||
|
||||
void proxy_unload(void);
|
||||
|
||||
void proxy_close_all(void);
|
||||
|
||||
proxy_ctx_t *proxy_get_by_conf(host_config_t *conf);
|
||||
|
||||
void proxy_unlock_ctx(proxy_ctx_t *ctx);
|
||||
|
||||
int proxy_request_header(http_req *req, sock *sock);
|
||||
|
||||
int proxy_response_header(http_req *req, http_res *res, host_config_t *conf);
|
||||
|
||||
int proxy_init(proxy_ctx_t **proxy, http_req *req, http_res *res, http_status_ctx *ctx, host_config_t *conf, sock *client, http_status *custom_status, char *err_msg);
|
||||
|
||||
int proxy_send(proxy_ctx_t *proxy, sock *client, unsigned long len_to_send, int flags);
|
||||
|
||||
int proxy_dump(proxy_ctx_t *proxy, char *buf, long len);
|
||||
|
||||
#endif //SESIMOS_PROXY_H
|
62
src/lib/res.h
Normal file
62
src/lib/res.h
Normal file
@ -0,0 +1,62 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief HTTP resources (header file)
|
||||
* @file src/lib/res.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-31
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_RES_H
|
||||
#define SESIMOS_RES_H
|
||||
|
||||
#define http_default_doc _binary_bin_res_default_txt_start
|
||||
#define http_default_doc_size ((unsigned int) (_binary_bin_res_default_txt_end - _binary_bin_res_default_txt_start) - 1)
|
||||
#define http_proxy_doc _binary_bin_res_proxy_txt_start
|
||||
#define http_proxy_doc_size ((unsigned int) (_binary_bin_res_proxy_txt_end - _binary_bin_res_proxy_txt_start) - 1)
|
||||
#define http_style_doc _binary_bin_res_style_txt_start
|
||||
#define http_style_doc_size ((unsigned int) (_binary_bin_res_style_txt_end - _binary_bin_res_style_txt_start) - 1)
|
||||
|
||||
#define http_icon_error _binary_bin_res_icon_error_txt_start
|
||||
#define http_icon_error_size ((unsigned int) (_binary_bin_res_icon_error_txt_end - _binary_bin_res_icon_error_txt_start) - 1)
|
||||
#define http_icon_info _binary_bin_res_icon_info_txt_start
|
||||
#define http_icon_info_size ((unsigned int) (_binary_bin_res_icon_info_txt_end - _binary_bin_res_icon_info_txt_start) - 1)
|
||||
#define http_icon_success _binary_bin_res_icon_success_txt_start
|
||||
#define http_icon_success_size ((unsigned int) (_binary_bin_res_icon_success_txt_end - _binary_bin_res_icon_success_txt_start) - 1)
|
||||
#define http_icon_warning _binary_bin_res_icon_warning_txt_start
|
||||
#define http_icon_warning_size ((unsigned int) (_binary_bin_res_icon_warning_txt_end - _binary_bin_res_icon_warning_txt_start) - 1)
|
||||
|
||||
#define http_icon_globe _binary_bin_res_globe_txt_start
|
||||
#define http_icon_globe_size ((unsigned int) (_binary_bin_res_globe_txt_end - _binary_bin_res_globe_txt_start) - 1)
|
||||
|
||||
typedef struct {
|
||||
const char *name;
|
||||
const char *type;
|
||||
const char *content;
|
||||
const unsigned int size;
|
||||
} res_t;
|
||||
|
||||
extern const char _binary_bin_res_default_txt_start[];
|
||||
extern const char _binary_bin_res_default_txt_end[];
|
||||
|
||||
extern const char _binary_bin_res_proxy_txt_start[];
|
||||
extern const char _binary_bin_res_proxy_txt_end[];
|
||||
|
||||
extern const char _binary_bin_res_style_txt_start[];
|
||||
extern const char _binary_bin_res_style_txt_end[];
|
||||
|
||||
extern const char _binary_bin_res_icon_error_txt_start[];
|
||||
extern const char _binary_bin_res_icon_error_txt_end[];
|
||||
|
||||
extern const char _binary_bin_res_icon_info_txt_start[];
|
||||
extern const char _binary_bin_res_icon_info_txt_end[];
|
||||
|
||||
extern const char _binary_bin_res_icon_success_txt_start[];
|
||||
extern const char _binary_bin_res_icon_success_txt_end[];
|
||||
|
||||
extern const char _binary_bin_res_icon_warning_txt_start[];
|
||||
extern const char _binary_bin_res_icon_warning_txt_end[];
|
||||
|
||||
extern const char _binary_bin_res_globe_txt_start[];
|
||||
extern const char _binary_bin_res_globe_txt_end[];
|
||||
|
||||
#endif //SESIMOS_RES_H
|
174
src/lib/sock.c
Normal file
174
src/lib/sock.c
Normal file
@ -0,0 +1,174 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Basic TCP and TLS socket
|
||||
* @file src/lib/sock.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-01-07
|
||||
*/
|
||||
|
||||
#include "sock.h"
|
||||
#include "utils.h"
|
||||
#include "error.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <openssl/ssl.h>
|
||||
#include <string.h>
|
||||
#include <sys/socket.h>
|
||||
#include <unistd.h>
|
||||
|
||||
int sock_set_socket_timeout_micros(sock *s, long recv_micros, long send_micros) {
|
||||
struct timeval recv_to = {.tv_sec = recv_micros / 1000000, .tv_usec = recv_micros % 1000000},
|
||||
send_to = {.tv_sec = send_micros / 1000000, .tv_usec = send_micros % 1000000};
|
||||
|
||||
if (setsockopt(s->socket, SOL_SOCKET, SO_RCVTIMEO, &recv_to, sizeof(recv_to)) != 0)
|
||||
return -1;
|
||||
|
||||
if (setsockopt(s->socket, SOL_SOCKET, SO_SNDTIMEO, &send_to, sizeof(send_to)) != 0)
|
||||
return -1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int sock_set_socket_timeout(sock *s, double sec) {
|
||||
return sock_set_socket_timeout_micros(s, (long) (sec * 1000000L), (long) (sec * 1000000L));
|
||||
}
|
||||
|
||||
int sock_set_timeout_micros(sock *s, long micros) {
|
||||
if (micros < 0)
|
||||
return -1;
|
||||
|
||||
s->timeout_us = micros;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int sock_set_timeout(sock *s, double sec) {
|
||||
return sock_set_timeout_micros(s, (long) (sec * 1000000));
|
||||
}
|
||||
|
||||
long sock_send(sock *s, void *buf, unsigned long len, int flags) {
|
||||
long ret;
|
||||
if (s->enc) {
|
||||
ret = SSL_write(s->ssl, buf, (int) len);
|
||||
if (ret <= 0) error_ssl(SSL_get_error(s->ssl, (int) ret));
|
||||
} else {
|
||||
ret = send(s->socket, buf, len, flags);
|
||||
}
|
||||
|
||||
if (ret >= 0) {
|
||||
s->ts_last = clock_micros();
|
||||
return ret;
|
||||
} else {
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
long sock_recv(sock *s, void *buf, unsigned long len, int flags) {
|
||||
long ret;
|
||||
if (s->enc) {
|
||||
int (*func)(SSL*, void*, int) = (flags & MSG_PEEK) ? SSL_peek : SSL_read;
|
||||
ret = func(s->ssl, buf, (int) len);
|
||||
if (ret <= 0) error_ssl(SSL_get_error(s->ssl, (int) ret));
|
||||
} else {
|
||||
ret = recv(s->socket, buf, len, flags);
|
||||
}
|
||||
|
||||
if (ret >= 0) {
|
||||
s->ts_last = clock_micros();
|
||||
return ret;
|
||||
} else {
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
long sock_splice(sock *dst, sock *src, void *buf, unsigned long buf_len, unsigned long len) {
|
||||
long ret;
|
||||
unsigned long send_len = 0;
|
||||
unsigned long next_len;
|
||||
while (send_len < len) {
|
||||
next_len = (buf_len < (len - send_len)) ? buf_len : (len - send_len);
|
||||
ret = sock_recv(src, buf, next_len, 0);
|
||||
if (ret <= 0) return -2;
|
||||
next_len = ret;
|
||||
ret = sock_send(dst, buf, next_len, send_len + next_len < len ? MSG_MORE : 0);
|
||||
if (ret < 0) return -1;
|
||||
if (ret != next_len) return -3;
|
||||
send_len += next_len;
|
||||
}
|
||||
return (long) send_len;
|
||||
}
|
||||
|
||||
long sock_splice_chunked(sock *dst, sock *src, void *buf, unsigned long buf_len) {
|
||||
long ret;
|
||||
unsigned long send_len = 0;
|
||||
unsigned long next_len;
|
||||
|
||||
while (1) {
|
||||
ret = sock_get_chunk_header(src);
|
||||
if (ret < 0) return -2;
|
||||
|
||||
next_len = ret;
|
||||
if (next_len <= 0) break;
|
||||
|
||||
ret = sock_splice(dst, src, buf, buf_len, next_len);
|
||||
if (ret < 0) return ret;
|
||||
}
|
||||
|
||||
return (long) send_len;
|
||||
}
|
||||
|
||||
int sock_close(sock *s) {
|
||||
int e = errno;
|
||||
if (s->enc && s->ssl != NULL) {
|
||||
SSL_shutdown(s->ssl);
|
||||
SSL_free(s->ssl);
|
||||
s->ssl = NULL;
|
||||
}
|
||||
close(s->socket);
|
||||
s->socket = 0;
|
||||
s->enc = 0;
|
||||
errno = e;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int sock_has_pending(sock *s) {
|
||||
char buf[1];
|
||||
int e = errno;
|
||||
long ret = sock_recv(s, &buf, 1, MSG_PEEK | MSG_DONTWAIT);
|
||||
errno = e;
|
||||
return ret == 1;
|
||||
}
|
||||
|
||||
long sock_parse_chunk_header(const char *buf, long len, long *ret_len) {
|
||||
for (int i = 0; i < len; i++) {
|
||||
char ch = buf[i];
|
||||
if (ch == '\r') {
|
||||
continue;
|
||||
} else if (ch == '\n') {
|
||||
if (ret_len != NULL) *ret_len = i + 1;
|
||||
return strtol(buf, NULL, 16);
|
||||
} else if (!((ch >= '0' && ch <= '9') || (ch >= 'a' && ch <= 'f') || (ch >= 'A' && ch <= 'F'))) {
|
||||
return -2;
|
||||
}
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
long sock_get_chunk_header(sock *s) {
|
||||
long ret, len;
|
||||
char buf[16];
|
||||
|
||||
do {
|
||||
ret = sock_recv(s, buf, sizeof(buf), MSG_PEEK);
|
||||
if (ret <= 0) return -2;
|
||||
else if (ret < 2) continue;
|
||||
|
||||
ret = sock_parse_chunk_header(buf, ret, &len);
|
||||
if (ret == -2) return -1;
|
||||
} while (ret < 0);
|
||||
|
||||
if (sock_recv(s, buf, len, 0) != len)
|
||||
return -2;
|
||||
|
||||
return ret;
|
||||
}
|
53
src/lib/sock.h
Normal file
53
src/lib/sock.h
Normal file
@ -0,0 +1,53 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Basic TCP and TLS socket (header file)
|
||||
* @file src/lib/sock.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2021-01-07
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_SOCK_H
|
||||
#define SESIMOS_SOCK_H
|
||||
|
||||
#include <openssl/crypto.h>
|
||||
#include <sys/socket.h>
|
||||
#include <arpa/inet.h>
|
||||
|
||||
typedef struct {
|
||||
unsigned int enc:1;
|
||||
int socket;
|
||||
union {
|
||||
struct sockaddr sock;
|
||||
struct sockaddr_in6 ipv6;
|
||||
} _addr;
|
||||
char *addr, *s_addr;
|
||||
SSL_CTX *ctx;
|
||||
SSL *ssl;
|
||||
long ts_start, ts_last, timeout_us;
|
||||
} sock;
|
||||
|
||||
int sock_set_socket_timeout_micros(sock *s, long recv_micros, long send_micros);
|
||||
|
||||
int sock_set_socket_timeout(sock *s, double sec);
|
||||
|
||||
int sock_set_timeout_micros(sock *s, long micros);
|
||||
|
||||
int sock_set_timeout(sock *s, double sec);
|
||||
|
||||
long sock_send(sock *s, void *buf, unsigned long len, int flags);
|
||||
|
||||
long sock_recv(sock *s, void *buf, unsigned long len, int flags);
|
||||
|
||||
long sock_splice(sock *dst, sock *src, void *buf, unsigned long buf_len, unsigned long len);
|
||||
|
||||
long sock_splice_chunked(sock *dst, sock *src, void *buf, unsigned long buf_len);
|
||||
|
||||
int sock_close(sock *s);
|
||||
|
||||
int sock_has_pending(sock *s);
|
||||
|
||||
long sock_parse_chunk_header(const char *buf, long len, long *ret_len);
|
||||
|
||||
long sock_get_chunk_header(sock *s);
|
||||
|
||||
#endif //SESIMOS_SOCK_H
|
226
src/lib/uri.c
Normal file
226
src/lib/uri.c
Normal file
@ -0,0 +1,226 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief URI and path handlers
|
||||
* @file src/lib/uri.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-13
|
||||
*/
|
||||
|
||||
#include "uri.h"
|
||||
#include "utils.h"
|
||||
|
||||
#include <errno.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
|
||||
int path_is_directory(const char *path) {
|
||||
struct stat statbuf;
|
||||
return stat(path, &statbuf) == 0 && S_ISDIR(statbuf.st_mode) != 0;
|
||||
}
|
||||
|
||||
int path_is_file(const char *path) {
|
||||
struct stat statbuf;
|
||||
int ret = stat(path, &statbuf);
|
||||
errno = 0;
|
||||
return ret == 0 && S_ISDIR(statbuf.st_mode) == 0;
|
||||
}
|
||||
|
||||
int path_exists(const char *path) {
|
||||
struct stat statbuf;
|
||||
int ret = stat(path, &statbuf);
|
||||
errno = 0;
|
||||
return ret == 0;
|
||||
}
|
||||
|
||||
int uri_init(http_uri *uri, const char *webroot, const char *uri_str, int dir_mode) {
|
||||
char buf0[1024];
|
||||
char buf1[1024];
|
||||
char buf2[1024];
|
||||
char buf3[1024];
|
||||
char buf4[1024];
|
||||
int p_len;
|
||||
uri->webroot = NULL;
|
||||
uri->req_path = NULL;
|
||||
uri->path = NULL;
|
||||
uri->pathinfo = NULL;
|
||||
uri->query = NULL;
|
||||
uri->filename = NULL;
|
||||
uri->uri = NULL;
|
||||
uri->meta = NULL;
|
||||
uri->is_static = 1;
|
||||
uri->is_dir = 0;
|
||||
if (uri_str[0] != '/') {
|
||||
return 1;
|
||||
}
|
||||
uri->webroot = malloc(strlen(webroot) + 1);
|
||||
strcpy(uri->webroot, webroot);
|
||||
|
||||
char *query = strchr(uri_str, '?');
|
||||
if (query == NULL) {
|
||||
uri->query = NULL;
|
||||
} else {
|
||||
query[0] = 0;
|
||||
query++;
|
||||
long size = (long) strlen(query) + 1;
|
||||
uri->query = malloc(size);
|
||||
strcpy(uri->query, query);
|
||||
}
|
||||
|
||||
long size = (long) strlen(uri_str) + 1;
|
||||
uri->req_path = malloc(size);
|
||||
url_decode(uri_str, uri->req_path, &size);
|
||||
if (query != NULL) {
|
||||
query[-1] = '?';
|
||||
}
|
||||
if (strstr(uri->req_path, "/../") != NULL || strstr(uri->req_path, "/./") != NULL) {
|
||||
return 2;
|
||||
}
|
||||
|
||||
size = (long) strlen(uri->req_path) + 1;
|
||||
uri->path = malloc(size);
|
||||
uri->pathinfo = malloc(size);
|
||||
|
||||
char last = 0;
|
||||
for (int i = 0, j = 0; i < size - 1; i++) {
|
||||
char ch = uri->req_path[i];
|
||||
if (last != '/' || ch != '/') {
|
||||
uri->path[j++] = ch;
|
||||
uri->path[j] = 0;
|
||||
}
|
||||
last = ch;
|
||||
}
|
||||
|
||||
if (dir_mode == URI_DIR_MODE_NO_VALIDATION) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (uri->path[strlen(uri->path) - 1] == '/') {
|
||||
uri->path[strlen(uri->path) - 1] = 0;
|
||||
strcpy(uri->pathinfo, "/");
|
||||
} else {
|
||||
strcpy(uri->pathinfo, "");
|
||||
}
|
||||
|
||||
if (!path_exists(uri->webroot)) {
|
||||
return 3;
|
||||
}
|
||||
|
||||
while (1) {
|
||||
sprintf(buf0, "%s%s", uri->webroot, uri->path);
|
||||
p_len = snprintf(buf1, sizeof(buf1), "%s.ncr", buf0);
|
||||
if (p_len < 0 || p_len >= sizeof(buf1)) return -1;
|
||||
p_len = snprintf(buf2, sizeof(buf2), "%s.php", buf0);
|
||||
if (p_len < 0 || p_len >= sizeof(buf2)) return -1;
|
||||
p_len = snprintf(buf3, sizeof(buf3), "%s.html", buf0);
|
||||
if (p_len < 0 || p_len >= sizeof(buf3)) return -1;
|
||||
|
||||
if (strlen(uri->path) <= 1 ||
|
||||
path_exists(buf0) ||
|
||||
path_is_file(buf1) ||
|
||||
path_is_file(buf2) ||
|
||||
path_is_file(buf3)) {
|
||||
break;
|
||||
}
|
||||
|
||||
char *ptr;
|
||||
parent_dir:
|
||||
ptr = strrchr(uri->path, '/');
|
||||
size = (long) strlen(ptr);
|
||||
sprintf(buf4, "%.*s%s", (int) size, ptr, uri->pathinfo);
|
||||
strcpy(uri->pathinfo, buf4);
|
||||
ptr[0] = 0;
|
||||
}
|
||||
if (uri->pathinfo[0] != 0) {
|
||||
sprintf(buf4, "%s", uri->pathinfo + 1);
|
||||
strcpy(uri->pathinfo, buf4);
|
||||
}
|
||||
|
||||
if (path_is_file(buf0)) {
|
||||
uri->filename = malloc(strlen(buf0) + 1);
|
||||
strcpy(uri->filename, buf0);
|
||||
long len = (long) strlen(uri->path);
|
||||
if (strcmp(uri->path + len - 4, ".ncr") == 0 || strcmp(uri->path + len - 4, ".php") == 0) {
|
||||
uri->path[len - 4] = 0;
|
||||
uri->is_static = 0;
|
||||
} else if (strcmp(uri->path + len - 5, ".html") == 0) {
|
||||
uri->path[len - 5] = 0;
|
||||
}
|
||||
} else if (path_is_file(buf1)) {
|
||||
uri->is_static = 0;
|
||||
uri->filename = malloc(strlen(buf1) + 1);
|
||||
strcpy(uri->filename, buf1);
|
||||
} else if (path_is_file(buf2)) {
|
||||
uri->is_static = 0;
|
||||
uri->filename = malloc(strlen(buf2) + 1);
|
||||
strcpy(uri->filename, buf2);
|
||||
} else if (path_is_file(buf3)) {
|
||||
uri->filename = malloc(strlen(buf3) + 1);
|
||||
strcpy(uri->filename, buf3);
|
||||
} else {
|
||||
uri->is_dir = 1;
|
||||
strcpy(uri->path + strlen(uri->path), "/");
|
||||
sprintf(buf1, "%s%sindex.ncr", uri->webroot, uri->path);
|
||||
sprintf(buf2, "%s%sindex.php", uri->webroot, uri->path);
|
||||
sprintf(buf3, "%s%sindex.html", uri->webroot, uri->path);
|
||||
if (path_is_file(buf1)) {
|
||||
uri->filename = malloc(strlen(buf1) + 1);
|
||||
strcpy(uri->filename, buf1);
|
||||
uri->is_static = 0;
|
||||
} else if (path_is_file(buf2)) {
|
||||
uri->filename = malloc(strlen(buf2) + 1);
|
||||
strcpy(uri->filename, buf2);
|
||||
uri->is_static = 0;
|
||||
} else if (path_is_file(buf3)) {
|
||||
uri->filename = malloc(strlen(buf3) + 1);
|
||||
strcpy(uri->filename, buf3);
|
||||
} else {
|
||||
if (dir_mode == URI_DIR_MODE_FORBIDDEN) {
|
||||
uri->is_static = 1;
|
||||
} else if (dir_mode == URI_DIR_MODE_LIST) {
|
||||
uri->is_static = 0;
|
||||
} else if (dir_mode == URI_DIR_MODE_INFO) {
|
||||
if (strlen(uri->path) > 1) {
|
||||
uri->path[strlen(uri->path) - 1] = 0;
|
||||
sprintf(buf0, "/%s", uri->pathinfo);
|
||||
strcpy(uri->pathinfo, buf0);
|
||||
goto parent_dir;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (strcmp(uri->path + strlen(uri->path) - 5, "index") == 0) {
|
||||
uri->path[strlen(uri->path) - 5] = 0;
|
||||
}
|
||||
if (strcmp(uri->pathinfo, "index.ncr") == 0 ||
|
||||
strcmp(uri->pathinfo, "index.php") == 0 ||
|
||||
strcmp(uri->pathinfo, "index.html") == 0) {
|
||||
uri->pathinfo[0] = 0;
|
||||
}
|
||||
|
||||
sprintf(buf0, "%s%s%s%s%s", uri->path,
|
||||
(strlen(uri->pathinfo) == 0 || uri->path[strlen(uri->path) - 1] == '/') ? "" : "/", uri->pathinfo,
|
||||
uri->query != NULL ? "?" : "", uri->query != NULL ? uri->query : "");
|
||||
uri->uri = malloc(strlen(buf0) + 1);
|
||||
strcpy(uri->uri, buf0);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void uri_free(http_uri *uri) {
|
||||
if (uri->webroot != NULL) free(uri->webroot);
|
||||
if (uri->req_path != NULL) free(uri->req_path);
|
||||
if (uri->path != NULL) free(uri->path);
|
||||
if (uri->pathinfo != NULL) free(uri->pathinfo);
|
||||
if (uri->query != NULL) free(uri->query);
|
||||
if (uri->filename != NULL) free(uri->filename);
|
||||
if (uri->uri != NULL) free(uri->uri);
|
||||
uri->webroot = NULL;
|
||||
uri->req_path = NULL;
|
||||
uri->path = NULL;
|
||||
uri->pathinfo = NULL;
|
||||
uri->query = NULL;
|
||||
uri->filename = NULL;
|
||||
uri->uri = NULL;
|
||||
}
|
52
src/lib/uri.h
Normal file
52
src/lib/uri.h
Normal file
@ -0,0 +1,52 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief URI and path handlers (header file)
|
||||
* @file src/lib/uri.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-13
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_URI_H
|
||||
#define SESIMOS_URI_H
|
||||
|
||||
#include <sys/stat.h>
|
||||
|
||||
#define URI_DIR_MODE_NO_VALIDATION 0
|
||||
#define URI_DIR_MODE_FORBIDDEN 1
|
||||
#define URI_DIR_MODE_LIST 2
|
||||
#define URI_DIR_MODE_INFO 3
|
||||
|
||||
#define URI_ETAG_SIZE 65 // SHA256 size (hex)
|
||||
#define URI_TYPE_SIZE 64
|
||||
#define URI_CHARSET_SIZE 16
|
||||
|
||||
typedef struct {
|
||||
char etag[URI_ETAG_SIZE];
|
||||
char type[URI_TYPE_SIZE];
|
||||
char charset[URI_CHARSET_SIZE];
|
||||
char filename_comp_gz[256];
|
||||
char filename_comp_br[256];
|
||||
struct stat stat;
|
||||
} metadata_t;
|
||||
|
||||
typedef struct {
|
||||
char *webroot; // "/srv/www/www.test.org"
|
||||
char *req_path; // "/account/login"
|
||||
char *path; // "/account/"
|
||||
char *pathinfo; // "login"
|
||||
char *query; // "username=test"
|
||||
char *filename; // "/account/index.php"
|
||||
char *uri; // "/account/login?username=test"
|
||||
metadata_t *meta;
|
||||
unsigned int is_static:1;
|
||||
unsigned int is_dir:1;
|
||||
} http_uri;
|
||||
|
||||
|
||||
int uri_init(http_uri *uri, const char *webroot, const char *uri_str, int dir_mode);
|
||||
|
||||
int uri_init_cache(http_uri *uri);
|
||||
|
||||
void uri_free(http_uri *uri);
|
||||
|
||||
#endif //SESIMOS_URI_H
|
218
src/lib/utils.c
Normal file
218
src/lib/utils.c
Normal file
@ -0,0 +1,218 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Utilities
|
||||
* @file src/lib/utils.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-03
|
||||
*/
|
||||
|
||||
#include "utils.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <time.h>
|
||||
|
||||
|
||||
static const char base64_encode_table[64] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
||||
static const int base64_mod_table[3] = {0, 2, 1};
|
||||
|
||||
|
||||
char *format_duration(unsigned long micros, char *buf) {
|
||||
if (micros < 10000) {
|
||||
sprintf(buf, "%.1f ms", (double) micros / 1000);
|
||||
} else if (micros < 1000000 - 1000) {
|
||||
sprintf(buf, "%.0f ms", (double) micros / 1000);
|
||||
} else if (micros < 60000000 - 1000000) {
|
||||
sprintf(buf, "%.1f s", (double) micros / 1000000);
|
||||
} else if (micros < 6000000000) {
|
||||
sprintf(buf, "%li:%02li min", micros / 1000000 / 60, micros / 1000000 % 60);
|
||||
} else {
|
||||
sprintf(buf, "%.0f min", (double) micros / 1000000 / 60);
|
||||
}
|
||||
return buf;
|
||||
}
|
||||
|
||||
int url_encode_component(const void *in, size_t size_in, char *out, size_t size_out) {
|
||||
int size = 0;
|
||||
|
||||
// Encode control characters
|
||||
for (int i = 0; i < size_in; i++) {
|
||||
unsigned char ch = ((unsigned char *) in)[i];
|
||||
if (ch == ' ') {
|
||||
ch = '+';
|
||||
} else if (
|
||||
ch <= 0x20 || ch >= 0x7F ||
|
||||
!((ch >= 'A' && ch <= 'Z') || (ch >= 'a' && ch <= 'z') || (ch >= '0' && ch <= '9') ||
|
||||
ch == '-' || ch == '_' || ch == '.' || ch == '!' || ch == '~' || ch == '*' || ch == '\'' ||
|
||||
ch == '(' || ch == ')')
|
||||
) {
|
||||
size += 3;
|
||||
if (size < size_out) sprintf(out + size - 3, "%%%02X", ch);
|
||||
ch = 0;
|
||||
}
|
||||
|
||||
if (ch != 0) {
|
||||
size++;
|
||||
if (size < size_out) out[size - 1] = (char) ch;
|
||||
}
|
||||
}
|
||||
|
||||
// Set terminating null byte
|
||||
if (size_out > 0) out[size < size_out ? size : size_out - 1] = 0;
|
||||
|
||||
// Return theoretical size
|
||||
return size;
|
||||
}
|
||||
|
||||
int url_encode(const void *in, size_t size_in, char *out, size_t size_out) {
|
||||
int size = 0;
|
||||
|
||||
// Encode control characters
|
||||
for (int i = 0; i < size_in; i++) {
|
||||
unsigned char ch = ((unsigned char *) in)[i];
|
||||
if (ch <= 0x20 || ch >= 0x7F) {
|
||||
size += 3;
|
||||
if (size < size_out) sprintf(out + size - 3, "%%%02X", ch);
|
||||
} else {
|
||||
size++;
|
||||
if (size < size_out) out[size - 1] = (char) ch;
|
||||
}
|
||||
}
|
||||
|
||||
// Set terminating null byte
|
||||
if (size_out > 0) out[size < size_out ? size : size_out - 1] = 0;
|
||||
|
||||
// Return theoretical size
|
||||
return size;
|
||||
}
|
||||
|
||||
int url_decode(const char *str, char *dec, long *size) {
|
||||
char *ptr = dec;
|
||||
char ch, buf[3];
|
||||
memset(dec, 0, *size);
|
||||
for (int i = 0; i < strlen(str); i++, ptr++) {
|
||||
if ((ptr - dec) >= *size) {
|
||||
return -1;
|
||||
}
|
||||
ch = str[i];
|
||||
if (ch == '+') {
|
||||
ch = ' ';
|
||||
} else if (ch == '%') {
|
||||
memcpy(buf, str + i + 1, 2);
|
||||
buf[2] = 0;
|
||||
ch = (char) strtol(buf, NULL, 16);
|
||||
i += 2;
|
||||
} else if (ch == '?') {
|
||||
strcpy(ptr, str + i);
|
||||
break;
|
||||
}
|
||||
ptr[0] = ch;
|
||||
}
|
||||
*size = ptr - dec;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int mime_is_compressible(const char *type) {
|
||||
if (type == NULL) return 0;
|
||||
char type_parsed[64];
|
||||
snprintf(type_parsed, sizeof(type_parsed), "%s", type);
|
||||
char *pos = strchr(type_parsed, ';');
|
||||
if (pos != NULL) pos[0] = 0;
|
||||
return
|
||||
strncmp(type_parsed, "text/", 5) == 0 ||
|
||||
strncmp(type_parsed, "message/", 7) == 0 ||
|
||||
strstr(type_parsed, "+xml") != NULL ||
|
||||
strstr(type_parsed, "+json") != NULL ||
|
||||
strcmp(type_parsed, "application/javascript") == 0 ||
|
||||
strcmp(type_parsed, "application/json") == 0 ||
|
||||
strcmp(type_parsed, "application/xml") == 0 ||
|
||||
strcmp(type_parsed, "application/x-www-form-urlencoded") == 0 ||
|
||||
strcmp(type_parsed, "application/x-tex") == 0 ||
|
||||
strcmp(type_parsed, "application/x-httpd-php") == 0 ||
|
||||
strcmp(type_parsed, "application/x-latex") == 0 ||
|
||||
strcmp(type_parsed, "application/vnd.ms-fontobject") == 0 ||
|
||||
strcmp(type_parsed, "application/x-font-ttf") == 0 ||
|
||||
strcmp(type_parsed, "application/x-javascript") == 0 ||
|
||||
strcmp(type_parsed, "font/eot") == 0 ||
|
||||
strcmp(type_parsed, "font/opentype") == 0 ||
|
||||
strcmp(type_parsed, "image/bmp") == 0 ||
|
||||
strcmp(type_parsed, "image/gif") == 0 ||
|
||||
strcmp(type_parsed, "image/vnd.microsoft.icon") == 0 ||
|
||||
strcmp(type_parsed, "image/vnd.microsoft.iconbinary") == 0 ||
|
||||
strcmp(type_parsed, "image/x-icon") == 0;
|
||||
}
|
||||
|
||||
int strcpy_rem_webroot(char *dst, const char *src, long len, const char *webroot) {
|
||||
memcpy(dst, src, len);
|
||||
dst[len] = 0;
|
||||
if (webroot == NULL)
|
||||
return 0;
|
||||
|
||||
char *pos;
|
||||
const unsigned long webroot_len = strlen(webroot);
|
||||
if (webroot_len == 0)
|
||||
return 0;
|
||||
|
||||
while ((pos = strstr(dst, webroot)) != NULL) {
|
||||
strcpy(pos, pos + webroot_len);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int str_trim(char **start, char **end) {
|
||||
if (start == NULL || end == NULL || *start == NULL || *end == NULL)
|
||||
return -1;
|
||||
|
||||
(*end)--;
|
||||
while (*start[0] == ' ' || *start[0] == '\t' || *start[0] == '\r' || *start[0] == '\n') (*start)++;
|
||||
while (*end[0] == ' ' || *end[0] == '\t' || *end[0] == '\r' || *end[0] == '\n') (*end)--;
|
||||
(*end)++;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int str_trim_lws(char **start, char **end) {
|
||||
if (start == NULL || end == NULL || *start == NULL || *end == NULL)
|
||||
return -1;
|
||||
|
||||
(*end)--;
|
||||
while (*start[0] == ' ' || *start[0] == '\t') (*start)++;
|
||||
while (*end[0] == ' ' || *end[0] == '\t') (*end)--;
|
||||
(*end)++;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int base64_encode(void *data, unsigned long data_len, char *output, unsigned long *output_len) {
|
||||
unsigned long out_len = 4 * ((data_len + 2) / 3);
|
||||
if (output_len != NULL) *output_len = out_len;
|
||||
|
||||
for (int i = 0, j = 0; i < data_len;) {
|
||||
unsigned int octet_a = (i < data_len) ? ((unsigned char *) data)[i++] : 0;
|
||||
unsigned int octet_b = (i < data_len) ? ((unsigned char *) data)[i++] : 0;
|
||||
unsigned int octet_c = (i < data_len) ? ((unsigned char *) data)[i++] : 0;
|
||||
unsigned int triple = (octet_a << 0x10) + (octet_b << 0x08) + octet_c;
|
||||
output[j++] = base64_encode_table[(triple >> 3 * 6) & 0x3F];
|
||||
output[j++] = base64_encode_table[(triple >> 2 * 6) & 0x3F];
|
||||
output[j++] = base64_encode_table[(triple >> 1 * 6) & 0x3F];
|
||||
output[j++] = base64_encode_table[(triple >> 0 * 6) & 0x3F];
|
||||
}
|
||||
|
||||
for (int i = 0; i < base64_mod_table[data_len % 3]; i++)
|
||||
output[out_len - 1 - i] = '=';
|
||||
output[out_len] = 0;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
long clock_micros(void) {
|
||||
struct timespec time;
|
||||
clock_gettime(CLOCK_MONOTONIC, &time);
|
||||
return time.tv_sec * 1000000 + time.tv_nsec / 1000;
|
||||
}
|
||||
|
||||
long clock_cpu(void) {
|
||||
struct timespec time;
|
||||
clock_gettime(CLOCK_THREAD_CPUTIME_ID, &time);
|
||||
return time.tv_sec * 1000000000 + time.tv_nsec;
|
||||
}
|
44
src/lib/utils.h
Normal file
44
src/lib/utils.h
Normal file
@ -0,0 +1,44 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Utilities (header file)
|
||||
* @file src/lib/utils.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-03
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_UTILS_H
|
||||
#define SESIMOS_UTILS_H
|
||||
|
||||
#include <stdio.h>
|
||||
|
||||
#define ERR_STR "\x1B[1;31m"
|
||||
#define CLR_STR "\x1B[0m"
|
||||
#define BLD_STR "\x1B[1m"
|
||||
#define WRN_STR "\x1B[1;33m"
|
||||
#define BLUE_STR "\x1B[34m"
|
||||
#define HTTP_STR "\x1B[1;31m"
|
||||
#define HTTPS_STR "\x1B[1;32m"
|
||||
|
||||
char *format_duration(unsigned long micros, char *buf);
|
||||
|
||||
int url_encode_component(const void *in, size_t size_in, char *out, size_t size_out);
|
||||
|
||||
int url_encode(const void *in, size_t size_in, char *out, size_t size_out);
|
||||
|
||||
int url_decode(const char *str, char *dec, long *size);
|
||||
|
||||
int mime_is_compressible(const char *type);
|
||||
|
||||
int strcpy_rem_webroot(char *dst, const char *str, long len, const char *webroot);
|
||||
|
||||
int str_trim(char **start, char **end);
|
||||
|
||||
int str_trim_lws(char **start, char **end);
|
||||
|
||||
int base64_encode(void *data, unsigned long data_len, char *output, unsigned long *output_len);
|
||||
|
||||
long clock_micros(void);
|
||||
|
||||
long clock_cpu(void);
|
||||
|
||||
#endif //SESIMOS_UTILS_H
|
138
src/lib/websocket.c
Normal file
138
src/lib/websocket.c
Normal file
@ -0,0 +1,138 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief WebSocket reverse proxy
|
||||
* @file src/lib/websocket.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-08-16
|
||||
*/
|
||||
|
||||
#include "../logger.h"
|
||||
#include "websocket.h"
|
||||
#include "utils.h"
|
||||
|
||||
#include <string.h>
|
||||
#include <openssl/sha.h>
|
||||
|
||||
static const char ws_key_uuid[] = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
|
||||
|
||||
int ws_calc_accept_key(const char *key, char *accept_key) {
|
||||
if (key == NULL || accept_key == NULL)
|
||||
return -1;
|
||||
|
||||
char input[256] = "";
|
||||
unsigned char output[SHA_DIGEST_LENGTH];
|
||||
strcat(input, key);
|
||||
strcat(input, ws_key_uuid);
|
||||
|
||||
if (SHA1((unsigned char *) input, strlen(input), output) == NULL) {
|
||||
return -2;
|
||||
}
|
||||
|
||||
base64_encode(output, sizeof(output), accept_key, NULL);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int ws_recv_frame_header(sock *s, ws_frame *frame) {
|
||||
unsigned char buf[12];
|
||||
|
||||
long ret = sock_recv(s, buf, 2, 0);
|
||||
if (ret < 0) {
|
||||
error("Unable to receive from socket");
|
||||
return -1;
|
||||
} else if (ret != 2) {
|
||||
error("Unable to receive 2 bytes from socket");
|
||||
return -2;
|
||||
}
|
||||
|
||||
unsigned short bits = (buf[0] << 8) | buf[1];
|
||||
frame->f_fin = (bits >> 15) & 1;
|
||||
frame->f_rsv1 = (bits >> 14) & 1;
|
||||
frame->f_rsv2 = (bits >> 13) & 1;
|
||||
frame->f_rsv3 = (bits >> 12) & 1;
|
||||
frame->opcode = (bits >> 8) & 0xF;
|
||||
frame->f_mask = (bits >> 7) & 1;
|
||||
unsigned short len = (bits & 0x7F);
|
||||
|
||||
int remaining = frame->f_mask ? 4 : 0;
|
||||
if (len == 126) {
|
||||
remaining += 2;
|
||||
} else if (len == 127) {
|
||||
remaining += 8;
|
||||
}
|
||||
|
||||
ret = sock_recv(s, buf, remaining, 0);
|
||||
if (ret < 0) {
|
||||
error("Unable to receive from socket");
|
||||
return -1;
|
||||
} else if (ret != remaining) {
|
||||
error("Unable to receive correct number of bytes from socket");
|
||||
return -2;
|
||||
}
|
||||
|
||||
if (len == 126) {
|
||||
frame->len = (((unsigned long) buf[0]) << 8) | ((unsigned long) buf[1]);
|
||||
} else if (len == 127) {
|
||||
frame->len =
|
||||
(((unsigned long) buf[0]) << 56) |
|
||||
(((unsigned long) buf[1]) << 48) |
|
||||
(((unsigned long) buf[2]) << 40) |
|
||||
(((unsigned long) buf[3]) << 32) |
|
||||
(((unsigned long) buf[4]) << 24) |
|
||||
(((unsigned long) buf[5]) << 16) |
|
||||
(((unsigned long) buf[6]) << 8) |
|
||||
(((unsigned long) buf[7]) << 0);
|
||||
} else {
|
||||
frame->len = len;
|
||||
}
|
||||
|
||||
if (frame->f_mask) memcpy(frame->masking_key, buf + (remaining - 4), 4);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int ws_send_frame_header(sock *s, ws_frame *frame) {
|
||||
unsigned char buf[14], *ptr = buf;
|
||||
|
||||
unsigned short len;
|
||||
if (frame->len > 0x7FFF) {
|
||||
len = 127;
|
||||
} else if (frame->len > 125) {
|
||||
len = 126;
|
||||
} else {
|
||||
len = frame->len;
|
||||
}
|
||||
|
||||
unsigned short bits =
|
||||
(frame->f_fin << 15) |
|
||||
(frame->f_rsv1 << 14) |
|
||||
(frame->f_rsv2 << 13) |
|
||||
(frame->f_rsv3 << 12) |
|
||||
(frame->opcode << 8) |
|
||||
(frame->f_mask << 7) |
|
||||
len;
|
||||
|
||||
ptr++[0] = bits >> 8;
|
||||
ptr++[0] = bits & 0xFF;
|
||||
|
||||
if (len >= 126) {
|
||||
for (int i = (len == 126 ? 2 : 8) - 1; i >= 0; i--)
|
||||
ptr++[0] = (unsigned char) ((frame->len >> (i * 8)) & 0xFF);
|
||||
}
|
||||
|
||||
if (frame->f_mask) {
|
||||
memcpy(ptr, frame->masking_key, 4);
|
||||
ptr += 4;
|
||||
}
|
||||
|
||||
long ret = sock_send(s, buf, ptr - buf, frame->len != 0 ? MSG_MORE : 0);
|
||||
if (ret < 0) {
|
||||
error("Unable to send to socket");
|
||||
return -1;
|
||||
} else if (ret != ptr - buf) {
|
||||
error("Unable to send to socket");
|
||||
return -2;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
33
src/lib/websocket.h
Normal file
33
src/lib/websocket.h
Normal file
@ -0,0 +1,33 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief WebSocket reverse proxy (header file)
|
||||
* @file src/lib/websocket.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-08-16
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_WEBSOCKET_H
|
||||
#define SESIMOS_WEBSOCKET_H
|
||||
|
||||
#include "sock.h"
|
||||
|
||||
#define WS_TIMEOUT 3600
|
||||
|
||||
typedef struct {
|
||||
unsigned char f_fin:1;
|
||||
unsigned char f_rsv1:1;
|
||||
unsigned char f_rsv2:1;
|
||||
unsigned char f_rsv3:1;
|
||||
unsigned char opcode:4;
|
||||
unsigned char f_mask:1;
|
||||
unsigned long len;
|
||||
char masking_key[4];
|
||||
} ws_frame;
|
||||
|
||||
int ws_calc_accept_key(const char *key, char *accept_key);
|
||||
|
||||
int ws_recv_frame_header(sock *s, ws_frame *frame);
|
||||
|
||||
int ws_send_frame_header(sock *s, ws_frame *frame);
|
||||
|
||||
#endif //SESIMOS_WEBSOCKET_H
|
281
src/logger.c
Normal file
281
src/logger.c
Normal file
@ -0,0 +1,281 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Logger
|
||||
* @file src/logger.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-10
|
||||
*/
|
||||
|
||||
#include "logger.h"
|
||||
#include "lib/utils.h"
|
||||
#include "lib/error.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <pthread.h>
|
||||
#include <semaphore.h>
|
||||
#include <errno.h>
|
||||
#include <string.h>
|
||||
#include <stdarg.h>
|
||||
#include <signal.h>
|
||||
#include <malloc.h>
|
||||
|
||||
#define LOG_MAX_MSG_SIZE 2048
|
||||
#define LOG_BUF_SIZE 16
|
||||
#define LOG_NAME_LEN 12
|
||||
#define LOG_PREFIX_LEN 256
|
||||
|
||||
#define LOG_PREFIX "[%-8s][%-6s]"
|
||||
|
||||
typedef struct {
|
||||
log_lvl_t lvl;
|
||||
char name[LOG_NAME_LEN];
|
||||
char prefix[LOG_PREFIX_LEN];
|
||||
char txt[LOG_MAX_MSG_SIZE];
|
||||
} log_msg_t;
|
||||
|
||||
typedef struct {
|
||||
int rd;
|
||||
int wr;
|
||||
log_msg_t msgs[LOG_BUF_SIZE];
|
||||
} buf_t;
|
||||
|
||||
static pthread_t thread;
|
||||
static volatile sig_atomic_t alive = 0;
|
||||
static sem_t sem_buf, sem_buf_free, sem_buf_used;
|
||||
static buf_t buffer;
|
||||
|
||||
static pthread_key_t key_name = -1, key_prefix = -1;
|
||||
static char global_name[LOG_NAME_LEN] = "", global_prefix[LOG_PREFIX_LEN] = "";
|
||||
|
||||
static const char *level_keywords[] = {
|
||||
"EMERG",
|
||||
"ALERT",
|
||||
"CRIT",
|
||||
"ERROR",
|
||||
"WARN",
|
||||
"NOTICE",
|
||||
"INFO",
|
||||
"DEBUG"
|
||||
};
|
||||
|
||||
static void err(const char *restrict msg) {
|
||||
char err_buf[64];
|
||||
fprintf(stderr, ERR_STR LOG_PREFIX " %s: %s" CLR_STR "\n", "logger",
|
||||
level_keywords[LOG_CRITICAL], msg, error_str(errno, err_buf, sizeof(err_buf)));
|
||||
}
|
||||
|
||||
void logmsgf(log_lvl_t level, const char *restrict format, ...) {
|
||||
char buf[256];
|
||||
char err_buf[64];
|
||||
va_list args;
|
||||
va_start(args, format);
|
||||
|
||||
const char *color = (level <= LOG_ERROR) ? ERR_STR : ((level <= LOG_WARNING) ? WRN_STR : "");
|
||||
if (errno != 0) {
|
||||
snprintf(buf, sizeof(buf), "%s%s: %s" CLR_STR, color, format, error_str(errno, err_buf, sizeof(err_buf)));
|
||||
} else {
|
||||
snprintf(buf, sizeof(buf), "%s%s" CLR_STR, color, format);
|
||||
}
|
||||
|
||||
void *name = pthread_getspecific(key_name);
|
||||
if (name == NULL && global_name[0] != 0) name = global_name;
|
||||
void *prefix = pthread_getspecific(key_prefix);
|
||||
if (prefix == NULL && global_prefix[0] != 0) prefix = global_prefix;
|
||||
|
||||
if (!alive) {
|
||||
// no logger thread running
|
||||
// simply write to stdout without synchronization
|
||||
printf("%s" LOG_PREFIX "%s%s ", color, (name != NULL) ? (char *) name : "", level_keywords[level], CLR_STR, (prefix != NULL) ? (char *) prefix : "");
|
||||
vprintf(buf, args);
|
||||
printf("\n");
|
||||
} else {
|
||||
// wait for free slot in buffer
|
||||
try_again_free:
|
||||
if (sem_wait(&sem_buf_free) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
goto try_again_free;
|
||||
} else {
|
||||
err("Unable to lock semaphore");
|
||||
errno = 0;
|
||||
}
|
||||
// cleanup
|
||||
va_end(args);
|
||||
return;
|
||||
}
|
||||
|
||||
// try to lock buffer
|
||||
try_again_buf:
|
||||
if (sem_wait(&sem_buf) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
goto try_again_buf;
|
||||
} else {
|
||||
err("Unable to lock semaphore");
|
||||
errno = 0;
|
||||
}
|
||||
// cleanup
|
||||
sem_post(&sem_buf_free);
|
||||
va_end(args);
|
||||
return;
|
||||
}
|
||||
|
||||
// write message to buffer
|
||||
log_msg_t *msg = &buffer.msgs[buffer.rd];
|
||||
buffer.rd = (buffer.rd + 1) % LOG_BUF_SIZE;
|
||||
|
||||
vsnprintf(msg->txt, sizeof(msg->txt), buf, args);
|
||||
msg->lvl = level;
|
||||
|
||||
if (name != NULL) {
|
||||
snprintf(msg->name, sizeof(msg->name), "%s", (char *) name);
|
||||
} else {
|
||||
msg->name[0] = 0;
|
||||
}
|
||||
|
||||
if (prefix != NULL) {
|
||||
snprintf(msg->prefix, sizeof(msg->prefix), "%s", (char *) prefix);
|
||||
} else {
|
||||
msg->prefix[0] = 0;
|
||||
}
|
||||
|
||||
// unlock buffer
|
||||
sem_post(&sem_buf);
|
||||
|
||||
// unlock slot in buffer for logger
|
||||
sem_post(&sem_buf_used);
|
||||
}
|
||||
|
||||
// cleanup
|
||||
va_end(args);
|
||||
}
|
||||
|
||||
static void logger_destroy(void) {
|
||||
sem_destroy(&sem_buf);
|
||||
sem_destroy(&sem_buf_free);
|
||||
sem_destroy(&sem_buf_used);
|
||||
}
|
||||
|
||||
static int logger_remaining(void) {
|
||||
int val = 0;
|
||||
sem_getvalue(&sem_buf_used, &val);
|
||||
return val;
|
||||
}
|
||||
|
||||
void logger_set_name(const char *restrict format, ...) {
|
||||
va_list args;
|
||||
|
||||
if (key_name == -1) {
|
||||
// not initialized
|
||||
va_start(args, format);
|
||||
vsnprintf(global_name, sizeof(global_name), format, args);
|
||||
} else {
|
||||
int ret;
|
||||
void *ptr = pthread_getspecific(key_name);
|
||||
if (!ptr) {
|
||||
ptr = malloc(LOG_NAME_LEN);
|
||||
if ((ret = pthread_setspecific(key_name, ptr)) != 0) {
|
||||
errno = ret;
|
||||
err("Unable to set thread specific values");
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
va_start(args, format);
|
||||
vsnprintf(ptr, LOG_NAME_LEN, format, args);
|
||||
}
|
||||
|
||||
// cleanup
|
||||
va_end(args);
|
||||
}
|
||||
|
||||
void logger_set_prefix(const char *restrict format, ...) {
|
||||
va_list args;
|
||||
|
||||
if (key_prefix == -1) {
|
||||
// not initialized
|
||||
va_start(args, format);
|
||||
vsnprintf(global_prefix, sizeof(global_prefix), format, args);
|
||||
} else {
|
||||
int ret;
|
||||
void *ptr = pthread_getspecific(key_prefix);
|
||||
if (!ptr) {
|
||||
ptr = malloc(LOG_PREFIX_LEN);
|
||||
if ((ret = pthread_setspecific(key_prefix, ptr)) != 0) {
|
||||
errno = ret;
|
||||
err("Unable to set thread specific values");
|
||||
return;
|
||||
}
|
||||
}
|
||||
va_start(args, format);
|
||||
vsnprintf(ptr, LOG_PREFIX_LEN, format, args);
|
||||
}
|
||||
|
||||
// cleanup
|
||||
va_end(args);
|
||||
}
|
||||
|
||||
static void *logger_thread(void *arg) {
|
||||
logger_set_name("logger");
|
||||
alive = 1;
|
||||
|
||||
while (alive || logger_remaining() > 0) {
|
||||
// wait for buffer to be filled
|
||||
if (sem_wait(&sem_buf_used) != 0) {
|
||||
if (errno == EINTR) {
|
||||
errno = 0;
|
||||
continue;
|
||||
} else {
|
||||
err("Unable to lock semaphore");
|
||||
errno = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
log_msg_t *msg = &buffer.msgs[buffer.wr];
|
||||
buffer.wr = (buffer.wr + 1) % LOG_BUF_SIZE;
|
||||
|
||||
printf("%s" LOG_PREFIX "%s%s %s\n",
|
||||
(msg->lvl <= LOG_ERROR) ? ERR_STR : ((msg->lvl <= LOG_WARNING) ? WRN_STR : ""),
|
||||
(msg->name[0] != 0) ? (char *) msg->name : "", level_keywords[msg->lvl], CLR_STR,
|
||||
(msg->prefix[0] != 0) ? (char *) msg->prefix : "", msg->txt);
|
||||
|
||||
// unlock slot in buffer
|
||||
sem_post(&sem_buf_free);
|
||||
}
|
||||
|
||||
logger_destroy();
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
int logger_init(void) {
|
||||
int ret;
|
||||
|
||||
// try to initialize all three semaphores
|
||||
if (sem_init(&sem_buf, 0, 1) != 0 || sem_init(&sem_buf_free, 0, LOG_BUF_SIZE) != 0 || sem_init(&sem_buf_used, 0, 0) != 0) {
|
||||
err("Unable to initialize semaphore");
|
||||
logger_destroy();
|
||||
return -1;
|
||||
}
|
||||
|
||||
// initialize read/write heads
|
||||
buffer.rd = 0;
|
||||
buffer.wr = 0;
|
||||
|
||||
// initialize thread specific values (keys)
|
||||
if ((ret = pthread_key_create(&key_name, free)) != 0 || (ret = pthread_key_create(&key_prefix, free)) != 0) {
|
||||
errno = ret;
|
||||
err("Unable to initialize thread specific values");
|
||||
logger_destroy();
|
||||
return -1;
|
||||
}
|
||||
|
||||
pthread_create(&thread, NULL, logger_thread, NULL);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void logger_stop(void) {
|
||||
alive = 0;
|
||||
}
|
40
src/logger.h
Normal file
40
src/logger.h
Normal file
@ -0,0 +1,40 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Logger (header file)
|
||||
* @file src/logger.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-10
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_LOGGER_H
|
||||
#define SESIMOS_LOGGER_H
|
||||
|
||||
#define LOG_DEBUG 7
|
||||
#define LOG_INFO 6
|
||||
#define LOG_NOTICE 5
|
||||
#define LOG_WARNING 4
|
||||
#define LOG_ERROR 3
|
||||
#define LOG_CRITICAL 2
|
||||
#define LOG_ALERT 1
|
||||
|
||||
typedef unsigned char log_lvl_t;
|
||||
|
||||
#define debug(...) logmsgf(LOG_DEBUG, __VA_ARGS__)
|
||||
#define info(...) logmsgf(LOG_INFO, __VA_ARGS__)
|
||||
#define notice(...) logmsgf(LOG_NOTICE, __VA_ARGS__)
|
||||
#define warning(...) logmsgf(LOG_WARNING, __VA_ARGS__)
|
||||
#define error(...) logmsgf(LOG_ERROR, __VA_ARGS__)
|
||||
#define critical(...) logmsgf(LOG_CRITICAL, __VA_ARGS__)
|
||||
#define alert(...) logmsgf(LOG_ALERT, __VA_ARGS__)
|
||||
|
||||
void logmsgf(log_lvl_t level, const char *restrict format, ...);
|
||||
|
||||
void logger_set_name(const char *restrict format, ...);
|
||||
|
||||
void logger_set_prefix(const char *restrict format, ...);
|
||||
|
||||
int logger_init(void);
|
||||
|
||||
void logger_stop(void);
|
||||
|
||||
#endif //SESIMOS_LOGGER_H
|
@ -1,284 +0,0 @@
|
||||
/**
|
||||
* Necronda Web Server 3.0
|
||||
* necronda-server.cpp - Main Executable
|
||||
* Lorenz Stechauner, 2018-05-09
|
||||
*/
|
||||
|
||||
|
||||
#include "necronda-server.h"
|
||||
#include <magic.h>
|
||||
#include <iostream>
|
||||
#include <thread>
|
||||
#include <sys/time.h>
|
||||
#include <sys/stat.h>
|
||||
#include <signal.h>
|
||||
#include <csignal>
|
||||
|
||||
|
||||
using namespace std;
|
||||
|
||||
const char* webroot = "/srv/necronda/";
|
||||
|
||||
|
||||
string getMimeType(string path) {
|
||||
|
||||
unsigned long pos = path.find_last_of('.');
|
||||
string ext;
|
||||
if (pos != string::npos) {
|
||||
ext = path.substr(pos + 1, path.length() - pos);
|
||||
}
|
||||
|
||||
magic_t magic = magic_open(MAGIC_MIME_TYPE);
|
||||
magic_load(magic, "/usr/share/misc/magic.mgc");
|
||||
string type = magic_file(magic, path.c_str());
|
||||
magic_setflags(magic, MAGIC_MIME_ENCODING);
|
||||
string charset = magic_file(magic, path.c_str());
|
||||
|
||||
if (type == "text/plain") {
|
||||
if (ext == "css") {
|
||||
type = "text/css";
|
||||
} else if (ext == "js") {
|
||||
type = "text/javascript";
|
||||
}
|
||||
}
|
||||
|
||||
magic_close(magic);
|
||||
|
||||
return type + "; charset=" + charset;
|
||||
}
|
||||
|
||||
/**
|
||||
* Sun, 06 Nov 1994 08:49:37 GMT
|
||||
* @return
|
||||
*/
|
||||
|
||||
std::string getTimestamp(string path) {
|
||||
struct stat attrib;
|
||||
stat(path.c_str(), &attrib);
|
||||
return getTimestamp(attrib.st_ctime);
|
||||
}
|
||||
|
||||
std::string getTimestamp(time_t time) {
|
||||
char buffer[64];
|
||||
struct tm *timeinfo = gmtime(&time);
|
||||
strftime(buffer, sizeof(buffer), "%Y%m%d%H%M%S", timeinfo);
|
||||
return string(buffer);
|
||||
}
|
||||
|
||||
long getFileSize(string filename) {
|
||||
struct stat stat_buf;
|
||||
int rc = stat(filename.c_str(), &stat_buf);
|
||||
return rc == 0 ? stat_buf.st_size : -1;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Returns a formatted time string
|
||||
* @param micros Delta time to be formatted
|
||||
* @return A formatted time string
|
||||
*/
|
||||
std::string formatTime(long micros) {
|
||||
char buffer[64];
|
||||
if (micros < 1000) {
|
||||
sprintf(buffer, "%.3f ms", micros / 1000.0);
|
||||
} else if (micros < 10000) {
|
||||
sprintf(buffer, "%.2f ms", micros / 1000.0);
|
||||
} else if (micros < 100000) {
|
||||
sprintf(buffer, "%.1f ms", micros / 1000.0);
|
||||
} else if (micros < 1000000) {
|
||||
sprintf(buffer, "%.0f ms", micros / 1000.0);
|
||||
} else {
|
||||
sprintf(buffer, "%.1f s", micros / 1000000.0);
|
||||
}
|
||||
return std::string(buffer);
|
||||
}
|
||||
|
||||
std::string formatSize(unsigned long bytes) {
|
||||
char buffer[64];
|
||||
if (bytes > 0x10000000000) {
|
||||
sprintf(buffer, "%.1f TiB", (double) bytes / 0x10000000000);
|
||||
} else if (bytes > 0x40000000) {
|
||||
sprintf(buffer, "%.1f GiB", (double) bytes / 0x40000000);
|
||||
} else if (bytes > 0x100000) {
|
||||
sprintf(buffer, "%.1f MiB", (double) bytes / 0x100000);
|
||||
} else if (bytes > 0x400) {
|
||||
sprintf(buffer, "%.1f KiB", (double) bytes / 0x400);
|
||||
} else {
|
||||
sprintf(buffer, "%ld B", bytes);
|
||||
}
|
||||
return std::string(buffer);
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
string url_decode(string url) {
|
||||
long pos = 0;
|
||||
while ((pos = url.find('+', pos + 1)) != string::npos) {
|
||||
url.replace(pos, 1, 1, ' ');
|
||||
}
|
||||
pos = 0;
|
||||
while ((pos = url.find('%', pos + 1)) != string::npos) {
|
||||
const char *num = url.substr(pos + 1, 2).c_str();
|
||||
auto c = (char) strtol(num, nullptr, 16);
|
||||
url.erase(pos, 3);
|
||||
url.insert(pos, 1, c);
|
||||
}
|
||||
|
||||
return url;
|
||||
}
|
||||
|
||||
string url_encode(string url) {
|
||||
char buff[4];
|
||||
for (long pos = 0; pos < url.length(); pos++) {
|
||||
auto c = (unsigned char) url[pos];
|
||||
if (c < ' ' || c > '~' || c == ' ' || c == '#' || c == '?' || c == '&' || c == '=' || c == '\\' || c == '%') {
|
||||
sprintf(buff, "%%%02X", c);
|
||||
url.replace(pos, 1, buff);
|
||||
}
|
||||
}
|
||||
return url;
|
||||
}
|
||||
|
||||
string html_decode(string text) {
|
||||
return text;
|
||||
}
|
||||
|
||||
string html_encode(string text) {
|
||||
return text;
|
||||
}
|
||||
|
||||
string cli_encode(string text) {
|
||||
char buff[5];
|
||||
for (long pos = 0; pos < text.length(); pos++) {
|
||||
auto c = (unsigned char) text[pos];
|
||||
if (!((c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || (c >= '0' && c <= '9') || c == ',' || c == '.' || c == '_' || c == '+' || c == ':' || c == '@' || c == '%' || c == '/' || c == '-')) {
|
||||
sprintf(buff, "\\%.1s", &c);
|
||||
text.replace(pos, 1, buff);
|
||||
pos++;
|
||||
}
|
||||
}
|
||||
return text;
|
||||
}
|
||||
|
||||
string read_line(FILE* file) {
|
||||
char *line;
|
||||
size_t len = 0;
|
||||
ssize_t read;
|
||||
if ((read = getline(&line, &len, file)) < 0 || line == nullptr) {
|
||||
return "";
|
||||
}
|
||||
string l = string(line);
|
||||
if (l[l.length()-1] == '\n') {
|
||||
l.erase(l.length()-1);
|
||||
}
|
||||
if (l[l.length()-1] == '\r') {
|
||||
l.erase(l.length()-1);
|
||||
}
|
||||
return l;
|
||||
}
|
||||
|
||||
|
||||
#include "procopen.cpp"
|
||||
#include "network/Address.cpp"
|
||||
#include "network/Socket.cpp"
|
||||
#include "URI.cpp"
|
||||
#include "network/http/Http.cpp"
|
||||
#include "network/http/HttpStatusCode.cpp"
|
||||
#include "network/http/HttpHeader.cpp"
|
||||
#include "network/http/HttpRequest.cpp"
|
||||
#include "network/http/HttpResponse.cpp"
|
||||
#include "network/http/HttpConnection.cpp"
|
||||
|
||||
string getWebRoot(string host) {
|
||||
string root = webroot + host;
|
||||
if (fileExists(root)) {
|
||||
return root;
|
||||
} else {
|
||||
return "";
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#include "client.cpp"
|
||||
|
||||
|
||||
long clientnum = 0;
|
||||
|
||||
int main() {
|
||||
cout << "Necronda Server 3.0" << endl << "by Lorenz Stechauner" << endl << endl;
|
||||
|
||||
signal(SIGPIPE, SIG_IGN);
|
||||
|
||||
SSL_load_error_strings();
|
||||
SSL_library_init();
|
||||
ERR_load_crypto_strings();
|
||||
OpenSSL_add_all_algorithms();
|
||||
|
||||
int ret = system("mkdir -p /var/necronda /etc/necronda /tmp/necronda; touch /var/necronda/ETags");
|
||||
|
||||
if (ret != 0) {
|
||||
cout << "Unable to create server files" << endl;
|
||||
exit(1);
|
||||
}
|
||||
|
||||
list<unsigned short> ports = {80, 443};
|
||||
|
||||
list<Socket> servers = {};
|
||||
auto it = ports.begin();
|
||||
|
||||
for (int i = 0; i < ports.size(); i++) {
|
||||
unsigned short port = *it;
|
||||
advance(it, 1);
|
||||
Socket server = Socket();
|
||||
servers.push_back(server);
|
||||
|
||||
try {
|
||||
server.setReuseAddress(true);
|
||||
server.setReceiveTimeout(0);
|
||||
server.setSendTimeout(0);
|
||||
} catch (char *msg) {
|
||||
cout << "Unable to set socket option: " << msg << endl;
|
||||
exit(2);
|
||||
}
|
||||
|
||||
try {
|
||||
server.bind(port);
|
||||
} catch (char *msg) {
|
||||
cout << "Unable to bind socket to port " << port << ": " << msg << endl;
|
||||
exit(3);
|
||||
}
|
||||
|
||||
try {
|
||||
server.listen(256);
|
||||
} catch (char *msg) {
|
||||
cout << "Unable to listen on socket: " << msg << endl;
|
||||
exit(4);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
cout << "Ready for connections" << endl;
|
||||
|
||||
while (true) {
|
||||
try {
|
||||
Socket::select(servers, {});
|
||||
for (Socket server : servers) {
|
||||
try {
|
||||
Socket *socket = server.accept();
|
||||
clientnum++;
|
||||
thread *t = new thread(client_handler, socket, clientnum, server.getSocketPort() == 443);
|
||||
} catch (char *msg) {
|
||||
// Nothing
|
||||
}
|
||||
}
|
||||
} catch (char *msg) {
|
||||
cout << "Select: " << msg << endl;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
@ -1,41 +0,0 @@
|
||||
//
|
||||
// Created by lorenz on 5/17/18.
|
||||
//
|
||||
|
||||
#include <string>
|
||||
|
||||
#ifndef NECRONDA_SERVER
|
||||
#define NECRONDA_SERVER
|
||||
|
||||
using namespace std;
|
||||
|
||||
unsigned long getMicros();
|
||||
|
||||
string formatTime(long micros);
|
||||
|
||||
string formatSize(unsigned long bytes);
|
||||
|
||||
string getWebRoot(string host);
|
||||
|
||||
string getMimeType(string path);
|
||||
|
||||
string getTimestamp(string path);
|
||||
|
||||
string getTimestamp(time_t time);
|
||||
|
||||
long getFileSize(string filename);
|
||||
|
||||
string url_decode(string url);
|
||||
|
||||
string url_encode(string url);
|
||||
|
||||
string html_decode(string text);
|
||||
|
||||
string html_encode(string text);
|
||||
|
||||
string cli_encode(string text);
|
||||
|
||||
string read_line(FILE *file);
|
||||
|
||||
|
||||
#endif
|
@ -1,68 +0,0 @@
|
||||
|
||||
|
||||
#include <iostream>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <cstring>
|
||||
#include "Address.h"
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
Address::Address() {
|
||||
|
||||
}
|
||||
|
||||
Address::Address(string addr) {
|
||||
// TODO
|
||||
}
|
||||
|
||||
Address::Address(struct sockaddr_in *addr) {
|
||||
address = ntohl(addr->sin_addr.s_addr);
|
||||
}
|
||||
|
||||
|
||||
struct sockaddr_in Address::toStruct(unsigned short port)const {
|
||||
struct sockaddr_in addr;
|
||||
addr.sin_family = AF_INET;
|
||||
addr.sin_addr.s_addr = htonl(address);
|
||||
addr.sin_port = htons(port);
|
||||
return addr;
|
||||
}
|
||||
|
||||
|
||||
|
||||
string Address::toString() const {
|
||||
struct sockaddr_in addr = toStruct(0);
|
||||
struct in_addr ipAddr = addr.sin_addr;
|
||||
return inet_ntoa(ipAddr);
|
||||
}
|
||||
|
||||
bool Address::isLocal() {
|
||||
string a = toString();
|
||||
return a.find("127.0.0.") == 0;
|
||||
}
|
||||
|
||||
|
||||
ostream& operator<<(ostream &str, const Address &addr) {
|
||||
return str << addr.toString();
|
||||
}
|
||||
|
||||
string operator+(string &str, const Address &addr) {
|
||||
return str + addr.toString();
|
||||
}
|
||||
|
||||
string operator+(string &str, const Address *addr) {
|
||||
return str + addr->toString();
|
||||
}
|
||||
|
||||
string operator+(const Address &addr, string &str) {
|
||||
return addr.toString() + str;
|
||||
}
|
||||
|
||||
string operator+(const Address *addr, string &str) {
|
||||
return addr->toString() + str;
|
||||
}
|
||||
|
||||
|
||||
|
@ -1,31 +0,0 @@
|
||||
/**
|
||||
* Necronda Web Server 3.0
|
||||
* HttpHeader.h - HttpHeader Class definition
|
||||
* Lorenz Stechauner, 2018-05-09
|
||||
*/
|
||||
|
||||
#ifndef NECRONDA_ADDRESS
|
||||
#define NECRONDA_ADDRESS
|
||||
|
||||
using namespace std;
|
||||
|
||||
class Address {
|
||||
private:
|
||||
unsigned int address;
|
||||
|
||||
public:
|
||||
Address();
|
||||
|
||||
explicit Address(string address);
|
||||
|
||||
explicit Address(struct sockaddr_in *address);
|
||||
|
||||
struct sockaddr_in toStruct(unsigned short port) const;
|
||||
|
||||
string toString() const;
|
||||
|
||||
bool isLocal();
|
||||
|
||||
};
|
||||
|
||||
#endif
|
@ -1,661 +0,0 @@
|
||||
/**
|
||||
* Necronda Web Server 3.0
|
||||
* Socket.cpp - Socket Class methods
|
||||
* Lorenz Stechauner, 2018-05-09
|
||||
*/
|
||||
|
||||
|
||||
#include <iostream>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <cstring>
|
||||
#include <utility>
|
||||
#include <unistd.h>
|
||||
#include <sstream>
|
||||
#include <ctime>
|
||||
#include <poll.h>
|
||||
#include <openssl/ssl.h>
|
||||
#include <openssl/bio.h>
|
||||
#include <openssl/err.h>
|
||||
#include <openssl/x509.h>
|
||||
#include <list>
|
||||
#include <sys/stat.h>
|
||||
|
||||
#include "Address.h"
|
||||
#include "Socket.h"
|
||||
#include "http/Http.h"
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
static void multi_ssl_init() {
|
||||
SSL_load_error_strings();
|
||||
SSL_library_init();
|
||||
ERR_load_crypto_strings();
|
||||
OpenSSL_add_all_algorithms();
|
||||
}
|
||||
|
||||
|
||||
char *multi_ssl_get_error(SSL *ssl, int ret) {
|
||||
if (ret > 0) {
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
unsigned long ret2 = ERR_get_error();
|
||||
const char *err2 = strerror(errno);
|
||||
const char *err1 = ERR_reason_error_string(ret2);
|
||||
|
||||
switch (SSL_get_error(ssl, ret)) {
|
||||
case SSL_ERROR_NONE:
|
||||
return (char *) "none";
|
||||
case SSL_ERROR_ZERO_RETURN:
|
||||
return (char *) "closed";
|
||||
case SSL_ERROR_WANT_READ:
|
||||
return (char *) "want_read";
|
||||
case SSL_ERROR_WANT_WRITE:
|
||||
return (char *) "want_write";
|
||||
case SSL_ERROR_WANT_CONNECT:
|
||||
return (char *) "want_connect";
|
||||
case SSL_ERROR_WANT_ACCEPT:
|
||||
return (char *) "want_accept";
|
||||
case SSL_ERROR_WANT_X509_LOOKUP:
|
||||
return (char *) "want_x509_lookup";
|
||||
case SSL_ERROR_SYSCALL:
|
||||
return (char *) ((ret2 == 0) ? (ret == 0) ? "protocol violation" : err2 : err1);
|
||||
case SSL_ERROR_SSL:
|
||||
return (char *) err1;
|
||||
default:
|
||||
return (char *) "unknown error";
|
||||
}
|
||||
}
|
||||
|
||||
char *strerror_socket(int nr) {
|
||||
if (nr == EAGAIN || nr == EWOULDBLOCK) {
|
||||
return (char *) "timeout";
|
||||
} else if (nr == ECONNRESET) {
|
||||
return (char *) "closed";
|
||||
} else {
|
||||
return strerror(nr);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Socket::Socket(int fd) {
|
||||
this->fd = fd;
|
||||
microsStart = getMicros();
|
||||
microsLast = microsStart;
|
||||
bytesSent = 0;
|
||||
bytesReceived = 0;
|
||||
enc = false;
|
||||
ssl = nullptr;
|
||||
ctx = nullptr;
|
||||
clients = false;
|
||||
servers = false;
|
||||
}
|
||||
|
||||
Socket::Socket() {
|
||||
fd = ::socket(AF_INET, SOCK_STREAM, 0);
|
||||
if (fd == 0) {
|
||||
throw strerror(errno);
|
||||
}
|
||||
enc = false;
|
||||
microsStart = getMicros();
|
||||
microsLast = microsStart;
|
||||
bytesSent = 0;
|
||||
bytesReceived = 0;
|
||||
ssl = nullptr;
|
||||
ctx = nullptr;
|
||||
clients = false;
|
||||
servers = false;
|
||||
}
|
||||
|
||||
int Socket::getFd() {
|
||||
return fd;
|
||||
}
|
||||
|
||||
void Socket::setSocketOption(int option, bool value = true) {
|
||||
int val = value ? 1 : 0;
|
||||
|
||||
if (::setsockopt(fd, SOL_SOCKET, option, &val, sizeof(val)) != 0) {
|
||||
throw strerror(errno);
|
||||
}
|
||||
}
|
||||
|
||||
void Socket::bind(Address *address, unsigned short port) {
|
||||
struct sockaddr_in addr;
|
||||
addr.sin_family = AF_INET;
|
||||
addr.sin_addr.s_addr = INADDR_ANY; // address.
|
||||
addr.sin_port = htons(port);
|
||||
|
||||
if (::bind(fd, (struct sockaddr *) &addr, sizeof(addr)) != 0) {
|
||||
throw strerror(errno);
|
||||
}
|
||||
}
|
||||
|
||||
void Socket::bind(unsigned short port) {
|
||||
struct sockaddr_in addr;
|
||||
addr.sin_family = AF_INET;
|
||||
addr.sin_addr.s_addr = INADDR_ANY;
|
||||
addr.sin_port = htons(port);
|
||||
|
||||
if (::bind(fd, (struct sockaddr *) &addr, sizeof(addr)) != 0) {
|
||||
throw strerror(errno);
|
||||
}
|
||||
}
|
||||
|
||||
void Socket::listen(int num) {
|
||||
if (::listen(fd, num) != 0) {
|
||||
throw strerror(errno);
|
||||
}
|
||||
}
|
||||
|
||||
void Socket::connect(Address, unsigned short) {
|
||||
|
||||
}
|
||||
|
||||
Socket* Socket::accept() {
|
||||
int newfd = ::accept(fd, nullptr, nullptr);
|
||||
if (newfd < 0) {
|
||||
throw strerror(errno);
|
||||
}
|
||||
Socket *socket = new Socket(newfd);
|
||||
socket->servers = true;
|
||||
return socket;
|
||||
}
|
||||
|
||||
void Socket::close() {
|
||||
if (isSecured()) {
|
||||
//SSL_shutdown(ssl);
|
||||
SSL_free(ssl);
|
||||
SSL_CTX_free(ctx);
|
||||
}
|
||||
|
||||
if (::close(fd) != 0) {
|
||||
throw strerror(errno);
|
||||
}
|
||||
}
|
||||
|
||||
Address *Socket::getPeerAddress() const {
|
||||
struct sockaddr_storage addr;
|
||||
socklen_t len = sizeof(addr);
|
||||
getpeername(fd, (struct sockaddr *) &addr, &len);
|
||||
struct sockaddr_in *s = (struct sockaddr_in *) &addr;
|
||||
return new Address(s);
|
||||
}
|
||||
|
||||
unsigned short Socket::getPeerPort() const {
|
||||
struct sockaddr_storage addr;
|
||||
socklen_t len = sizeof(addr);
|
||||
getpeername(fd, (struct sockaddr *) &addr, &len);
|
||||
return ntohs(((struct sockaddr_in *) &addr)->sin_port);
|
||||
}
|
||||
|
||||
Address *Socket::getSocketAddress() const {
|
||||
struct sockaddr_storage addr;
|
||||
socklen_t len = sizeof(addr);
|
||||
getsockname(fd, (struct sockaddr *) &addr, &len);
|
||||
struct sockaddr_in *s = (struct sockaddr_in *) &addr;
|
||||
return new Address(s);
|
||||
}
|
||||
|
||||
unsigned short Socket::getSocketPort() const {
|
||||
struct sockaddr_storage addr;
|
||||
socklen_t len = sizeof(addr);
|
||||
getsockname(fd, (struct sockaddr *) &addr, &len);
|
||||
return ntohs(((struct sockaddr_in *) &addr)->sin_port);
|
||||
}
|
||||
|
||||
|
||||
void Socket::setReuseAddress(bool value) {
|
||||
setSocketOption(SO_REUSEADDR, value);
|
||||
}
|
||||
|
||||
void Socket::setReusePort(bool value) {
|
||||
setSocketOption(SO_REUSEPORT, value);
|
||||
}
|
||||
|
||||
|
||||
string Socket::toString() const {
|
||||
return "{[Socket]" + getSocketAddress()->toString() + ":" + to_string(getSocketPort()) + "<->" +
|
||||
getPeerAddress()->toString() + ":" + to_string(getPeerPort()) + "}";
|
||||
}
|
||||
|
||||
long Socket::send(string *str) {
|
||||
return send(str->c_str(), str->length());
|
||||
}
|
||||
|
||||
long Socket::send(string str) {
|
||||
return send(str.c_str(), str.length());
|
||||
}
|
||||
|
||||
long Socket::send(const char *str, long length) {
|
||||
return send((void*) str, length);
|
||||
}
|
||||
|
||||
long Socket::send(const char *str) {
|
||||
return send(str, strlen(str));
|
||||
}
|
||||
|
||||
long Socket::send(FILE *file) {
|
||||
char buffer[CPPNET_CHUNK];
|
||||
long all_len = 0;
|
||||
long len = 0;
|
||||
do {
|
||||
len = fread(buffer, 1, CPPNET_CHUNK, file);
|
||||
send(buffer, len);
|
||||
all_len += len;
|
||||
} while (len > 0 && len == CPPNET_CHUNK);
|
||||
return all_len;
|
||||
}
|
||||
|
||||
Socket::~Socket() {
|
||||
|
||||
}
|
||||
|
||||
long Socket::receive(void *buffer, int size) {
|
||||
long len;
|
||||
if (isSecured()) {
|
||||
len = SSL_read(ssl, buffer, size);
|
||||
if (len < 0) {
|
||||
throw multi_ssl_get_error(ssl, (int) len);
|
||||
}
|
||||
} else {
|
||||
len = recv(fd, buffer, (size_t) size, 0);
|
||||
if (len < 0) {
|
||||
throw strerror_socket(errno);
|
||||
}
|
||||
}
|
||||
bytesReceived += len;
|
||||
return len;
|
||||
}
|
||||
|
||||
long Socket::peek(void *buffer, int size) {
|
||||
long len;
|
||||
if (isSecured()) {
|
||||
len = SSL_peek(ssl, buffer, size);
|
||||
if (len < 0) {
|
||||
throw multi_ssl_get_error(ssl, (int) len);
|
||||
}
|
||||
} else {
|
||||
len = recv(fd, buffer, (size_t) size, MSG_PEEK);
|
||||
if (len < 0) {
|
||||
throw strerror_socket(errno);
|
||||
}
|
||||
}
|
||||
return len;
|
||||
}
|
||||
|
||||
long Socket::send(void *buffer, int size) {
|
||||
long len;
|
||||
if (isSecured()) {
|
||||
if (size != 0) {
|
||||
len = SSL_write(ssl, buffer, size);
|
||||
if (len <= 0) {
|
||||
throw multi_ssl_get_error(ssl, (int) len);
|
||||
}
|
||||
} else {
|
||||
len = 0;
|
||||
}
|
||||
} else {
|
||||
len = ::send(fd, buffer, (size_t) size, 0);
|
||||
if (len < 0) {
|
||||
throw strerror_socket(errno);
|
||||
}
|
||||
}
|
||||
bytesSent += len;
|
||||
return len;
|
||||
}
|
||||
|
||||
|
||||
string Socket::receive() {
|
||||
string *str = new string();
|
||||
|
||||
char buffer[CPPNET_CHUNK];
|
||||
long len = 0;
|
||||
do {
|
||||
len = receive((void*) buffer, CPPNET_CHUNK);
|
||||
str->append(buffer, (unsigned) len);
|
||||
} while (len > 0 && len == CPPNET_CHUNK);
|
||||
|
||||
return *str;
|
||||
}
|
||||
|
||||
string Socket::receive(long length) {
|
||||
string *str = new string();
|
||||
|
||||
char buffer[CPPNET_CHUNK];
|
||||
long len = 0;
|
||||
long reclen = 0;
|
||||
do {
|
||||
len = receive((void*) buffer, CPPNET_CHUNK);
|
||||
reclen += len;
|
||||
str->append(buffer, (unsigned) len);
|
||||
} while (reclen < length);
|
||||
|
||||
return *str;
|
||||
}
|
||||
|
||||
string Socket::receive(string until) {
|
||||
string *str = new string();
|
||||
|
||||
struct pollfd ufds[1];
|
||||
ufds[0].fd = fd;
|
||||
ufds[0].events = POLLIN | POLLOUT;
|
||||
|
||||
char buffer[CPPNET_CHUNK];
|
||||
long len = 0;
|
||||
do {
|
||||
len = peek((void*) buffer, CPPNET_CHUNK);
|
||||
if (len != 0) {
|
||||
string s = string(buffer, (size_t) len);
|
||||
size_t found = s.find(until);
|
||||
long l = (found != string::npos) ? found + 1 : len;
|
||||
long l2 = (found != string::npos) ? found : len;
|
||||
str->append(buffer, (unsigned) l2);
|
||||
receive((void *) buffer, (int) l);
|
||||
if (found != string::npos) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (poll(ufds, 1, 0) < 0) {
|
||||
throw strerror_socket(errno);
|
||||
} else if ((ufds[0].revents & POLLIN) == 0) {
|
||||
if ((ufds[0].revents & POLLOUT) != 0) {
|
||||
throw (char *) "error";
|
||||
} else {
|
||||
throw (char *) "want_write";
|
||||
}
|
||||
} else if ((ufds[0].revents & POLLERR) != 0) {
|
||||
throw (char *) "error";
|
||||
} else if (ufds[0].revents & (POLLRDHUP | POLLHUP | POLLNVAL) != 0) {
|
||||
throw (char *) "closed";
|
||||
}
|
||||
} while (true);
|
||||
|
||||
return *str;
|
||||
}
|
||||
|
||||
string Socket::receive(const char *until) {
|
||||
return receive(until, (int) (strlen(until)));
|
||||
}
|
||||
|
||||
string Socket::receive(const char *until, unsigned long strlen) {
|
||||
return receive(string(until, strlen));
|
||||
}
|
||||
|
||||
long Socket::receive(FILE *file) {
|
||||
char buffer[CPPNET_CHUNK];
|
||||
long len;
|
||||
long rec = 0;
|
||||
do {
|
||||
len = receive((void*) buffer, CPPNET_CHUNK);
|
||||
fwrite(buffer, 1, CPPNET_CHUNK, file);
|
||||
rec += len;
|
||||
} while (len > 0 && len == CPPNET_CHUNK);
|
||||
return len;
|
||||
}
|
||||
|
||||
long Socket::receive(FILE *file, long size) {
|
||||
char buffer[CPPNET_CHUNK];
|
||||
long len = 0;
|
||||
long rec = 0;
|
||||
while (size > rec) {
|
||||
len = receive((void*) buffer, (CPPNET_CHUNK > (size - rec) && size >= 0)?(size - rec):CPPNET_CHUNK);
|
||||
fwrite(buffer, 1, len, file);
|
||||
rec += len;
|
||||
}
|
||||
return rec;
|
||||
}
|
||||
|
||||
string Socket::receiveLine() {
|
||||
string str = receive("\n");
|
||||
if (str.length() > 0 && str.at(str.length() - 1) == '\r') {
|
||||
str = str.substr(0, str.length() - 1);
|
||||
}
|
||||
return str;
|
||||
}
|
||||
|
||||
|
||||
long Socket::getDuration() {
|
||||
return getMicros() - microsStart;
|
||||
}
|
||||
|
||||
|
||||
void Socket::setReceiveTimeout(unsigned long ms) {
|
||||
struct timeval timeout;
|
||||
if (ms == 0) {
|
||||
timeout.tv_sec = 0;
|
||||
timeout.tv_usec = 1;
|
||||
} else {
|
||||
timeout.tv_sec = ms / 1000;
|
||||
timeout.tv_usec = (ms % 1000) * 1000;
|
||||
}
|
||||
if (setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, sizeof(timeout)) < 0) {
|
||||
throw strerror(errno);
|
||||
}
|
||||
}
|
||||
|
||||
void Socket::setSendTimeout(unsigned long ms) {
|
||||
struct timeval timeout;
|
||||
if (ms == 0) {
|
||||
timeout.tv_sec = 0;
|
||||
timeout.tv_usec = 1;
|
||||
} else {
|
||||
timeout.tv_sec = ms / 1000;
|
||||
timeout.tv_usec = (ms % 1000) * 1000;
|
||||
}
|
||||
if (setsockopt(fd, SOL_SOCKET, SO_SNDTIMEO, (char *) &timeout, sizeof(timeout)) < 0) {
|
||||
throw strerror(errno);
|
||||
}
|
||||
}
|
||||
|
||||
bool Socket::isServerSide() {
|
||||
return servers;
|
||||
}
|
||||
|
||||
bool Socket::isSecured() {
|
||||
return enc;
|
||||
}
|
||||
|
||||
bool Socket::isClientSide() {
|
||||
return clients;
|
||||
}
|
||||
|
||||
void Socket::sslHandshake(map<string, KeyPair> sni) {
|
||||
/*if (isSecured()) {
|
||||
throw (char *) "Socket already secured";
|
||||
}
|
||||
|
||||
const SSL_METHOD *method;
|
||||
if (isServerSide()) {
|
||||
method = TLSv1_2_server_method();
|
||||
} else if (isClientSide()) {
|
||||
method = TLSv1_2_client_method();
|
||||
} else {
|
||||
method = TLSv1_2_method();
|
||||
}
|
||||
|
||||
SSL_CTX *ctx = SSL_CTX_new(method);
|
||||
SSL_CTX_set_options(ctx, SSL_OP_SINGLE_DH_USE | SSL_OP_NO_SSLv2 | SSL_OP_NO_SSLv3);
|
||||
SSL_CTX_set_mode(ctx, SSL_MODE_ENABLE_PARTIAL_WRITE);
|
||||
SSL_CTX_set_cipher_list(ctx, "HIGH:!aNULL:!kRSA:!PSK:!SRP:!MD5:!RC4");
|
||||
SSL_CTX_set_ecdh_auto(ctx, 1);
|
||||
|
||||
const char *certfile = keypair.fullchain.c_str();
|
||||
const char *keyfile = keypair.privkey.c_str();
|
||||
|
||||
if (isServerSide()) {
|
||||
if (SSL_CTX_use_certificate_file(ctx, certfile, SSL_FILETYPE_PEM) != 1) {
|
||||
throw (char *) ERR_reason_error_string(ERR_get_error());
|
||||
}
|
||||
|
||||
if (SSL_CTX_use_PrivateKey_file(ctx, keyfile, SSL_FILETYPE_PEM) != 1) {
|
||||
throw (char *) ERR_reason_error_string(ERR_get_error());
|
||||
}
|
||||
}
|
||||
|
||||
SSL_CTX_set_tlsext_servername_callback
|
||||
|
||||
this->ctx = ctx;
|
||||
this->ssl = SSL_new(ctx);
|
||||
SSL_set_fd(ssl, fd);
|
||||
enc = true;
|
||||
|
||||
while (true) {
|
||||
int ret = 0;
|
||||
if (isServerSide()) {
|
||||
ret = SSL_accept(ssl);
|
||||
} else if (isClientSide()) {
|
||||
ret = SSL_connect(ssl);
|
||||
} else {
|
||||
ret = SSL_do_handshake(ssl);
|
||||
}
|
||||
|
||||
if (ret <= 0 && ((isServerSide() && SSL_get_error(ssl, ret) != SSL_ERROR_WANT_READ) ||
|
||||
(isClientSide() && SSL_get_error(ssl, ret) != SSL_ERROR_WANT_WRITE))) {
|
||||
throw multi_ssl_get_error(ssl, ret);
|
||||
} else if (ret == 1) {
|
||||
break;
|
||||
}
|
||||
}*/
|
||||
|
||||
}
|
||||
|
||||
void Socket::sslHandshake() {
|
||||
sslHandshake(KeyPair{"", ""});
|
||||
}
|
||||
|
||||
void Socket::sslHandshake(KeyPair keypair) {
|
||||
if (isSecured()) {
|
||||
throw (char *) "Socket already secured";
|
||||
}
|
||||
|
||||
const SSL_METHOD *method;
|
||||
if (isServerSide()) {
|
||||
method = TLSv1_2_server_method();
|
||||
} else if (isClientSide()) {
|
||||
method = TLSv1_2_client_method();
|
||||
} else {
|
||||
method = TLSv1_2_method();
|
||||
}
|
||||
|
||||
SSL_CTX *ctx = SSL_CTX_new(method);
|
||||
SSL_CTX_set_options(ctx, SSL_OP_SINGLE_DH_USE);
|
||||
SSL_CTX_set_verify(ctx, SSL_VERIFY_NONE, nullptr);
|
||||
SSL_CTX_set_min_proto_version(ctx, SSL3_VERSION); // TLS1_VERSION
|
||||
SSL_CTX_set_mode(ctx, SSL_MODE_ENABLE_PARTIAL_WRITE);
|
||||
SSL_CTX_set_cipher_list(ctx, "HIGH:!aNULL:!kRSA:!PSK:!SRP:!MD5:!RC4");
|
||||
SSL_CTX_set_ecdh_auto(ctx, 1);
|
||||
|
||||
const char *certfile = keypair.fullchain.c_str();
|
||||
const char *keyfile = keypair.privkey.c_str();
|
||||
|
||||
if (isServerSide()) {
|
||||
if (SSL_CTX_use_certificate_chain_file(ctx, certfile) != 1) {
|
||||
throw (char *) ERR_reason_error_string(ERR_get_error());
|
||||
}
|
||||
|
||||
if (SSL_CTX_use_PrivateKey_file(ctx, keyfile, SSL_FILETYPE_PEM) != 1) {
|
||||
throw (char *) ERR_reason_error_string(ERR_get_error());
|
||||
}
|
||||
}
|
||||
|
||||
this->ctx = ctx;
|
||||
this->ssl = SSL_new(ctx);
|
||||
SSL_set_fd(ssl, fd);
|
||||
enc = true;
|
||||
|
||||
while (true) {
|
||||
int ret = 0;
|
||||
if (isServerSide()) {
|
||||
ret = SSL_accept(ssl);
|
||||
} else if (isClientSide()) {
|
||||
ret = SSL_connect(ssl);
|
||||
} else {
|
||||
ret = SSL_do_handshake(ssl);
|
||||
}
|
||||
|
||||
if (ret <= 0 && ((isServerSide() && SSL_get_error(ssl, ret) != SSL_ERROR_WANT_READ) ||
|
||||
(isClientSide() && SSL_get_error(ssl, ret) != SSL_ERROR_WANT_WRITE))) {
|
||||
throw multi_ssl_get_error(ssl, ret);
|
||||
} else if (ret == 1) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
void Socket::sslHandshake(string privkey, string fullchain) {
|
||||
sslHandshake(KeyPair{std::move(privkey), std::move(fullchain)});
|
||||
}
|
||||
|
||||
long Socket::select(list<Socket> read, list<Socket> write, long millis) {
|
||||
fd_set readfd, writefd;
|
||||
int maxfd = 0;
|
||||
FD_ZERO(&readfd);
|
||||
FD_ZERO(&writefd);
|
||||
|
||||
for (Socket s : read) {
|
||||
if (s.fd > maxfd) {
|
||||
maxfd = s.fd;
|
||||
}
|
||||
FD_SET(s.fd, &readfd);
|
||||
}
|
||||
|
||||
for (Socket s : write) {
|
||||
if (s.fd > maxfd) {
|
||||
maxfd = s.fd;
|
||||
}
|
||||
FD_SET(s.fd, &writefd);
|
||||
}
|
||||
|
||||
struct timeval *tv = new struct timeval;
|
||||
if (millis < 0) {
|
||||
tv = nullptr;
|
||||
} else if (millis == 0) {
|
||||
tv->tv_sec = 0;
|
||||
tv->tv_usec = 1;
|
||||
} else {
|
||||
tv->tv_sec = millis / 1000;
|
||||
tv->tv_usec = (millis % 1000) * 1000;
|
||||
}
|
||||
|
||||
int ret = ::select(maxfd + 1, &readfd, &writefd, nullptr, tv);
|
||||
if (ret < 0) {
|
||||
throw (char *) strerror(errno);
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
long Socket::select(list<Socket> read, list<Socket> write) {
|
||||
Socket::select(std::move(read), std::move(write), -1);
|
||||
}
|
||||
|
||||
unsigned long Socket::getBytesSent() {
|
||||
return bytesSent;
|
||||
}
|
||||
|
||||
unsigned long Socket::getBytesReceived() {
|
||||
return bytesReceived;
|
||||
}
|
||||
|
||||
|
||||
ostream &operator<<(ostream &str, const Socket &socket) {
|
||||
return str << socket.toString();
|
||||
}
|
||||
|
||||
ostream &operator<<(ostream &str, const Socket *socket) {
|
||||
return str << socket->toString();
|
||||
}
|
||||
|
||||
string operator+(string &str, const Socket &socket) {
|
||||
return str + socket.toString();
|
||||
}
|
||||
|
||||
string operator+(const Socket &socket, string &str) {
|
||||
return socket.toString() + str;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
@ -1,174 +0,0 @@
|
||||
/**
|
||||
* Necronda Web Server 3.0
|
||||
* Socket.h - Socket Class definition
|
||||
* Lorenz Stechauner, 2018-05-09
|
||||
*/
|
||||
|
||||
#ifndef NECRONDA_SOCKET
|
||||
#define NECRONDA_SOCKET
|
||||
|
||||
#include <map>
|
||||
|
||||
#define CPPNET_CHUNK 16384
|
||||
|
||||
typedef struct {
|
||||
string privkey;
|
||||
string fullchain;
|
||||
} KeyPair;
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
class Socket {
|
||||
private:
|
||||
int fd;
|
||||
SSL *ssl;
|
||||
SSL_CTX *ctx;
|
||||
bool enc;
|
||||
bool servers;
|
||||
bool clients;
|
||||
unsigned long bytesSent;
|
||||
unsigned long bytesReceived;
|
||||
long microsStart;
|
||||
long microsLast;
|
||||
|
||||
void setSocketOption(int, bool);
|
||||
|
||||
long send(void *buffer, int size);
|
||||
|
||||
long receive(void *buffer, int size);
|
||||
|
||||
long peek(void *buffer, int size);
|
||||
|
||||
public:
|
||||
Socket();
|
||||
|
||||
explicit Socket(int filedescriptor);
|
||||
|
||||
~Socket();
|
||||
|
||||
void bind(Address *address, unsigned short port);
|
||||
|
||||
void bind(unsigned short port);
|
||||
|
||||
void listen(int count = 1);
|
||||
|
||||
void connect(Address address, unsigned short port);
|
||||
|
||||
Socket* accept();
|
||||
|
||||
void sslHandshake();
|
||||
|
||||
void sslHandshake(map<string, KeyPair> sni);
|
||||
|
||||
void sslHandshake(KeyPair keypair);
|
||||
|
||||
void sslHandshake(string privkey, string fullchain);
|
||||
|
||||
long send(string *str);
|
||||
|
||||
long send(string str);
|
||||
|
||||
long send(const char *str);
|
||||
|
||||
long send(const char *str, long length);
|
||||
|
||||
long send(FILE *file);
|
||||
|
||||
string receive();
|
||||
|
||||
string receive(long length);
|
||||
|
||||
string receive(string until);
|
||||
|
||||
string receive(const char *until, unsigned long strlen);
|
||||
|
||||
string receive(const char *until);
|
||||
|
||||
long receive(FILE *file);
|
||||
|
||||
string receiveLine();
|
||||
|
||||
void shutdown();
|
||||
|
||||
void close();
|
||||
|
||||
int getFd();
|
||||
|
||||
long getDuration();
|
||||
|
||||
Address *getSocketAddress() const;
|
||||
|
||||
unsigned short getSocketPort() const;
|
||||
|
||||
Address *getPeerAddress() const;
|
||||
|
||||
unsigned short getPeerPort() const;
|
||||
|
||||
string toString() const;
|
||||
|
||||
|
||||
bool isServerSide();
|
||||
|
||||
bool isClientSide();
|
||||
|
||||
bool isSecured();
|
||||
|
||||
|
||||
void setReuseAddress(bool value = true);
|
||||
|
||||
void setReusePort(bool value = true);
|
||||
|
||||
void setSendBufferSize(int value);
|
||||
|
||||
void setReceiveBufferSize(int value);
|
||||
|
||||
void setMinReceiveBytes(int value);
|
||||
|
||||
void setMinSendBytes(int value);
|
||||
|
||||
void setSendTimeout(unsigned long ms);
|
||||
|
||||
void setReceiveTimeout(unsigned long ms);
|
||||
|
||||
|
||||
bool getReuseAddress();
|
||||
|
||||
bool getReusePort();
|
||||
|
||||
int getSendBufferSize();
|
||||
|
||||
int getReceiveBufferSize();
|
||||
|
||||
int getMinReceiveBytes();
|
||||
|
||||
int getMinSendBytes();
|
||||
|
||||
long getSendTimeout();
|
||||
|
||||
long getReceiveTimeout();
|
||||
|
||||
unsigned long getBytesSent();
|
||||
|
||||
unsigned long getBytesReceived();
|
||||
|
||||
static long select(list<Socket> read, list<Socket> write, long millis);
|
||||
|
||||
static long select(list<Socket> read, list<Socket> write);
|
||||
|
||||
long receive(FILE *file, long size);
|
||||
};
|
||||
|
||||
Socket operator<<(Socket sock, const char *str);
|
||||
|
||||
Socket operator<<(Socket sock, string str);
|
||||
|
||||
ostream &operator<<(ostream &str, const Socket &socket);
|
||||
|
||||
ostream &operator<<(ostream &str, const Socket *socket);
|
||||
|
||||
string operator+(string &str, const Socket &socket);
|
||||
|
||||
string operator+(const Socket &socket, string &str);
|
||||
|
||||
#endif
|
@ -1,32 +0,0 @@
|
||||
//
|
||||
// Created by lorenz on 7/10/18.
|
||||
//
|
||||
|
||||
#include <sys/stat.h>
|
||||
#include <sys/time.h>
|
||||
#include "Http.h"
|
||||
|
||||
unsigned long getMicros() {
|
||||
struct timeval tv;
|
||||
gettimeofday(&tv, nullptr);
|
||||
return (unsigned long) (1000000 * tv.tv_sec + tv.tv_usec);
|
||||
}
|
||||
|
||||
string getHttpDate() {
|
||||
time_t rawtime;
|
||||
time(&rawtime);
|
||||
return getHttpDate(rawtime);
|
||||
}
|
||||
|
||||
string getHttpDate(string filename) {
|
||||
struct stat attrib;
|
||||
stat(filename.c_str(), &attrib);
|
||||
return getHttpDate(attrib.st_ctime);
|
||||
}
|
||||
|
||||
string getHttpDate(time_t time) {
|
||||
char buffer[64];
|
||||
struct tm *timeinfo = gmtime(&time);
|
||||
strftime(buffer, sizeof(buffer), "%a, %d %b %Y %H:%M:%S GMT", timeinfo);
|
||||
return string(buffer);
|
||||
}
|
@ -1,22 +0,0 @@
|
||||
//
|
||||
// Created by lorenz on 7/10/18.
|
||||
//
|
||||
|
||||
#ifndef CPPNET_HTTP_H
|
||||
#define CPPNET_HTTP_H
|
||||
|
||||
#include <ctime>
|
||||
#include <string>
|
||||
|
||||
using namespace std;
|
||||
|
||||
unsigned long getMicros();
|
||||
|
||||
string getHttpDate(time_t time);
|
||||
|
||||
string getHttpDate();
|
||||
|
||||
string getHttpDate(string filename);
|
||||
|
||||
|
||||
#endif //CPPNET_HTTP_H
|
@ -1,194 +0,0 @@
|
||||
|
||||
|
||||
#include <zlib.h>
|
||||
#include <cassert>
|
||||
#include <iostream>
|
||||
#include <utility>
|
||||
#include "HttpConnection.h"
|
||||
#include "../Socket.h"
|
||||
#include "HttpStatusCode.h"
|
||||
#include "Http.h"
|
||||
|
||||
|
||||
HttpConnection::HttpConnection() = default;
|
||||
|
||||
HttpConnection::HttpConnection(Socket *socket) {
|
||||
this->socket = socket;
|
||||
this->request = new HttpRequest(socket);
|
||||
this->response = new HttpResponse();
|
||||
microsStart = getMicros();
|
||||
response->setVersion("1.1");
|
||||
response->setField("Server", "Necronda/3.0");
|
||||
}
|
||||
|
||||
void HttpConnection::respond(int statuscode) {
|
||||
if (statuscode >= 400 && statuscode < 600) {
|
||||
respond(statuscode,
|
||||
"<!DOCTYPE html><html><head><title>" + to_string(statuscode) + " " +
|
||||
::getStatusCode(statuscode).message +
|
||||
"</title></head><body><center><h1>" + to_string(statuscode) + " " +
|
||||
::getStatusCode(statuscode).message +
|
||||
"</h1>" +
|
||||
((request->isExistingField("Host")) ?
|
||||
(request->isExistingField("Referer") &&
|
||||
request->getField("Referer").find(request->getField("Host")) != string::npos) ?
|
||||
"<p>Go back to the last page you visited: <a href=\"" + request->getField("Referer") + "\">" +
|
||||
request->getField("Referer") + "</a></p>" :
|
||||
"<p>Go back to the home page of <a href=\"//" +
|
||||
request->getField("Host") + "/\">" +
|
||||
request->getField("Host") +
|
||||
"</a></p>" : "") + "</center></body></html>\r\n"
|
||||
);
|
||||
} else {
|
||||
respond(statuscode, "");
|
||||
}
|
||||
}
|
||||
|
||||
void HttpConnection::respond(int statuscode, string payload) {
|
||||
response->setStatusCode(statuscode);
|
||||
response->setField("Date", getHttpDate());
|
||||
response->setField("Content-Length", to_string(payload.length()));
|
||||
response->sendHeader(socket);
|
||||
socket->send(std::move(payload));
|
||||
}
|
||||
|
||||
void HttpConnection::respond(int statuscode, FILE *file, bool compress, long start, long end) {
|
||||
response->setStatusCode(statuscode);
|
||||
response->setField("Transfer-Encoding", "chunked");
|
||||
response->setField("Date", getHttpDate());
|
||||
|
||||
long shouldTransfer;
|
||||
long transfered = 0;
|
||||
|
||||
fseek(file, 0, SEEK_END);
|
||||
long len = ftell(file);
|
||||
|
||||
if (start != -1 && end != -1) {
|
||||
fseek(file, start, SEEK_SET);
|
||||
response->setField("Content-Length", to_string(end - start + 1));
|
||||
shouldTransfer = end - start + 1;
|
||||
compress = false;
|
||||
} else {
|
||||
fseek(file, 0, SEEK_SET);
|
||||
shouldTransfer = len;
|
||||
if (len >= 0 && !compress) {
|
||||
response->setField("Content-Length", to_string(len));
|
||||
}
|
||||
}
|
||||
|
||||
if (compress) {
|
||||
response->setField("Content-Encoding", "deflate");
|
||||
}
|
||||
|
||||
response->sendHeader(socket);
|
||||
|
||||
if (compress) {
|
||||
int level = 1;
|
||||
int ret, flush;
|
||||
unsigned have;
|
||||
z_stream strm;
|
||||
unsigned char in[CPPNET_CHUNK];
|
||||
unsigned char out[CPPNET_CHUNK];
|
||||
|
||||
strm.zalloc = Z_NULL;
|
||||
strm.zfree = Z_NULL;
|
||||
strm.opaque = Z_NULL;
|
||||
ret = deflateInit(&strm, level);
|
||||
if (ret != Z_OK) {
|
||||
throw (char *) "Unable to open file";
|
||||
}
|
||||
|
||||
do {
|
||||
strm.avail_in = (uInt) fread(in, 1, CPPNET_CHUNK, file);
|
||||
|
||||
if (ferror(file)) {
|
||||
(void) deflateEnd(&strm);
|
||||
throw (char *) strerror(errno);
|
||||
}
|
||||
flush = feof(file) ? Z_FINISH : Z_NO_FLUSH;
|
||||
strm.next_in = in;
|
||||
do {
|
||||
strm.avail_out = CPPNET_CHUNK;
|
||||
strm.next_out = out;
|
||||
ret = deflate(&strm, flush);
|
||||
assert(ret != Z_STREAM_ERROR);
|
||||
have = CPPNET_CHUNK - strm.avail_out;
|
||||
|
||||
if (have != 0) {
|
||||
char buffer[64];
|
||||
sprintf(buffer, "%X\r\n", have);
|
||||
socket->send(buffer);
|
||||
socket->send((const char *) out, have);
|
||||
socket->send("\r\n");
|
||||
}
|
||||
} while (strm.avail_out == 0);
|
||||
assert(strm.avail_in == 0);
|
||||
} while (flush != Z_FINISH);
|
||||
assert(ret == Z_STREAM_END);
|
||||
socket->send("0\r\n\r\n");
|
||||
deflateEnd(&strm);
|
||||
} else {
|
||||
char buffer[CPPNET_CHUNK];
|
||||
char buff[64];
|
||||
while (true) {
|
||||
unsigned long size = fread(buffer, 1, (size_t) ((CPPNET_CHUNK > (shouldTransfer - transfered) && shouldTransfer > 0) ? (shouldTransfer - transfered) : CPPNET_CHUNK), file);
|
||||
transfered += size;
|
||||
sprintf(buff, "%lX\r\n", size);
|
||||
socket->send(buff);
|
||||
socket->send((const char *) buffer, size);
|
||||
socket->send("\r\n");
|
||||
if (size == 0) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
string HttpConnection::getField(string index) {
|
||||
return request->getField(std::move(index));
|
||||
}
|
||||
|
||||
string HttpConnection::getPath() {
|
||||
return request->getPath();
|
||||
}
|
||||
|
||||
void HttpConnection::setField(string index, string data) {
|
||||
response->setField(std::move(index), std::move(data));
|
||||
}
|
||||
|
||||
bool HttpConnection::isExistingField(string index) {
|
||||
return request->isExistingField(std::move(index));
|
||||
}
|
||||
|
||||
string HttpConnection::getMethod() {
|
||||
return request->getMethod();
|
||||
}
|
||||
|
||||
long HttpConnection::getDuration() {
|
||||
return getMicros() - microsStart;
|
||||
}
|
||||
|
||||
HttpStatusCode HttpConnection::getStatusCode() {
|
||||
return response->getStatusCode();
|
||||
}
|
||||
|
||||
void HttpConnection::redirect(int statuscode, string location) {
|
||||
setField("Location", std::move(location));
|
||||
respond(statuscode, "");
|
||||
}
|
||||
|
||||
string HttpConnection::getResponseField(string index) {
|
||||
return response->getField(std::move(index));
|
||||
}
|
||||
|
||||
bool HttpConnection::isExistingResponseField(string index) {
|
||||
return response->isExistingField(std::move(index));
|
||||
}
|
||||
|
||||
string HttpConnection::cgiExport() {
|
||||
return request->cgiExport();
|
||||
}
|
||||
|
||||
void HttpConnection::removeField(string index) {
|
||||
response->removeField(std::move(index));
|
||||
}
|
@ -1,55 +0,0 @@
|
||||
|
||||
|
||||
#ifndef NECRONDA_HTTP_CONNECTION
|
||||
#define NECRONDA_HTTP_CONNECTION
|
||||
|
||||
|
||||
#include "../Socket.h"
|
||||
#include "HttpResponse.h"
|
||||
#include "HttpRequest.h"
|
||||
|
||||
|
||||
class HttpConnection {
|
||||
private:
|
||||
Socket *socket{};
|
||||
HttpRequest *request{};
|
||||
HttpResponse *response{};
|
||||
long microsStart{};
|
||||
|
||||
public:
|
||||
explicit HttpConnection();
|
||||
|
||||
explicit HttpConnection(Socket *socket);
|
||||
|
||||
void respond(int statuscode);
|
||||
|
||||
void respond(int statuscode, string payload);
|
||||
|
||||
void respond(int statuscode, FILE *file, bool compress = false, long start = -1, long end = -1);
|
||||
|
||||
void redirect(int statuscode, string location);
|
||||
|
||||
bool isExistingField(string index);
|
||||
|
||||
bool isExistingResponseField(string index);
|
||||
|
||||
string getField(string index);
|
||||
|
||||
string getResponseField(string index);
|
||||
|
||||
string getPath();
|
||||
|
||||
string getMethod();
|
||||
|
||||
void setField(string index, string data);
|
||||
|
||||
long getDuration();
|
||||
|
||||
HttpStatusCode getStatusCode();
|
||||
|
||||
string cgiExport();
|
||||
|
||||
void removeField(string index);
|
||||
};
|
||||
|
||||
#endif
|
@ -1,122 +0,0 @@
|
||||
/**
|
||||
* Necronda Web Server 3.0
|
||||
* HttpHeader.cpp - HttpHeader Class methods
|
||||
* Lorenz Stechauner, 2018-05-09
|
||||
*/
|
||||
|
||||
|
||||
#include <map>
|
||||
#include <iostream>
|
||||
#include "../Socket.h"
|
||||
|
||||
#include "HttpHeader.h"
|
||||
|
||||
|
||||
using namespace std;
|
||||
|
||||
string to_cgi(string text) {
|
||||
for (auto & c: text) c = (char) toupper(c);
|
||||
long pos = 0;
|
||||
while ((pos = text.find('-', pos + 1)) != string::npos) {
|
||||
text.replace(pos, 1, 1, '_');
|
||||
}
|
||||
return text;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Default Constructor
|
||||
*/
|
||||
HttpHeader::HttpHeader() {
|
||||
fields = fields;
|
||||
}
|
||||
|
||||
HttpHeader::HttpHeader(Socket *socket) : HttpHeader::HttpHeader() {
|
||||
parse(socket);
|
||||
}
|
||||
|
||||
|
||||
void HttpHeader::parse(Socket *socket) {
|
||||
while (true) {
|
||||
string line = socket->receiveLine();
|
||||
if (line.length() == 0) {
|
||||
break;
|
||||
} else {
|
||||
unsigned long pos = line.find(':');
|
||||
if (pos == string::npos) {
|
||||
throw (char *) "Malformed header";
|
||||
}
|
||||
string index = line.substr(0, pos);
|
||||
string data = line.substr(pos + 1, line.length() - pos);
|
||||
while (index[0] == ' ') index.erase(index.begin() + 0);
|
||||
while (index[index.length() - 1] == ' ') index.erase(index.end() - 1);
|
||||
while (data[0] == ' ') data.erase(data.begin() + 0);
|
||||
while (data[data.length() - 1] == ' ') data.erase(data.end() - 1);
|
||||
setField(index, data);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Default Destructor
|
||||
*/
|
||||
HttpHeader::~HttpHeader() {
|
||||
fields.clear();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Sets a field in the HTTP header
|
||||
* e.g. Content-Length: 42
|
||||
* @param index The field index
|
||||
* @param data The field data
|
||||
*/
|
||||
void HttpHeader::setField(string index, string data) {
|
||||
removeField(index);
|
||||
fields.insert(make_pair(index, data));
|
||||
}
|
||||
|
||||
void HttpHeader::removeField(string index) {
|
||||
fields.erase(index);
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets a field from the HTTP header
|
||||
* e.g. Content-Length: 42
|
||||
* @param index The field index
|
||||
* @return The field data
|
||||
*/
|
||||
string HttpHeader::getField(string index) {
|
||||
auto i = fields.find(index);
|
||||
if (i != fields.end()) {
|
||||
return fields.at(index);
|
||||
} else {
|
||||
return "";
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
bool HttpHeader::isExistingField(string index) {
|
||||
auto i = fields.find(index);
|
||||
return i != fields.end();
|
||||
}
|
||||
|
||||
string HttpHeader::toString() {
|
||||
string header = "";
|
||||
for (auto it = fields.begin(); it != fields.end(); it++ ) {
|
||||
header += it->first + ": " + it->second + "\r\n";
|
||||
}
|
||||
return header;
|
||||
}
|
||||
|
||||
string HttpHeader::cgiExport() {
|
||||
string header = "";
|
||||
for (auto it = fields.begin(); it != fields.end(); it++ ) {
|
||||
header += "HTTP_" + to_cgi(it->first) + "=" + cli_encode(it->second) + " ";
|
||||
}
|
||||
return header;
|
||||
}
|
||||
|
||||
|
||||
|
@ -1,53 +0,0 @@
|
||||
/**
|
||||
* Necronda Web Server 3.0
|
||||
* HttpHeader.h - HttpHeader Class definition
|
||||
* Lorenz Stechauner, 2018-05-09
|
||||
*/
|
||||
|
||||
#ifndef NECRONDA_HTTP_HEADER
|
||||
#define NECRONDA_HTTP_HEADER
|
||||
|
||||
#include <cstring>
|
||||
|
||||
using namespace std;
|
||||
|
||||
struct comp {
|
||||
bool operator()(const std::string& lhs, const std::string& rhs) const {
|
||||
return strcasecmp(lhs.c_str(), rhs.c_str()) < 0;
|
||||
}
|
||||
};
|
||||
|
||||
/**
|
||||
* Stores Key-Value Pairs for a HTTP header
|
||||
* e.g.
|
||||
* Content-Length: 64
|
||||
* Host: example.org
|
||||
*/
|
||||
class HttpHeader {
|
||||
private:
|
||||
map<string, string, comp> fields;
|
||||
|
||||
public:
|
||||
HttpHeader();
|
||||
|
||||
explicit HttpHeader(Socket *socket);
|
||||
|
||||
~HttpHeader();
|
||||
|
||||
void setField(string index, string data);
|
||||
|
||||
string getField(string index);
|
||||
|
||||
void removeField(string index);
|
||||
|
||||
bool isExistingField(string index);
|
||||
|
||||
void parse(Socket *socket);
|
||||
|
||||
string toString();
|
||||
|
||||
string cgiExport();
|
||||
|
||||
};
|
||||
|
||||
#endif
|
@ -1,107 +0,0 @@
|
||||
|
||||
|
||||
#include <string>
|
||||
#include <utility>
|
||||
#include <iostream>
|
||||
#include "../Socket.h"
|
||||
#include "HttpHeader.h"
|
||||
#include "HttpRequest.h"
|
||||
|
||||
|
||||
HttpRequest::HttpRequest() {
|
||||
this->header = HttpHeader();
|
||||
}
|
||||
|
||||
HttpRequest::HttpRequest(Socket *socket) : HttpRequest::HttpRequest() {
|
||||
parseHeader(socket);
|
||||
}
|
||||
|
||||
HttpRequest::HttpRequest(string method, string path, string version) : HttpRequest::HttpRequest() {
|
||||
this->method = std::move(method);
|
||||
this->path = std::move(path);
|
||||
this->version = std::move(version);
|
||||
}
|
||||
|
||||
void HttpRequest::parseHeader(Socket *socket) {
|
||||
string line = socket->receiveLine();
|
||||
|
||||
unsigned long pos1 = line.find(' ');
|
||||
unsigned long pos2;
|
||||
|
||||
bool invalid = false;
|
||||
|
||||
if (pos1 != string::npos) {
|
||||
pos2 = line.find(' ', pos1 + 1);
|
||||
if (pos2 != string::npos) {
|
||||
method = line.substr(0, pos1);
|
||||
for (auto &c: method) c = (char) toupper(c);
|
||||
path = line.substr(pos1 + 1, pos2 - pos1 - 1);
|
||||
version = line.substr(pos2 + 6, 3);
|
||||
} else {
|
||||
invalid = true;
|
||||
}
|
||||
} else {
|
||||
pos2 = string::npos;
|
||||
invalid = true;
|
||||
}
|
||||
|
||||
|
||||
if (!invalid && (line.substr(pos2 + 1, 5) != "HTTP/" || version[1] != '.' || path[0] != '/' || !(version[0] >= '0' && version[0] <= '9') || !(version[2] >= '0' && version[2] <= '9'))) {
|
||||
invalid = true;
|
||||
}
|
||||
|
||||
if (invalid) {
|
||||
method = "";
|
||||
path = "";
|
||||
version = "";
|
||||
throw (char *) "Malformed header";
|
||||
}
|
||||
|
||||
header.parse(socket);
|
||||
}
|
||||
|
||||
string HttpRequest::getMethod() {
|
||||
return method;
|
||||
}
|
||||
|
||||
string HttpRequest::getPath() {
|
||||
return path;
|
||||
}
|
||||
|
||||
string HttpRequest::getVersion() {
|
||||
return version;
|
||||
}
|
||||
|
||||
void HttpRequest::setMethod(string method) {
|
||||
this->method = std::move(method);
|
||||
}
|
||||
|
||||
void HttpRequest::setPath(string path) {
|
||||
this->path = std::move(path);
|
||||
}
|
||||
|
||||
void HttpRequest::setVersion(string version) {
|
||||
this->version = std::move(version);
|
||||
}
|
||||
|
||||
string HttpRequest::getField(string index) {
|
||||
return header.getField(std::move(index));
|
||||
}
|
||||
|
||||
void HttpRequest::setField(string index, string data) {
|
||||
header.setField(std::move(index), std::move(data));
|
||||
}
|
||||
|
||||
bool HttpRequest::isExistingField(string index) {
|
||||
return header.isExistingField(std::move(index));
|
||||
}
|
||||
|
||||
string HttpRequest::cgiExport() {
|
||||
return header.cgiExport();
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
@ -1,52 +0,0 @@
|
||||
/**
|
||||
* Necronda Web Server 3.0
|
||||
* HttpHeader.h - HttpHeader Class definition
|
||||
* Lorenz Stechauner, 2018-05-09
|
||||
*/
|
||||
|
||||
#ifndef NECRONDA_HTTP_REQUEST
|
||||
#define NECRONDA_HTTP_REQUEST
|
||||
|
||||
using namespace std;
|
||||
|
||||
class HttpRequest {
|
||||
private:
|
||||
HttpHeader header;
|
||||
string method;
|
||||
string path;
|
||||
string version;
|
||||
|
||||
public:
|
||||
HttpRequest();
|
||||
|
||||
explicit HttpRequest(Socket *socket);
|
||||
|
||||
HttpRequest(string method, string path, string version = "1.1");
|
||||
|
||||
void parseHeader(Socket *socket);
|
||||
|
||||
void sendHeader(Socket *socket);
|
||||
|
||||
string getField(string index);
|
||||
|
||||
void setField(string index, string data);
|
||||
|
||||
bool isExistingField(string index);
|
||||
|
||||
string getMethod();
|
||||
|
||||
string getPath();
|
||||
|
||||
string getVersion();
|
||||
|
||||
void setMethod(string method);
|
||||
|
||||
void setPath(string path);
|
||||
|
||||
void setVersion(string version);
|
||||
|
||||
string cgiExport();
|
||||
|
||||
};
|
||||
|
||||
#endif
|
@ -1,70 +0,0 @@
|
||||
//
|
||||
// Created by lorenz on 5/17/18.
|
||||
//
|
||||
|
||||
#include "HttpResponse.h"
|
||||
#include <utility>
|
||||
#include <iostream>
|
||||
#include "HttpStatusCode.h"
|
||||
|
||||
|
||||
HttpResponse::HttpResponse() {
|
||||
this->header = HttpHeader();
|
||||
}
|
||||
|
||||
HttpResponse::HttpResponse(Socket *socket) : HttpResponse::HttpResponse() {
|
||||
this->parseHeader(socket);
|
||||
}
|
||||
|
||||
HttpResponse::HttpResponse(int statuscode, string version) : HttpResponse::HttpResponse(::getStatusCode(statuscode), std::move(version)) {
|
||||
}
|
||||
|
||||
HttpResponse::HttpResponse(HttpStatusCode statuscode, string version) : HttpResponse::HttpResponse() {
|
||||
this->statuscode = statuscode;
|
||||
this->version = std::move(version);
|
||||
}
|
||||
|
||||
void HttpResponse::sendHeader(Socket *socket) {
|
||||
socket->send("HTTP/" + version + " " + to_string(statuscode.code) + " " + statuscode.message + "\r\n" +
|
||||
header.toString() + "\r\n");
|
||||
}
|
||||
|
||||
string HttpResponse::getField(string index) {
|
||||
return header.getField(std::move(index));
|
||||
}
|
||||
|
||||
void HttpResponse::setField(string index, string data) {
|
||||
header.setField(std::move(index), std::move(data));
|
||||
}
|
||||
|
||||
bool HttpResponse::isExistingField(string index) {
|
||||
return header.isExistingField(std::move(index));
|
||||
}
|
||||
|
||||
HttpStatusCode HttpResponse::getStatusCode() {
|
||||
return statuscode;
|
||||
}
|
||||
|
||||
string HttpResponse::getVersion() {
|
||||
return version;
|
||||
}
|
||||
|
||||
void HttpResponse::setStatusCode(HttpStatusCode statuscode) {
|
||||
this->statuscode = statuscode;
|
||||
}
|
||||
|
||||
void HttpResponse::setStatusCode(int statuscode) {
|
||||
this->statuscode = ::getStatusCode(statuscode);
|
||||
}
|
||||
|
||||
void HttpResponse::setVersion(string version) {
|
||||
this->version = std::move(version);
|
||||
}
|
||||
|
||||
void HttpResponse::parseHeader(Socket *socket) {
|
||||
|
||||
}
|
||||
|
||||
void HttpResponse::removeField(string index) {
|
||||
header.removeField(std::move(index));
|
||||
}
|
@ -1,50 +0,0 @@
|
||||
|
||||
|
||||
#ifndef NECRONDA_HTTP_RESPONSE
|
||||
#define NECRONDA_HTTP_RESPONSE
|
||||
|
||||
|
||||
#include <string>
|
||||
#include "HttpHeader.h"
|
||||
#include "HttpStatusCode.h"
|
||||
#include "../Socket.h"
|
||||
|
||||
class HttpResponse {
|
||||
private:
|
||||
HttpHeader header;
|
||||
HttpStatusCode statuscode;
|
||||
string version;
|
||||
|
||||
public:
|
||||
HttpResponse();
|
||||
|
||||
explicit HttpResponse(Socket *socket);
|
||||
|
||||
explicit HttpResponse(int statuscode, string version = "1.1");
|
||||
|
||||
explicit HttpResponse(HttpStatusCode statuscode, string version = "1.1");
|
||||
|
||||
void parseHeader(Socket *socket);
|
||||
|
||||
void sendHeader(Socket *socket);
|
||||
|
||||
string getField(string index);
|
||||
|
||||
void setField(string index, string data);
|
||||
|
||||
bool isExistingField(string index);
|
||||
|
||||
HttpStatusCode getStatusCode();
|
||||
|
||||
string getVersion();
|
||||
|
||||
void setStatusCode(HttpStatusCode statuscode);
|
||||
|
||||
void setStatusCode(int statuscode);
|
||||
|
||||
void setVersion(string version);
|
||||
|
||||
void removeField(string index);
|
||||
};
|
||||
|
||||
#endif
|
@ -1,66 +0,0 @@
|
||||
#include "HttpStatusCode.h"
|
||||
|
||||
/**
|
||||
* Necronda Web Server 3.0
|
||||
* HttpStatusCode.cpp - HTTP Status Code definition
|
||||
* Lorenz Stechauner, 2018-05-16
|
||||
* Reference: https://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html
|
||||
*/
|
||||
|
||||
|
||||
HttpStatusCode httpStatusCodes[] = {
|
||||
HttpStatusCode{100, "Informational", "Continue", ""},
|
||||
HttpStatusCode{101, "Informational", "Switching Protocols", ""},
|
||||
|
||||
HttpStatusCode{200, "Success", "OK", ""},
|
||||
HttpStatusCode{201, "Success", "Created", ""},
|
||||
HttpStatusCode{202, "Success", "Accepted", ""},
|
||||
HttpStatusCode{203, "Success", "Non-Authoritative Information", ""},
|
||||
HttpStatusCode{204, "Success", "No Content", ""},
|
||||
HttpStatusCode{205, "Success", "Reset Content", ""},
|
||||
HttpStatusCode{206, "Success", "Partial Content", ""},
|
||||
|
||||
HttpStatusCode{300, "Redirection", "Multiple Choices", ""},
|
||||
HttpStatusCode{301, "Redirection", "Moved Permanently", ""},
|
||||
HttpStatusCode{302, "Redirection", "Found", ""},
|
||||
HttpStatusCode{303, "Redirection", "See Other", ""},
|
||||
HttpStatusCode{304, "Redirection", "Not Modified", ""},
|
||||
HttpStatusCode{305, "Redirection", "Use Proxy", ""},
|
||||
HttpStatusCode{307, "Redirection", "Temporary Redirect", ""},
|
||||
HttpStatusCode{308, "Redirection", "Permanent Redirect", ""},
|
||||
|
||||
HttpStatusCode{400, "Client Error", "Bad Request", ""},
|
||||
HttpStatusCode{401, "Client Error", "Unauthorized", ""},
|
||||
HttpStatusCode{402, "Client Error", "Payment Required", ""},
|
||||
HttpStatusCode{403, "Client Error", "Forbidden", ""},
|
||||
HttpStatusCode{404, "Client Error", "Not Found", ""},
|
||||
HttpStatusCode{405, "Client Error", "Method Not Allowed", ""},
|
||||
HttpStatusCode{406, "Client Error", "Not Acceptable", ""},
|
||||
HttpStatusCode{407, "Client Error", "Proxy Authentication Required", ""},
|
||||
HttpStatusCode{408, "Client Error", "Request Timeout", ""},
|
||||
HttpStatusCode{409, "Client Error", "Conflict", ""},
|
||||
HttpStatusCode{410, "Client Error", "Gone", ""},
|
||||
HttpStatusCode{411, "Client Error", "Length Required", ""},
|
||||
HttpStatusCode{412, "Client Error", "Precondition Failed", ""},
|
||||
HttpStatusCode{413, "Client Error", "Request Entity Too Large", ""},
|
||||
HttpStatusCode{414, "Client Error", "Request-URI Too Long", ""},
|
||||
HttpStatusCode{415, "Client Error", "Unsupported Media Type", ""},
|
||||
HttpStatusCode{416, "Client Error", "Requested Range Not Satisfiable", ""},
|
||||
HttpStatusCode{417, "Client Error", "Expectation Failed", ""},
|
||||
|
||||
HttpStatusCode{500, "Server Error", "Internal Server Error", ""},
|
||||
HttpStatusCode{501, "Server Error", "Not Implemented", ""},
|
||||
HttpStatusCode{502, "Server Error", "Bad Gateway", ""},
|
||||
HttpStatusCode{503, "Server Error", "Service Unavailable", ""},
|
||||
HttpStatusCode{504, "Server Error", "Gateway Timeout", ""},
|
||||
HttpStatusCode{505, "Server Error", "HTTP Version Not Supported", ""},
|
||||
};
|
||||
|
||||
HttpStatusCode getStatusCode(int statuscode) {
|
||||
for (HttpStatusCode sc : httpStatusCodes) {
|
||||
if (sc.code == statuscode) {
|
||||
return sc;
|
||||
}
|
||||
}
|
||||
throw (char *) "Invalid status code";
|
||||
}
|
@ -1,15 +0,0 @@
|
||||
|
||||
|
||||
#ifndef NECRONDA_HTTP_STATUSCODE
|
||||
#define NECRONDA_HTTP_STATUSCODE
|
||||
|
||||
typedef struct {
|
||||
short code; // The status code (e.g. 200)
|
||||
const char *type; // The status type type (e.g Success)
|
||||
const char *message; // The status code message (e.g. OK)
|
||||
const char *description; // The status code description (currently not used)
|
||||
} HttpStatusCode;
|
||||
|
||||
HttpStatusCode getStatusCode(int statuscode);
|
||||
|
||||
#endif
|
@ -1,41 +0,0 @@
|
||||
//
|
||||
// Created by lorenz on 5/30/18.
|
||||
//
|
||||
|
||||
#include "procopen.h"
|
||||
|
||||
stds procopen(const char* command) {
|
||||
|
||||
int pipes[3][2];
|
||||
|
||||
pipe(pipes[PARENT_READ_PIPE]);
|
||||
pipe(pipes[PARENT_WRITE_PIPE]);
|
||||
pipe(pipes[PARENT_ERROR_PIPE]);
|
||||
|
||||
int pid = fork();
|
||||
|
||||
if(pid == 0) {
|
||||
dup2(CHILD_READ_FD, STDIN_FILENO);
|
||||
dup2(CHILD_WRITE_FD, STDOUT_FILENO);
|
||||
dup2(CHILD_ERROR_FD, STDERR_FILENO);
|
||||
|
||||
close(CHILD_READ_FD);
|
||||
close(CHILD_WRITE_FD);
|
||||
close(CHILD_ERROR_FD);
|
||||
|
||||
close(PARENT_READ_FD);
|
||||
close(PARENT_WRITE_FD);
|
||||
close(PARENT_ERROR_FD);
|
||||
|
||||
system(command);
|
||||
exit(0);
|
||||
} else {
|
||||
close(CHILD_READ_FD);
|
||||
close(CHILD_WRITE_FD);
|
||||
close(CHILD_ERROR_FD);
|
||||
|
||||
return stds{fdopen(PARENT_WRITE_FD, "w"), fdopen(PARENT_READ_FD, "r"), fdopen(PARENT_ERROR_FD, "r"), (pid_t) pid};
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -1,41 +0,0 @@
|
||||
//
|
||||
// Created by lorenz on 5/30/18.
|
||||
//
|
||||
|
||||
|
||||
#include <zconf.h>
|
||||
#include <cstdlib>
|
||||
#include <cstdio>
|
||||
|
||||
#ifndef NECRONDA_PROCOPEN
|
||||
#define NECRONDA_PROCOPEN
|
||||
|
||||
|
||||
#define PARENT_WRITE_PIPE 0
|
||||
#define PARENT_READ_PIPE 1
|
||||
#define PARENT_ERROR_PIPE 2
|
||||
|
||||
#define READ_FD 0
|
||||
#define WRITE_FD 1
|
||||
|
||||
#define PARENT_READ_FD ( pipes[PARENT_READ_PIPE][READ_FD] )
|
||||
#define PARENT_WRITE_FD ( pipes[PARENT_WRITE_PIPE][WRITE_FD] )
|
||||
#define PARENT_ERROR_FD ( pipes[PARENT_ERROR_PIPE][READ_FD] )
|
||||
|
||||
#define CHILD_READ_FD ( pipes[PARENT_WRITE_PIPE][READ_FD] )
|
||||
#define CHILD_WRITE_FD ( pipes[PARENT_READ_PIPE][WRITE_FD] )
|
||||
#define CHILD_ERROR_FD ( pipes[PARENT_ERROR_PIPE][WRITE_FD] )
|
||||
|
||||
|
||||
typedef struct {
|
||||
FILE* stdin;
|
||||
FILE* stdout;
|
||||
FILE* stderr;
|
||||
pid_t pid;
|
||||
} stds;
|
||||
|
||||
|
||||
stds procopen(const char* command);
|
||||
|
||||
|
||||
#endif
|
325
src/server.c
Normal file
325
src/server.c
Normal file
@ -0,0 +1,325 @@
|
||||
/**
|
||||
* Sesimos - secure, simple, modern web server
|
||||
* @brief Main executable
|
||||
* @file src/server.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-03
|
||||
*/
|
||||
|
||||
#include "defs.h"
|
||||
#include "server.h"
|
||||
#include "logger.h"
|
||||
#include "async.h"
|
||||
|
||||
#include "cache_handler.h"
|
||||
#include "lib/config.h"
|
||||
#include "lib/proxy.h"
|
||||
#include "lib/geoip.h"
|
||||
#include "workers.h"
|
||||
#include "worker/func.h"
|
||||
#include "lib/list.h"
|
||||
#include "lib/utils.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <getopt.h>
|
||||
#include <sys/socket.h>
|
||||
#include <signal.h>
|
||||
#include <unistd.h>
|
||||
#include <string.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <sys/types.h>
|
||||
#include <openssl/err.h>
|
||||
#include <openssl/pem.h>
|
||||
#include <openssl/ssl.h>
|
||||
#include <openssl/conf.h>
|
||||
|
||||
|
||||
const char *config_file;
|
||||
|
||||
static int sockets[NUM_SOCKETS];
|
||||
static SSL_CTX *contexts[CONFIG_MAX_CERT_CONFIG];
|
||||
|
||||
static client_ctx_t **clients;
|
||||
|
||||
static const char *color_table[] = {"\x1B[31m", "\x1B[32m", "\x1B[33m", "\x1B[34m", "\x1B[35m", "\x1B[36m"};
|
||||
|
||||
static int clean() {
|
||||
remove("/var/sesimos/server/cache");
|
||||
rmdir("/var/sesimos/server/");
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int ssl_servername_cb(SSL *ssl, int *ad, void *arg) {
|
||||
const char *servername = SSL_get_servername(ssl, TLSEXT_NAMETYPE_host_name);
|
||||
if (servername != NULL) {
|
||||
const host_config_t *conf = get_host_config(servername);
|
||||
if (conf != NULL) SSL_set_SSL_CTX(ssl, contexts[conf->cert]);
|
||||
}
|
||||
return SSL_TLSEXT_ERR_OK;
|
||||
}
|
||||
|
||||
void server_free_client(client_ctx_t *ctx) {
|
||||
for (int i = 0; i < list_size(clients); i++) {
|
||||
if (clients[i] == ctx) {
|
||||
clients = list_remove(clients, i);
|
||||
break;
|
||||
}
|
||||
}
|
||||
free(ctx);
|
||||
}
|
||||
|
||||
static void accept_cb(void *arg) {
|
||||
int i = (int) (((int *) arg) - sockets);
|
||||
int fd = sockets[i];
|
||||
|
||||
client_ctx_t *client_ctx = malloc(sizeof(client_ctx_t));
|
||||
if (client_ctx == NULL) {
|
||||
critical("Unable to allocate memory for client context");
|
||||
errno = 0;
|
||||
return;
|
||||
}
|
||||
|
||||
clients = list_append(clients, &client_ctx);
|
||||
if (clients == NULL) {
|
||||
critical("Unable to add client context to list");
|
||||
errno = 0;
|
||||
return;
|
||||
}
|
||||
|
||||
client_ctx->in_use = 1;
|
||||
sock *client = &client_ctx->socket;
|
||||
|
||||
client->ctx = contexts[0];
|
||||
socklen_t addr_len = sizeof(client->_addr);
|
||||
int client_fd = accept(fd, &client->_addr.sock, &addr_len);
|
||||
if (client_fd < 0) {
|
||||
critical("Unable to accept connection");
|
||||
return;
|
||||
}
|
||||
|
||||
client->socket = client_fd;
|
||||
client->enc = (i == 1);
|
||||
client->ts_start = clock_micros();
|
||||
client->ts_last = client->ts_start;
|
||||
client_ctx->cnx_s = client->ts_start;
|
||||
client_ctx->cnx_e = -1, client_ctx->req_s = -1, client_ctx->req_e = -1, client_ctx->res_ts = -1;
|
||||
|
||||
tcp_accept(client_ctx);
|
||||
}
|
||||
|
||||
static void accept_err_cb(void *arg) {
|
||||
int i = (int) (((int *) arg) - sockets);
|
||||
int fd = sockets[i];
|
||||
// TODO accept error callback
|
||||
}
|
||||
|
||||
static void terminate_forcefully(int sig) {
|
||||
fprintf(stderr, "\n");
|
||||
notice("Terminating forcefully!");
|
||||
|
||||
geoip_free();
|
||||
|
||||
notice("Goodbye");
|
||||
exit(2);
|
||||
}
|
||||
|
||||
static void terminate_gracefully(int sig) {
|
||||
fprintf(stderr, "\n");
|
||||
notice("Terminating gracefully...");
|
||||
|
||||
struct sigaction act = {0};
|
||||
act.sa_handler = terminate_forcefully;
|
||||
sigaction(SIGINT, &act, NULL);
|
||||
sigaction(SIGTERM, &act, NULL);
|
||||
|
||||
for (int i = 0; i < NUM_SOCKETS; i++) {
|
||||
close(sockets[i]);
|
||||
}
|
||||
|
||||
cache_stop();
|
||||
workers_stop();
|
||||
workers_destroy();
|
||||
|
||||
for (int i = 0; i < list_size(clients); i++) {
|
||||
tcp_close(clients[i]);
|
||||
}
|
||||
proxy_close_all();
|
||||
logger_set_prefix("");
|
||||
|
||||
async_stop();
|
||||
}
|
||||
|
||||
static void nothing(int sig) {}
|
||||
|
||||
int main(int argc, char *const argv[]) {
|
||||
const int YES = 1;
|
||||
int ret;
|
||||
|
||||
memset(sockets, 0, sizeof(sockets));
|
||||
clients = list_create(sizeof(void *), 64);
|
||||
if (clients == NULL) {
|
||||
critical("Unable to initialize client list");
|
||||
return 1;
|
||||
}
|
||||
|
||||
const struct sockaddr_in6 addresses[2] = {
|
||||
{.sin6_family = AF_INET6, .sin6_addr = IN6ADDR_ANY_INIT, .sin6_port = htons(80)},
|
||||
{.sin6_family = AF_INET6, .sin6_addr = IN6ADDR_ANY_INIT, .sin6_port = htons(443)}
|
||||
};
|
||||
|
||||
logger_init();
|
||||
|
||||
logger_set_name("server");
|
||||
|
||||
if (setvbuf(stdout, NULL, _IOLBF, 0) != 0 || setvbuf(stderr, NULL, _IOLBF, 0) != 0) {
|
||||
critical("Unable to set stdout/stderr to line-buffered mode");
|
||||
return 1;
|
||||
}
|
||||
printf("sesimos web server " SERVER_VERSION "\n");
|
||||
|
||||
static const struct option long_opts[] = {
|
||||
{"help", no_argument, 0, 'h'},
|
||||
{"config", required_argument, 0, 'c'},
|
||||
{ 0, 0, 0, 0 }
|
||||
};
|
||||
|
||||
config_file = NULL;
|
||||
int c, opt_idx;
|
||||
while ((c = getopt_long(argc, argv, "hc:", long_opts, &opt_idx)) != -1) {
|
||||
switch (c) {
|
||||
case 'h':
|
||||
fprintf(stderr,
|
||||
"Usage: sesimos [-h] [-c <CONFIG FILE>]\n"
|
||||
"\n"
|
||||
"Options:\n"
|
||||
" -c, --config <CONFIG-FILE> path to the config file. If not provided, default will be used\n"
|
||||
" -h, --help print this dialogue\n");
|
||||
return 0;
|
||||
case 'c':
|
||||
config_file = optarg;
|
||||
break;
|
||||
case '?':
|
||||
default:
|
||||
critical("Unable to parse arguments");
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (optind != argc) {
|
||||
critical("No positional arguments expected");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (config_load(config_file == NULL ? DEFAULT_CONFIG_FILE : config_file) != 0)
|
||||
return 1;
|
||||
|
||||
if ((sockets[0] = socket(AF_INET6, SOCK_STREAM, 0)) == -1 ||
|
||||
(sockets[1] = socket(AF_INET6, SOCK_STREAM, 0)) == -1)
|
||||
{
|
||||
critical("Unable to create socket");
|
||||
return 1;
|
||||
}
|
||||
|
||||
for (int i = 0; i < NUM_SOCKETS; i++) {
|
||||
if (setsockopt(sockets[i], SOL_SOCKET, SO_REUSEADDR, &YES, sizeof(YES)) < 0) {
|
||||
critical("Unable to set options for socket %i", i);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (bind(sockets[0], (struct sockaddr *) &addresses[0], sizeof(addresses[0])) == -1 ||
|
||||
bind(sockets[1], (struct sockaddr *) &addresses[1], sizeof(addresses[1])) == -1)
|
||||
{
|
||||
critical("Unable to bind socket to address");
|
||||
return 1;
|
||||
}
|
||||
|
||||
struct sigaction act = {0};
|
||||
act.sa_handler = terminate_gracefully;
|
||||
sigaction(SIGINT, &act, NULL);
|
||||
sigaction(SIGTERM, &act, NULL);
|
||||
act.sa_handler = nothing;
|
||||
sigaction(SIGUSR1, &act, NULL);
|
||||
sigaction(SIGPIPE, &act, NULL);
|
||||
|
||||
if ((ret = geoip_init(config.geoip_dir)) != 0) {
|
||||
if (ret == -1) {
|
||||
critical("Unable to initialize geoip");
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
if ((ret = cache_init()) != 0) {
|
||||
geoip_free();
|
||||
if (ret == -1) critical("Unable to initialize cache");
|
||||
return 1;
|
||||
}
|
||||
|
||||
for (int i = 0; i < CONFIG_MAX_CERT_CONFIG; i++) {
|
||||
const cert_config_t *conf = &config.certs[i];
|
||||
if (conf->name[0] == 0) break;
|
||||
|
||||
contexts[i] = SSL_CTX_new(TLS_server_method());
|
||||
SSL_CTX *ctx = contexts[i];
|
||||
SSL_CTX_set_options(ctx, SSL_OP_SINGLE_DH_USE);
|
||||
SSL_CTX_set_verify(ctx, SSL_VERIFY_NONE, NULL);
|
||||
SSL_CTX_set_min_proto_version(ctx, TLS1_2_VERSION);
|
||||
SSL_CTX_set_mode(ctx, SSL_MODE_ENABLE_PARTIAL_WRITE);
|
||||
SSL_CTX_set_cipher_list(ctx, "HIGH:!aNULL:!kRSA:!PSK:!SRP:!MD5:!RC4");
|
||||
SSL_CTX_set_ecdh_auto(ctx, 1);
|
||||
SSL_CTX_set_tlsext_servername_callback(ctx, ssl_servername_cb);
|
||||
|
||||
if (SSL_CTX_use_certificate_chain_file(ctx, conf->full_chain) != 1) {
|
||||
critical("Unable to load certificate chain file: %s: %s", ERR_reason_error_string(ERR_get_error()), conf->full_chain);
|
||||
geoip_free();
|
||||
return 1;
|
||||
}
|
||||
if (SSL_CTX_use_PrivateKey_file(ctx, conf->priv_key, SSL_FILETYPE_PEM) != 1) {
|
||||
critical("Unable to load private key file: %s: %s", ERR_reason_error_string(ERR_get_error()), conf->priv_key);
|
||||
geoip_free();
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (async_init() != 0) {
|
||||
critical("Unable to initialize async thread");
|
||||
geoip_free();
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (proxy_preload() != 0) {
|
||||
critical("Unable to initialize proxy");
|
||||
geoip_free();
|
||||
async_free();
|
||||
return 1;
|
||||
}
|
||||
|
||||
for (int i = 0; i < NUM_SOCKETS; i++) {
|
||||
if (listen(sockets[i], LISTEN_BACKLOG) < 0) {
|
||||
critical("Unable to listen on socket %i", i);
|
||||
geoip_free();
|
||||
proxy_unload();
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
workers_init();
|
||||
|
||||
for (int i = 0; i < NUM_SOCKETS; i++) {
|
||||
async_fd(sockets[i], ASYNC_WAIT_READ, ASYNC_KEEP, &sockets[i], accept_cb, accept_err_cb, accept_err_cb);
|
||||
}
|
||||
|
||||
notice("Ready to accept connections");
|
||||
|
||||
async_thread();
|
||||
|
||||
notice("Goodbye!");
|
||||
|
||||
// cleanup
|
||||
list_free(clients);
|
||||
geoip_free();
|
||||
proxy_unload();
|
||||
cache_join();
|
||||
async_free();
|
||||
return 0;
|
||||
}
|
26
src/server.h
Normal file
26
src/server.h
Normal file
@ -0,0 +1,26 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Main executable (header file)
|
||||
* @file src/server.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2020-12-03
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_SERVER_H
|
||||
#define SESIMOS_SERVER_H
|
||||
|
||||
#include "worker/func.h"
|
||||
|
||||
#define NUM_SOCKETS 2
|
||||
#define LISTEN_BACKLOG 16
|
||||
#define REQ_PER_CONNECTION 200
|
||||
#define CLIENT_TIMEOUT 3600
|
||||
#define SERVER_TIMEOUT_INIT 4
|
||||
#define SERVER_TIMEOUT 3600
|
||||
|
||||
#define CNX_HANDLER_WORKERS 8
|
||||
#define REQ_HANDLER_WORKERS 16
|
||||
|
||||
void server_free_client(client_ctx_t *ctx);
|
||||
|
||||
#endif //SESIMOS_SERVER_H
|
175
src/worker/fastcgi_handler.c
Normal file
175
src/worker/fastcgi_handler.c
Normal file
@ -0,0 +1,175 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief FastCGI handler
|
||||
* @file src/worker/fastcgi_handler.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-28
|
||||
*/
|
||||
|
||||
#include "func.h"
|
||||
#include "../logger.h"
|
||||
#include "../lib/utils.h"
|
||||
#include "../lib/compress.h"
|
||||
#include "../workers.h"
|
||||
#include "../lib/fastcgi.h"
|
||||
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
|
||||
static int fastcgi_handler_1(client_ctx_t *ctx, fastcgi_cnx_t *fcgi_cnx);
|
||||
static int fastcgi_handler_2(client_ctx_t *ctx, fastcgi_cnx_t *fcgi_cnx);
|
||||
|
||||
void fastcgi_handler_func(client_ctx_t *ctx) {
|
||||
logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, ctx->req_host, CLR_STR, ctx->log_prefix);
|
||||
|
||||
fastcgi_cnx_t fcgi_cnx;
|
||||
int ret = fastcgi_handler_1(ctx, &fcgi_cnx);
|
||||
respond(ctx);
|
||||
if (ret == 0) fastcgi_handler_2(ctx, &fcgi_cnx);
|
||||
request_complete(ctx);
|
||||
|
||||
handle_request(ctx);
|
||||
}
|
||||
|
||||
static int fastcgi_handler_1(client_ctx_t *ctx, fastcgi_cnx_t *fcgi_cnx) {
|
||||
http_res *res = &ctx->res;
|
||||
http_req *req = &ctx->req;
|
||||
http_uri *uri = &ctx->uri;
|
||||
sock *client = &ctx->socket;
|
||||
char *err_msg = ctx->err_msg;
|
||||
|
||||
fcgi_cnx->socket = 0;
|
||||
fcgi_cnx->req_id = 0;
|
||||
fcgi_cnx->r_addr = ctx->socket.addr;
|
||||
fcgi_cnx->r_host = (ctx->host[0] != 0) ? ctx->host : NULL;
|
||||
|
||||
char buf[1024];
|
||||
|
||||
int mode, ret;
|
||||
if (strcmp(uri->filename + strlen(uri->filename) - 4, ".ncr") == 0) {
|
||||
mode = FASTCGI_SESIMOS;
|
||||
} else if (strcmp(uri->filename + strlen(uri->filename) - 4, ".php") == 0) {
|
||||
mode = FASTCGI_PHP;
|
||||
} else {
|
||||
res->status = http_get_status(500);
|
||||
error("Invalid FastCGI extension: %s", uri->filename);
|
||||
return 0;
|
||||
}
|
||||
|
||||
struct stat statbuf;
|
||||
stat(uri->filename, &statbuf);
|
||||
char *last_modified = http_format_date(statbuf.st_mtime, buf, sizeof(buf));
|
||||
http_add_header_field(&res->hdr, "Last-Modified", last_modified);
|
||||
|
||||
res->status = http_get_status(200);
|
||||
if (fastcgi_init(fcgi_cnx, mode, ctx->req_num, client, req, uri) != 0) {
|
||||
res->status = http_get_status(503);
|
||||
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
||||
return 2;
|
||||
}
|
||||
|
||||
const char *client_content_length = http_get_header_field(&req->hdr, "Content-Length");
|
||||
const char *client_transfer_encoding = http_get_header_field(&req->hdr, "Transfer-Encoding");
|
||||
if (client_content_length != NULL) {
|
||||
unsigned long client_content_len = strtoul(client_content_length, NULL, 10);
|
||||
ret = fastcgi_receive(fcgi_cnx, client, client_content_len);
|
||||
} else if (client_transfer_encoding != NULL && strstr(client_transfer_encoding, "chunked") != NULL) {
|
||||
ret = fastcgi_receive_chunked(fcgi_cnx, client);
|
||||
} else {
|
||||
ret = 0;
|
||||
}
|
||||
if (ret != 0) {
|
||||
if (ret < 0) {
|
||||
return -1;
|
||||
} else {
|
||||
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
||||
}
|
||||
res->status = http_get_status(502);
|
||||
return 2;
|
||||
}
|
||||
fastcgi_close_stdin(fcgi_cnx);
|
||||
|
||||
ret = fastcgi_header(fcgi_cnx, res, err_msg);
|
||||
if (ret != 0) {
|
||||
return (ret < 0) ? -1 : 1;
|
||||
}
|
||||
|
||||
const char *status_hdr = http_get_header_field(&res->hdr, "Status");
|
||||
if (status_hdr != NULL) {
|
||||
int status_code = (int) strtoul(status_hdr, NULL, 10);
|
||||
res->status = http_get_status(status_code);
|
||||
http_remove_header_field(&res->hdr, "Status", HTTP_REMOVE_ALL);
|
||||
if (res->status == NULL && status_code >= 100 && status_code <= 999) {
|
||||
ctx->custom_status.code = status_code;
|
||||
ctx->custom_status.type = 0;
|
||||
strcpy(ctx->custom_status.msg, status_hdr + 4);
|
||||
res->status = &ctx->custom_status;
|
||||
} else if (res->status == NULL) {
|
||||
res->status = http_get_status(500);
|
||||
sprintf(err_msg, "The status_hdr code was set to an invalid or unknown value.");
|
||||
return 2;
|
||||
}
|
||||
}
|
||||
|
||||
const char *content_length_f = http_get_header_field(&res->hdr, "Content-Length");
|
||||
ctx->content_length = (content_length_f == NULL) ? -1 : strtol(content_length_f, NULL, 10);
|
||||
|
||||
const char *content_type = http_get_header_field(&res->hdr, "Content-Type");
|
||||
const char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding");
|
||||
if (content_encoding == NULL &&
|
||||
content_type != NULL &&
|
||||
strncmp(content_type, "text/html", 9) == 0 &&
|
||||
ctx->content_length != -1 &&
|
||||
ctx->content_length <= sizeof(ctx->msg_content) - 1)
|
||||
{
|
||||
fastcgi_dump(fcgi_cnx, ctx->msg_content, sizeof(ctx->msg_content));
|
||||
return 1;
|
||||
}
|
||||
|
||||
ctx->use_fastcgi = 1;
|
||||
|
||||
if (ctx->content_length != -1 && ctx->content_length < 1024000) {
|
||||
ctx->use_fastcgi |= FASTCGI_COMPRESS_HOLD;
|
||||
}
|
||||
|
||||
ctx->content_length = -1;
|
||||
|
||||
int http_comp = http_get_compression(req, res);
|
||||
if (http_comp & COMPRESS) {
|
||||
if (http_comp & COMPRESS_BR) {
|
||||
ctx->use_fastcgi |= FASTCGI_COMPRESS_BR;
|
||||
sprintf(buf, "br");
|
||||
} else if (http_comp & COMPRESS_GZ) {
|
||||
ctx->use_fastcgi |= FASTCGI_COMPRESS_GZ;
|
||||
sprintf(buf, "gzip");
|
||||
}
|
||||
http_add_header_field(&res->hdr, "Vary", "Accept-Encoding");
|
||||
http_add_header_field(&res->hdr, "Content-Encoding", buf);
|
||||
http_remove_header_field(&res->hdr, "Content-Length", HTTP_REMOVE_ALL);
|
||||
}
|
||||
|
||||
if (http_get_header_field(&res->hdr, "Content-Length") == NULL) {
|
||||
http_add_header_field(&res->hdr, "Transfer-Encoding", "chunked");
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int fastcgi_handler_2(client_ctx_t *ctx, fastcgi_cnx_t *fcgi_cnx) {
|
||||
const char *transfer_encoding = http_get_header_field(&ctx->res.hdr, "Transfer-Encoding");
|
||||
int chunked = (transfer_encoding != NULL && strstr(transfer_encoding, "chunked") != NULL);
|
||||
|
||||
int flags = (chunked ? FASTCGI_CHUNKED : 0) | (ctx->use_fastcgi & (FASTCGI_COMPRESS | FASTCGI_COMPRESS_HOLD));
|
||||
int ret = fastcgi_send(fcgi_cnx, &ctx->socket, flags);
|
||||
if (ret < 0) {
|
||||
ctx->c_keep_alive = 0;
|
||||
errno = 0;
|
||||
}
|
||||
|
||||
if (fcgi_cnx->socket != 0) {
|
||||
close(fcgi_cnx->socket);
|
||||
fcgi_cnx->socket = 0;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
71
src/worker/func.h
Normal file
71
src/worker/func.h
Normal file
@ -0,0 +1,71 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Worker function header file
|
||||
* @file src/worker/func.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-29
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_FUNC_H
|
||||
#define SESIMOS_FUNC_H
|
||||
|
||||
#include "../lib/sock.h"
|
||||
#include "../lib/http.h"
|
||||
#include "../lib/uri.h"
|
||||
#include "../lib/config.h"
|
||||
#include "../lib/proxy.h"
|
||||
|
||||
typedef struct {
|
||||
sock socket;
|
||||
int req_num;
|
||||
unsigned char in_use: 1, s_keep_alive:1, c_keep_alive:1, use_fastcgi:4, use_proxy:2, ws_close:2;
|
||||
char cc[3], host[256];
|
||||
char req_host[256], err_msg[256];
|
||||
char log_prefix[128];
|
||||
char _c_addr[INET6_ADDRSTRLEN + 1], _s_addr[INET6_ADDRSTRLEN + 1];
|
||||
long cnx_s, cnx_e, req_s, res_ts, req_e;
|
||||
http_req req;
|
||||
http_res res;
|
||||
http_uri uri;
|
||||
http_status_ctx status;
|
||||
http_status custom_status;
|
||||
host_config_t *conf;
|
||||
FILE *file;
|
||||
long content_length;
|
||||
char *msg_buf, *msg_buf_ptr, msg_content[1024];
|
||||
proxy_ctx_t *proxy;
|
||||
} client_ctx_t;
|
||||
|
||||
typedef struct {
|
||||
client_ctx_t *client;
|
||||
sock *socket;
|
||||
void *other;
|
||||
} ws_ctx_t;
|
||||
|
||||
void tcp_acceptor_func(client_ctx_t *ctx);
|
||||
|
||||
void request_handler_func(client_ctx_t *ctx);
|
||||
|
||||
void local_handler_func(client_ctx_t *ctx);
|
||||
|
||||
void fastcgi_handler_func(client_ctx_t *ctx);
|
||||
|
||||
void proxy_handler_func(client_ctx_t *ctx);
|
||||
|
||||
void ws_frame_handler_func(ws_ctx_t *ctx);
|
||||
|
||||
int respond(client_ctx_t *ctx);
|
||||
|
||||
void request_complete(client_ctx_t *ctx);
|
||||
|
||||
void timeout_request(client_ctx_t *ctx);
|
||||
|
||||
void tcp_close(client_ctx_t *ctx);
|
||||
|
||||
void proxy_close(proxy_ctx_t *ctx);
|
||||
|
||||
int ws_handle_connection(client_ctx_t *ctx);
|
||||
|
||||
void ws_close(ws_ctx_t *ctx);
|
||||
|
||||
#endif //SESIMOS_FUNC_H
|
217
src/worker/local_handler.c
Normal file
217
src/worker/local_handler.c
Normal file
@ -0,0 +1,217 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Local filesystem handler
|
||||
* @file src/worker/local_handler.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-29
|
||||
*/
|
||||
|
||||
#include "func.h"
|
||||
#include "../logger.h"
|
||||
#include "../lib/utils.h"
|
||||
#include "../lib/compress.h"
|
||||
#include "../workers.h"
|
||||
#include "../lib/list.h"
|
||||
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
|
||||
static int local_handler(client_ctx_t *ctx);
|
||||
|
||||
void local_handler_func(client_ctx_t *ctx) {
|
||||
logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, ctx->req_host, CLR_STR, ctx->log_prefix);
|
||||
|
||||
switch (local_handler(ctx)) {
|
||||
case 0:
|
||||
respond(ctx);
|
||||
request_complete(ctx);
|
||||
handle_request(ctx);
|
||||
break;
|
||||
case 1:
|
||||
fastcgi_handle(ctx);
|
||||
break;
|
||||
default:
|
||||
tcp_close(ctx);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static int local_handler(client_ctx_t *ctx) {
|
||||
http_res *res = &ctx->res;
|
||||
http_req *req = &ctx->req;
|
||||
http_uri *uri = &ctx->uri;
|
||||
char *err_msg = ctx->err_msg;
|
||||
|
||||
char buf1[1024], buf2[1024];
|
||||
int accept_if_modified_since = 0;
|
||||
|
||||
if (strcmp(req->method, "TRACE") == 0) {
|
||||
res->status = http_get_status(200);
|
||||
http_add_header_field(&res->hdr, "Content-Type", "message/http");
|
||||
|
||||
ctx->msg_buf_ptr = malloc(4096);
|
||||
ctx->msg_buf = ctx->msg_buf_ptr;
|
||||
ctx->content_length = snprintf(ctx->msg_buf, 4096 - ctx->content_length, "%s %s HTTP/%s\r\n", req->method, req->uri, req->version);
|
||||
for (int i = 0; i < list_size(&req->hdr); i++) {
|
||||
const http_field *f = &req->hdr.fields[i];
|
||||
ctx->content_length += snprintf(ctx->msg_buf + ctx->content_length, 4096 - ctx->content_length, "%s: %s\r\n", http_field_get_name(f), http_field_get_value(f));
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (strncmp(uri->req_path, "/.well-known/", 13) == 0) {
|
||||
http_add_header_field(&res->hdr, "Access-Control-Allow-Origin", "*");
|
||||
}
|
||||
|
||||
if (strncmp(uri->req_path, "/.well-known/", 13) != 0 && strstr(uri->path, "/.") != NULL) {
|
||||
res->status = http_get_status(403);
|
||||
sprintf(err_msg, "Parts of this URI are hidden.");
|
||||
return 0;
|
||||
} else if (uri->filename == NULL && (int) uri->is_static && (int) uri->is_dir && strlen(uri->pathinfo) == 0) {
|
||||
res->status = http_get_status(403);
|
||||
sprintf(err_msg, "It is not allowed to list the contents of this directory.");
|
||||
return 0;
|
||||
} else if (uri->filename == NULL && (int) !uri->is_static && (int) uri->is_dir && strlen(uri->pathinfo) == 0) {
|
||||
// TODO list directory contents
|
||||
res->status = http_get_status(501);
|
||||
sprintf(err_msg, "Listing contents of an directory is currently not implemented.");
|
||||
return 0;
|
||||
} else if (uri->filename == NULL || (strlen(uri->pathinfo) > 0 && (int) uri->is_static)) {
|
||||
res->status = http_get_status(404);
|
||||
return 0;
|
||||
} else if (strlen(uri->pathinfo) != 0 && ctx->conf->local.dir_mode != URI_DIR_MODE_INFO) {
|
||||
res->status = http_get_status(404);
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (uri->is_static) {
|
||||
res->status = http_get_status(200);
|
||||
http_add_header_field(&res->hdr, "Accept-Ranges", "bytes");
|
||||
if (strcmp(req->method, "GET") != 0 && strcmp(req->method, "HEAD") != 0) {
|
||||
res->status = http_get_status(405);
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (http_get_header_field(&req->hdr, "Content-Length") != NULL || http_get_header_field(&req->hdr, "Transfer-Encoding") != NULL) {
|
||||
res->status = http_get_status(400);
|
||||
sprintf(err_msg, "A GET request must not contain a payload");
|
||||
return 0;
|
||||
}
|
||||
|
||||
cache_init_uri(ctx->conf->cache, uri);
|
||||
|
||||
const char *last_modified = http_format_date(uri->meta->stat.st_mtime, buf1, sizeof(buf1));
|
||||
http_add_header_field(&res->hdr, "Last-Modified", last_modified);
|
||||
sprintf(buf2, "%s; charset=%s", uri->meta->type, uri->meta->charset);
|
||||
http_add_header_field(&res->hdr, "Content-Type", buf2);
|
||||
|
||||
|
||||
const char *accept_encoding = http_get_header_field(&req->hdr, "Accept-Encoding");
|
||||
int enc = 0;
|
||||
if (accept_encoding != NULL) {
|
||||
if (uri->meta->filename_comp_br[0] != 0 && strstr(accept_encoding, "br") != NULL) {
|
||||
ctx->file = fopen(uri->meta->filename_comp_br, "rb");
|
||||
if (ctx->file == NULL) {
|
||||
cache_mark_dirty(ctx->conf->cache, uri->filename);
|
||||
errno = 0;
|
||||
} else {
|
||||
http_add_header_field(&res->hdr, "Content-Encoding", "br");
|
||||
enc = COMPRESS_BR;
|
||||
}
|
||||
} else if (uri->meta->filename_comp_gz[0] != 0 && strstr(accept_encoding, "gzip") != NULL) {
|
||||
ctx->file = fopen(uri->meta->filename_comp_gz, "rb");
|
||||
if (ctx->file == NULL) {
|
||||
cache_mark_dirty(ctx->conf->cache, uri->filename);
|
||||
errno = 0;
|
||||
} else {
|
||||
http_add_header_field(&res->hdr, "Content-Encoding", "gzip");
|
||||
enc = COMPRESS_GZ;
|
||||
}
|
||||
}
|
||||
if (enc != 0) {
|
||||
http_add_header_field(&res->hdr, "Vary", "Accept-Encoding");
|
||||
}
|
||||
}
|
||||
|
||||
if (uri->meta->etag[0] != 0) {
|
||||
if (enc) {
|
||||
sprintf(buf1, "%s-%s", uri->meta->etag, (enc & COMPRESS_BR) ? "br" : (enc & COMPRESS_GZ) ? "gzip" : "");
|
||||
http_add_header_field(&res->hdr, "ETag", buf1);
|
||||
} else {
|
||||
http_add_header_field(&res->hdr, "ETag", uri->meta->etag);
|
||||
}
|
||||
}
|
||||
|
||||
if (strncmp(uri->meta->type, "text/", 5) == 0) {
|
||||
http_add_header_field(&res->hdr, "Cache-Control", "public, max-age=3600");
|
||||
} else {
|
||||
http_add_header_field(&res->hdr, "Cache-Control", "public, max-age=86400");
|
||||
}
|
||||
|
||||
const char *if_modified_since = http_get_header_field(&req->hdr, "If-Modified-Since");
|
||||
const char *if_none_match = http_get_header_field(&req->hdr, "If-None-Match");
|
||||
if ((if_none_match != NULL && strstr(if_none_match, uri->meta->etag) == NULL) ||
|
||||
(accept_if_modified_since && if_modified_since != NULL && strcmp(if_modified_since, last_modified) == 0))
|
||||
{
|
||||
res->status = http_get_status(304);
|
||||
return 0;
|
||||
}
|
||||
|
||||
const char *range = http_get_header_field(&req->hdr, "Range");
|
||||
if (range != NULL) {
|
||||
if (strlen(range) <= 6 || strncmp(range, "bytes=", 6) != 0) {
|
||||
res->status = http_get_status(416);
|
||||
http_remove_header_field(&res->hdr, "Content-Type", HTTP_REMOVE_ALL);
|
||||
http_remove_header_field(&res->hdr, "Last-Modified", HTTP_REMOVE_ALL);
|
||||
http_remove_header_field(&res->hdr, "ETag", HTTP_REMOVE_ALL);
|
||||
http_remove_header_field(&res->hdr, "Cache-Control", HTTP_REMOVE_ALL);
|
||||
return 0;
|
||||
}
|
||||
range += 6;
|
||||
char *ptr = strchr(range, '-');
|
||||
if (ptr == NULL) {
|
||||
res->status = http_get_status(416);
|
||||
return 0;
|
||||
}
|
||||
ctx->file = fopen(uri->filename, "rb");
|
||||
fseek(ctx->file, 0, SEEK_END);
|
||||
unsigned long file_len = ftell(ctx->file);
|
||||
fseek(ctx->file, 0, SEEK_SET);
|
||||
if (file_len == 0) {
|
||||
ctx->content_length = 0;
|
||||
return 0;
|
||||
}
|
||||
long num1 = 0;
|
||||
long num2 = (long) file_len - 1;
|
||||
|
||||
if (ptr != range) num1 = (long) strtoul(range, NULL, 10);
|
||||
if (ptr[1] != 0) num2 = (long) strtoul(ptr + 1, NULL, 10);
|
||||
|
||||
if (num1 >= file_len || num2 >= file_len || num1 > num2) {
|
||||
res->status = http_get_status(416);
|
||||
return 0;
|
||||
}
|
||||
sprintf(buf1, "bytes %li-%li/%li", num1, num2, file_len);
|
||||
http_add_header_field(&res->hdr, "Content-Range", buf1);
|
||||
|
||||
res->status = http_get_status(206);
|
||||
fseek(ctx->file, num1, SEEK_SET);
|
||||
ctx->content_length = num2 - num1 + 1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (ctx->file == NULL) {
|
||||
ctx->file = fopen(uri->filename, "rb");
|
||||
}
|
||||
|
||||
fseek(ctx->file, 0, SEEK_END);
|
||||
ctx->content_length = ftell(ctx->file);
|
||||
fseek(ctx->file, 0, SEEK_SET);
|
||||
} else {
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
170
src/worker/proxy_handler.c
Normal file
170
src/worker/proxy_handler.c
Normal file
@ -0,0 +1,170 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Proxy handler
|
||||
* @file src/worker/proxy_handler_1.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-29
|
||||
*/
|
||||
|
||||
#include "func.h"
|
||||
#include "../logger.h"
|
||||
#include "../lib/utils.h"
|
||||
#include "../lib/proxy.h"
|
||||
#include "../lib/websocket.h"
|
||||
#include "../workers.h"
|
||||
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
|
||||
static int proxy_handler_1(client_ctx_t *ctx);
|
||||
static int proxy_handler_2(client_ctx_t *ctx);
|
||||
|
||||
void proxy_handler_func(client_ctx_t *ctx) {
|
||||
logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, ctx->req_host, CLR_STR, ctx->log_prefix);
|
||||
|
||||
// TODO handle 1xx responses
|
||||
|
||||
int ret = proxy_handler_1(ctx);
|
||||
respond(ctx);
|
||||
|
||||
if (ret == 1) {
|
||||
proxy_unlock_ctx(ctx->proxy);
|
||||
ctx->proxy = NULL;
|
||||
} else if (ctx->use_proxy == 0) {
|
||||
proxy_close(ctx->proxy);
|
||||
} else if (ctx->use_proxy == 1) {
|
||||
proxy_handler_2(ctx);
|
||||
proxy_unlock_ctx(ctx->proxy);
|
||||
ctx->proxy = NULL;
|
||||
} else if (ctx->use_proxy == 2) {
|
||||
// WebSocket
|
||||
ws_handle_connection(ctx);
|
||||
return;
|
||||
}
|
||||
|
||||
request_complete(ctx);
|
||||
handle_request(ctx);
|
||||
}
|
||||
|
||||
static int proxy_handler_1(client_ctx_t *ctx) {
|
||||
http_res *res = &ctx->res;
|
||||
http_status_ctx *status = &ctx->status;
|
||||
|
||||
char buf[1024];
|
||||
|
||||
info("Reverse proxy for " BLD_STR "%s:%i" CLR_STR, ctx->conf->proxy.hostname, ctx->conf->proxy.port);
|
||||
http_remove_header_field(&res->hdr, "Date", HTTP_REMOVE_ALL);
|
||||
http_remove_header_field(&res->hdr, "Server", HTTP_REMOVE_ALL);
|
||||
|
||||
ctx->use_proxy = proxy_init(&ctx->proxy, &ctx->req, res, status, ctx->conf, &ctx->socket, &ctx->custom_status, ctx->err_msg) == 0;
|
||||
ctx->proxy->client = ctx;
|
||||
|
||||
if (res->status->code == 101) {
|
||||
const char *connection = http_get_header_field(&res->hdr, "Connection");
|
||||
const char *upgrade = http_get_header_field(&res->hdr, "Upgrade");
|
||||
if (connection != NULL && upgrade != NULL &&
|
||||
(strstr(connection, "upgrade") != NULL || strstr(connection, "Upgrade") != NULL) &&
|
||||
strcmp(upgrade, "websocket") == 0)
|
||||
{
|
||||
const char *ws_accept = http_get_header_field(&res->hdr, "Sec-WebSocket-Accept");
|
||||
if (ws_calc_accept_key(status->ws_key, buf) == 0) {
|
||||
ctx->use_proxy = (strcmp(buf, ws_accept) == 0) ? 2 : 1;
|
||||
}
|
||||
} else {
|
||||
status->status = 101;
|
||||
status->origin = INTERNAL;
|
||||
res->status = http_get_status(501);
|
||||
}
|
||||
}
|
||||
|
||||
// Let 300 be formatted by origin server
|
||||
if (ctx->use_proxy && res->status->code >= 301 && res->status->code < 600) {
|
||||
const char *content_type = http_get_header_field(&res->hdr, "Content-Type");
|
||||
const char *content_length_f = http_get_header_field(&res->hdr, "Content-Length");
|
||||
const char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding");
|
||||
if (content_encoding == NULL && (
|
||||
content_length_f == NULL ||
|
||||
(content_length_f != NULL && strcmp(content_length_f, "0") == 0) ||
|
||||
(content_type != NULL && content_length_f != NULL && strncmp(content_type, "text/html", 9) == 0)))
|
||||
{
|
||||
long content_len = (strcmp(ctx->req.method, "HEAD") != 0 && content_length_f != NULL) ? strtol(content_length_f, NULL, 10) : 0;
|
||||
if (content_len <= sizeof(ctx->msg_content) - 1) {
|
||||
if (status->status != 101) {
|
||||
status->status = res->status->code;
|
||||
status->origin = res->status->code >= 400 ? SERVER : NONE;
|
||||
}
|
||||
ctx->use_proxy = 0;
|
||||
|
||||
if (content_len > 0)
|
||||
proxy_dump(ctx->proxy, ctx->msg_content, content_len);
|
||||
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (strcmp(ctx->req.method, "HEAD") == 0) {
|
||||
return 1;
|
||||
}
|
||||
|
||||
/*
|
||||
char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding");
|
||||
if (use_proxy && content_encoding == NULL) {
|
||||
int http_comp = http_get_compression(&req, &res);
|
||||
if (http_comp & COMPRESS_BR) {
|
||||
use_proxy |= PROXY_COMPRESS_BR;
|
||||
} else if (http_comp & COMPRESS_GZ) {
|
||||
use_proxy |= PROXY_COMPRESS_GZ;
|
||||
}
|
||||
}
|
||||
|
||||
char *transfer_encoding = http_get_header_field(&res->hdr, "Transfer-Encoding");
|
||||
int chunked = transfer_encoding != NULL && strcmp(transfer_encoding, "chunked") == 0;
|
||||
http_remove_header_field(&res->hdr, "Transfer-Encoding", HTTP_REMOVE_ALL);
|
||||
ret = sprintf(buf, "%s%s%s",
|
||||
(use_proxy & PROXY_COMPRESS_BR) ? "br" :
|
||||
((use_proxy & PROXY_COMPRESS_GZ) ? "gzip" : ""),
|
||||
((use_proxy & PROXY_COMPRESS) && chunked) ? ", " : "",
|
||||
chunked ? "chunked" : "");
|
||||
if (ret > 0) {
|
||||
http_add_header_field(&res->hdr, "Transfer-Encoding", buf);
|
||||
}
|
||||
*/
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int proxy_handler_2(client_ctx_t *ctx) {
|
||||
const char *transfer_encoding = http_get_header_field(&ctx->res.hdr, "Transfer-Encoding");
|
||||
int chunked = transfer_encoding != NULL && strstr(transfer_encoding, "chunked") != NULL;
|
||||
|
||||
const char *content_len = http_get_header_field(&ctx->res.hdr, "Content-Length");
|
||||
unsigned long len_to_send = 0;
|
||||
if (content_len != NULL) {
|
||||
len_to_send = strtol(content_len, NULL, 10);
|
||||
}
|
||||
|
||||
int flags = (chunked ? PROXY_CHUNKED : 0) | (ctx->use_proxy & PROXY_COMPRESS);
|
||||
int ret = proxy_send(ctx->proxy, &ctx->socket, len_to_send, flags);
|
||||
|
||||
if (ret < 0) {
|
||||
ctx->c_keep_alive = 0;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void proxy_close(proxy_ctx_t *ctx) {
|
||||
client_ctx_t *cctx = ctx->client;
|
||||
if (cctx) {
|
||||
logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, cctx->req_host, CLR_STR, cctx->log_prefix);
|
||||
} else {
|
||||
logger_set_prefix("");
|
||||
}
|
||||
|
||||
info(BLUE_STR "Closing proxy connection");
|
||||
sock_close(&ctx->proxy);
|
||||
|
||||
memset(ctx, 0, sizeof(*ctx));
|
||||
errno = 0;
|
||||
}
|
396
src/worker/request_handler.c
Normal file
396
src/worker/request_handler.c
Normal file
@ -0,0 +1,396 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Client request handler
|
||||
* @file src/worker/request_handler.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-28
|
||||
*/
|
||||
|
||||
#include "../defs.h"
|
||||
#include "func.h"
|
||||
#include "../workers.h"
|
||||
#include "../lib/mpmc.h"
|
||||
#include "../logger.h"
|
||||
#include "../lib/utils.h"
|
||||
#include "../server.h"
|
||||
#include "../lib/res.h"
|
||||
#include "../lib/error.h"
|
||||
|
||||
#include <string.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <errno.h>
|
||||
|
||||
static int request_handler(client_ctx_t *ctx);
|
||||
|
||||
void request_handler_func(client_ctx_t *ctx) {
|
||||
logger_set_prefix("[%*s]%s", INET6_ADDRSTRLEN, ctx->socket.s_addr, ctx->log_prefix);
|
||||
|
||||
switch (request_handler(ctx)) {
|
||||
case 0:
|
||||
respond(ctx);
|
||||
request_complete(ctx);
|
||||
handle_request(ctx);
|
||||
break;
|
||||
case 1:
|
||||
local_handle(ctx);
|
||||
break;
|
||||
case 2:
|
||||
proxy_handle(ctx);
|
||||
break;
|
||||
default:
|
||||
tcp_close(ctx);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static void init_ctx(client_ctx_t *ctx) {
|
||||
ctx->conf = NULL;
|
||||
ctx->file = NULL;
|
||||
ctx->proxy = NULL;
|
||||
ctx->use_fastcgi = 0;
|
||||
ctx->use_proxy = 0;
|
||||
ctx->ws_close = 0;
|
||||
ctx->proxy = NULL;
|
||||
ctx->msg_content[0] = 0;
|
||||
ctx->msg_buf = NULL;
|
||||
ctx->msg_buf_ptr = NULL;
|
||||
ctx->req_host[0] = 0;
|
||||
ctx->err_msg[0] = 0;
|
||||
ctx->req_s = ctx->socket.ts_last;
|
||||
|
||||
memset(&ctx->uri, 0, sizeof(ctx->uri));
|
||||
memset(&ctx->req, 0, sizeof(ctx->req));
|
||||
memset(&ctx->res, 0, sizeof(ctx->res));
|
||||
|
||||
|
||||
ctx->res.status = http_get_status(501);
|
||||
http_init_hdr(&ctx->res.hdr);
|
||||
ctx->res.hdr.last_field_num = -1;
|
||||
sprintf(ctx->res.version, "1.1");
|
||||
|
||||
ctx->status.status = 0;
|
||||
ctx->status.origin = NONE;
|
||||
ctx->status.ws_key = NULL;
|
||||
}
|
||||
|
||||
static int request_handler(client_ctx_t *ctx) {
|
||||
sock *client = &ctx->socket;
|
||||
char *err_msg = ctx->err_msg;
|
||||
http_res *res = &ctx->res;
|
||||
|
||||
long ret;
|
||||
char buf0[1024], buf1[1024];
|
||||
|
||||
ctx->req_s = clock_micros();
|
||||
|
||||
init_ctx(ctx);
|
||||
|
||||
http_add_header_field(&res->hdr, "Date", http_get_date(buf0, sizeof(buf0)));
|
||||
http_add_header_field(&res->hdr, "Server", SERVER_STR);
|
||||
/*if (ret <= 0) {
|
||||
if (errno != 0) return 0;
|
||||
|
||||
ctx->c_keep_alive = 0;
|
||||
res->status = http_get_status(408);
|
||||
return 0;
|
||||
}*/
|
||||
|
||||
http_req *req = &ctx->req;
|
||||
ret = http_receive_request(client, req);
|
||||
if (ret != 0) {
|
||||
ctx->c_keep_alive = 0;
|
||||
error("Unable to receive http header");
|
||||
sprintf(err_msg, "Unable to receive http header: %s.", error_str(errno, buf0, sizeof(buf0)));
|
||||
int err = error_get_http();
|
||||
res->status = http_get_status(err == HTTP_ERROR_URI_TOO_LONG ? 414 : (err == HTTP_ERROR_TOO_MANY_HEADER_FIELDS ? 431 : 400));
|
||||
errno = 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
const char *hdr_connection = http_get_header_field(&req->hdr, "Connection");
|
||||
ctx->c_keep_alive = (hdr_connection != NULL && (strstr(hdr_connection, "keep-alive") != NULL || strstr(hdr_connection, "Keep-Alive") != NULL));
|
||||
const char *host_ptr = http_get_header_field(&req->hdr, "Host");
|
||||
if (host_ptr != NULL && strlen(host_ptr) > 255) {
|
||||
ctx->req_host[0] = 0;
|
||||
res->status = http_get_status(400);
|
||||
sprintf(err_msg, "Host header field is too long.");
|
||||
return 0;
|
||||
} else if (host_ptr == NULL || strchr(host_ptr, '/') != NULL) {
|
||||
if (strchr(ctx->socket.addr, ':') == NULL) {
|
||||
strcpy(ctx->req_host, ctx->socket.addr);
|
||||
} else {
|
||||
sprintf(ctx->req_host, "[%s]", ctx->socket.addr);
|
||||
}
|
||||
res->status = http_get_status(400);
|
||||
sprintf(err_msg, "The client provided no or an invalid Host header field.");
|
||||
return 0;
|
||||
} else {
|
||||
strcpy(ctx->req_host, host_ptr);
|
||||
}
|
||||
|
||||
logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, ctx->req_host, CLR_STR, ctx->log_prefix);
|
||||
info(BLD_STR "%s %s", req->method, req->uri);
|
||||
|
||||
if (strncmp(req->uri, "/.sesimos/res/", 14) == 0) {
|
||||
if (strcmp(req->method, "GET") != 0 && strcmp(req->method, "HEAD") != 0) {
|
||||
res->status = http_get_status(405);
|
||||
http_add_header_field(&res->hdr, "Allow", "GET, HEAD");
|
||||
return 0;
|
||||
}
|
||||
|
||||
const res_t resources[] = {
|
||||
{"style.css", "text/css; charset=UTF-8", http_style_doc, http_style_doc_size},
|
||||
{"icon-error.svg", "image/svg+xml; charset=UTF-8", http_icon_error, http_icon_error_size},
|
||||
{"icon-info.svg", "image/svg+xml; charset=UTF-8", http_icon_info, http_icon_info_size},
|
||||
{"icon-success.svg", "image/svg+xml; charset=UTF-8", http_icon_success, http_icon_success_size},
|
||||
{"icon-warning.svg", "image/svg+xml; charset=UTF-8", http_icon_warning, http_icon_warning_size},
|
||||
{"globe.svg", "image/svg+xml; charset=UTF-8", http_icon_globe, http_icon_globe_size},
|
||||
};
|
||||
|
||||
res->status = http_get_status(404);
|
||||
for (int i = 0; i < sizeof(resources) / sizeof(res_t); i++) {
|
||||
const res_t *r = &resources[i];
|
||||
if (strcmp(req->uri + 14, r->name) == 0) {
|
||||
res->status = http_get_status(203);
|
||||
http_add_header_field(&res->hdr, "Content-Type", r->type);
|
||||
http_add_header_field(&res->hdr, "Cache-Control", "public, max-age=86400");
|
||||
ctx->msg_buf = (char *) r->content;
|
||||
ctx->content_length = r->size;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
ctx->conf = get_host_config(ctx->req_host);
|
||||
if (ctx->conf == NULL) {
|
||||
res->status = http_get_status(421);
|
||||
strcpy(ctx->err_msg, "The requested host name is not configured on the server.");
|
||||
return 0;
|
||||
}
|
||||
|
||||
http_uri *uri = &ctx->uri;
|
||||
unsigned char dir_mode = (ctx->conf->type == CONFIG_TYPE_LOCAL ? ctx->conf->local.dir_mode : URI_DIR_MODE_NO_VALIDATION);
|
||||
ret = uri_init(uri, ctx->conf->local.webroot, req->uri, dir_mode);
|
||||
if (ret != 0) {
|
||||
if (ret == 1) {
|
||||
sprintf(err_msg, "Invalid URI: has to start with slash.");
|
||||
res->status = http_get_status(400);
|
||||
} else if (ret == 2) {
|
||||
sprintf(err_msg, "Invalid URI: contains relative path change (/../).");
|
||||
res->status = http_get_status(400);
|
||||
} else if (ret == 3) {
|
||||
sprintf(err_msg, "The specified webroot directory does not exist.");
|
||||
res->status = http_get_status(404);
|
||||
} else {
|
||||
res->status = http_get_status(500);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (ctx->conf->type == CONFIG_TYPE_LOCAL && strcmp(req->method, "TRACE") == 0) {
|
||||
return 1;
|
||||
} else if (dir_mode != URI_DIR_MODE_NO_VALIDATION) {
|
||||
ssize_t size = sizeof(buf0);
|
||||
url_decode(req->uri, buf0, &size);
|
||||
int change_proto = (!client->enc && strncmp(uri->uri, "/.well-known/", 13) != 0);
|
||||
if (strcmp(uri->uri, buf0) != 0 || change_proto) {
|
||||
res->status = http_get_status(308);
|
||||
size = url_encode(uri->uri, strlen(uri->uri), buf0, sizeof(buf0));
|
||||
if (change_proto) {
|
||||
int p_len = snprintf(buf1, sizeof(buf1), "https://%s%s", ctx->req_host, buf0);
|
||||
if (p_len < 0 || p_len >= sizeof(buf1)) {
|
||||
res->status = http_get_status(500);
|
||||
error("Header field 'Location' too long");
|
||||
return 0;
|
||||
}
|
||||
http_add_header_field(&res->hdr, "Location", buf1);
|
||||
} else {
|
||||
http_add_header_field(&res->hdr, "Location", buf0);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
} else if (!client->enc) {
|
||||
res->status = http_get_status(308);
|
||||
sprintf(buf0, "https://%s%s", ctx->req_host, req->uri);
|
||||
http_add_header_field(&res->hdr, "Location", buf0);
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (ctx->conf->type == CONFIG_TYPE_LOCAL) {
|
||||
return 1;
|
||||
} else if (ctx->conf->type == CONFIG_TYPE_REVERSE_PROXY) {
|
||||
return 2;
|
||||
} else {
|
||||
error("Unknown host type: %i", ctx->conf->type);
|
||||
res->status = http_get_status(501);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int respond(client_ctx_t *ctx) {
|
||||
http_req *req = &ctx->req;
|
||||
http_res *res = &ctx->res;
|
||||
sock *client = &ctx->socket;
|
||||
http_status_ctx *status = &ctx->status;
|
||||
char *err_msg = ctx->err_msg;
|
||||
|
||||
long ret = 0;
|
||||
char buf0[1024];
|
||||
char msg_pre_buf_1[4096], msg_pre_buf_2[4096];
|
||||
char buffer[CHUNK_SIZE];
|
||||
|
||||
if (!ctx->use_proxy) {
|
||||
if (ctx->conf != NULL && ctx->conf->type == CONFIG_TYPE_LOCAL && ctx->uri.is_static && res->status->code == 405) {
|
||||
http_add_header_field(&res->hdr, "Allow", "GET, HEAD, TRACE");
|
||||
}
|
||||
if (http_get_header_field(&res->hdr, "Accept-Ranges") == NULL) {
|
||||
http_add_header_field(&res->hdr, "Accept-Ranges", "none");
|
||||
}
|
||||
if (!ctx->use_fastcgi && ctx->file == NULL && ctx->msg_buf == NULL) {
|
||||
http_remove_header_field(&res->hdr, "Date", HTTP_REMOVE_ALL);
|
||||
http_remove_header_field(&res->hdr, "Server", HTTP_REMOVE_ALL);
|
||||
http_remove_header_field(&res->hdr, "Cache-Control", HTTP_REMOVE_ALL);
|
||||
http_remove_header_field(&res->hdr, "Content-Type", HTTP_REMOVE_ALL);
|
||||
http_remove_header_field(&res->hdr, "Content-Encoding", HTTP_REMOVE_ALL);
|
||||
http_add_header_field(&res->hdr, "Date", http_get_date(buf0, sizeof(buf0)));
|
||||
http_add_header_field(&res->hdr, "Server", SERVER_STR);
|
||||
http_add_header_field(&res->hdr, "Cache-Control", "no-cache");
|
||||
http_add_header_field(&res->hdr, "Content-Type", "text/html; charset=UTF-8");
|
||||
|
||||
// TODO list Locations on 3xx Redirects
|
||||
const http_doc_info *http_info = http_get_status_info(res->status);
|
||||
const http_status_msg *http_msg = http_get_error_msg(res->status);
|
||||
|
||||
if (ctx->msg_content[0] == 0) {
|
||||
if (res->status->code >= 300 && res->status->code < 400) {
|
||||
const char *location = http_get_header_field(&res->hdr, "Location");
|
||||
if (location != NULL) {
|
||||
snprintf(ctx->msg_content, sizeof(ctx->msg_content), " <ul>\n <li><a href=\"%s\">%s</a></li>\n </ul>\n", location, location);
|
||||
}
|
||||
}
|
||||
} else if (strncmp(ctx->msg_content, "<!DOCTYPE html>", 15) == 0 || strncmp(ctx->msg_content, "<html", 5) == 0) {
|
||||
ctx->msg_content[0] = 0;
|
||||
// TODO let relevant information pass?
|
||||
}
|
||||
|
||||
char *proxy_doc = "";
|
||||
if (ctx->conf != NULL && ctx->conf->type == CONFIG_TYPE_REVERSE_PROXY) {
|
||||
const http_status *status_hdr = http_get_status(status->status);
|
||||
char stat_str[8];
|
||||
sprintf(stat_str, "%03i", status->status);
|
||||
snprintf(msg_pre_buf_2, sizeof(msg_pre_buf_2), http_proxy_doc,
|
||||
" success",
|
||||
(status->origin == CLIENT_REQ) ? " error" : " success",
|
||||
(status->origin == INTERNAL) ? " error" : " success",
|
||||
(status->origin == SERVER_REQ) ? " error" : (status->status == 0 ? "" : " success"),
|
||||
(status->origin == CLIENT_RES) ? " error" : " success",
|
||||
(status->origin == SERVER) ? " error" : (status->status == 0 ? "" : " success"),
|
||||
(status->origin == SERVER_RES) ? " error" : (status->status == 0 ? "" : " success"),
|
||||
(status->origin == INTERNAL) ? " error" : " success",
|
||||
(status->origin == INTERNAL || status->origin == SERVER) ? " error" : " success",
|
||||
res->status->code,
|
||||
res->status->msg,
|
||||
(status->status == 0) ? "???" : stat_str,
|
||||
(status_hdr != NULL) ? status_hdr->msg : "",
|
||||
ctx->req_host, SERVER_NAME);
|
||||
proxy_doc = msg_pre_buf_2;
|
||||
}
|
||||
|
||||
ctx->msg_buf_ptr = malloc(4096);
|
||||
ctx->msg_buf = ctx->msg_buf_ptr;
|
||||
snprintf(msg_pre_buf_1, sizeof(msg_pre_buf_1), http_info->doc,
|
||||
res->status->code, res->status->msg, http_msg != NULL ? http_msg->msg : "", err_msg[0] != 0 ? err_msg : "");
|
||||
ctx->content_length = snprintf(ctx->msg_buf, 4096, http_default_doc, res->status->code,
|
||||
res->status->msg, msg_pre_buf_1, http_info->mode, http_info->icon, http_info->color, ctx->req_host,
|
||||
proxy_doc, ctx->msg_content[0] != 0 ? ctx->msg_content : "", SERVER_STR_HTML);
|
||||
}
|
||||
if (ctx->content_length >= 0) {
|
||||
sprintf(buf0, "%li", ctx->content_length);
|
||||
http_remove_header_field(&res->hdr, "Content-Length", HTTP_REMOVE_ALL);
|
||||
http_add_header_field(&res->hdr, "Content-Length", buf0);
|
||||
} else if (http_get_header_field(&res->hdr, "Transfer-Encoding") == NULL) {
|
||||
ctx->s_keep_alive = 0;
|
||||
}
|
||||
}
|
||||
|
||||
if (ctx->use_proxy != 2) {
|
||||
http_remove_header_field(&res->hdr, "Connection", HTTP_REMOVE_ALL);
|
||||
http_remove_header_field(&res->hdr, "Keep-Alive", HTTP_REMOVE_ALL);
|
||||
if (ctx->s_keep_alive && ctx->c_keep_alive) {
|
||||
http_add_header_field(&res->hdr, "Connection", "keep-alive");
|
||||
sprintf(buf0, "timeout=%i, max=%i", CLIENT_TIMEOUT, REQ_PER_CONNECTION);
|
||||
http_add_header_field(&res->hdr, "Keep-Alive", buf0);
|
||||
} else {
|
||||
http_add_header_field(&res->hdr, "Connection", "close");
|
||||
}
|
||||
}
|
||||
|
||||
http_send_response(client, res);
|
||||
ctx->res_ts = clock_micros();
|
||||
const char *location = http_get_header_field(&res->hdr, "Location");
|
||||
info("%s%s%03i %s%s%s (%s)%s", http_get_status_color(res->status), ctx->use_proxy ? "-> " : "", res->status->code,
|
||||
res->status->msg, location != NULL ? " -> " : "", location != NULL ? location : "",
|
||||
format_duration(ctx->res_ts - ctx->req_s, buf0), CLR_STR);
|
||||
|
||||
// TODO access/error log file
|
||||
|
||||
if (ctx->use_proxy) {
|
||||
// reverse proxy
|
||||
return 3;
|
||||
} else if (strcmp(req->method, "HEAD") != 0) {
|
||||
// default response
|
||||
if (ctx->msg_buf != NULL) {
|
||||
ret = sock_send(client, ctx->msg_buf, ctx->content_length, 0);
|
||||
if (ret <= 0) {
|
||||
error("Unable to send");
|
||||
}
|
||||
} else if (ctx->file != NULL) {
|
||||
unsigned long len, snd_len = 0;
|
||||
while (snd_len < ctx->content_length) {
|
||||
len = fread(buffer, 1, CHUNK_SIZE, ctx->file);
|
||||
if (snd_len + len > ctx->content_length) {
|
||||
len = ctx->content_length - snd_len;
|
||||
}
|
||||
ret = sock_send(client, buffer, len, feof(ctx->file) ? 0 : MSG_MORE);
|
||||
if (ret <= 0) {
|
||||
error("Unable to send");
|
||||
break;
|
||||
}
|
||||
snd_len += ret;
|
||||
}
|
||||
} else if (ctx->use_fastcgi) {
|
||||
// FastCGI
|
||||
return 2;
|
||||
}
|
||||
|
||||
if (ret < 0) ctx->c_keep_alive = 0;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void request_complete(client_ctx_t *ctx) {
|
||||
char buf[32];
|
||||
ctx->req_e = clock_micros();
|
||||
info("Transfer complete: %s", format_duration(ctx->req_e - ctx->req_s, buf));
|
||||
|
||||
if (ctx->file) fclose(ctx->file);
|
||||
free(ctx->msg_buf_ptr);
|
||||
uri_free(&ctx->uri);
|
||||
http_free_req(&ctx->req);
|
||||
http_free_res(&ctx->res);
|
||||
}
|
||||
|
||||
void timeout_request(client_ctx_t *ctx) {
|
||||
init_ctx(ctx);
|
||||
logger_set_prefix("[%*s]%s", INET6_ADDRSTRLEN, ctx->socket.s_addr, ctx->log_prefix);
|
||||
|
||||
ctx->s_keep_alive = 0;
|
||||
ctx->res.status = http_get_status(408);
|
||||
|
||||
respond(ctx);
|
||||
tcp_close(ctx);
|
||||
}
|
130
src/worker/tcp_acceptor.c
Normal file
130
src/worker/tcp_acceptor.c
Normal file
@ -0,0 +1,130 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief TCP acceptor
|
||||
* @file src/worker/tcp_acceptor.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-28
|
||||
*/
|
||||
|
||||
#include "func.h"
|
||||
#include "../logger.h"
|
||||
#include "../lib/utils.h"
|
||||
#include "../lib/geoip.h"
|
||||
#include "../workers.h"
|
||||
#include "../server.h"
|
||||
#include "../lib/error.h"
|
||||
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
#include <openssl/ssl.h>
|
||||
#include <openssl/err.h>
|
||||
|
||||
static int tcp_acceptor(client_ctx_t *ctx);
|
||||
|
||||
void tcp_acceptor_func(client_ctx_t *ctx) {
|
||||
if (tcp_acceptor(ctx) == 0) {
|
||||
handle_request(ctx);
|
||||
} else {
|
||||
tcp_close(ctx);
|
||||
}
|
||||
}
|
||||
|
||||
static int tcp_acceptor(client_ctx_t *ctx) {
|
||||
struct sockaddr_in6 server_addr;
|
||||
|
||||
inet_ntop(ctx->socket._addr.ipv6.sin6_family, &ctx->socket._addr.ipv6.sin6_addr, ctx->_c_addr, sizeof(ctx->_c_addr));
|
||||
if (strncmp(ctx->_c_addr, "::ffff:", 7) == 0) {
|
||||
ctx->socket.addr = ctx->_c_addr + 7;
|
||||
} else {
|
||||
ctx->socket.addr = ctx->_c_addr;
|
||||
}
|
||||
|
||||
socklen_t len = sizeof(server_addr);
|
||||
getsockname(ctx->socket.socket, (struct sockaddr *) &server_addr, &len);
|
||||
inet_ntop(server_addr.sin6_family, (void *) &server_addr.sin6_addr, ctx->_s_addr, sizeof(ctx->_s_addr));
|
||||
if (strncmp(ctx->_s_addr, "::ffff:", 7) == 0) {
|
||||
ctx->socket.s_addr = ctx->_s_addr + 7;
|
||||
} else {
|
||||
ctx->socket.s_addr = ctx->_s_addr;
|
||||
}
|
||||
|
||||
sprintf(ctx->log_prefix, "[%s%4i%s]%s[%*s][%5i]%s", (int) ctx->socket.enc ? HTTPS_STR : HTTP_STR,
|
||||
ntohs(server_addr.sin6_port), CLR_STR, /*color_table[0]*/ "", INET6_ADDRSTRLEN, ctx->socket.addr,
|
||||
ntohs(ctx->socket._addr.ipv6.sin6_port), CLR_STR);
|
||||
|
||||
logger_set_prefix("[%*s]%s", INET6_ADDRSTRLEN, ctx->socket.s_addr, ctx->log_prefix);
|
||||
|
||||
int ret;
|
||||
char buf[1024];
|
||||
sock *client = &ctx->socket;
|
||||
|
||||
ctx->cnx_s = clock_micros();
|
||||
|
||||
if (config.dns_server[0] != 0) {
|
||||
sprintf(buf, "dig @%s +short +time=1 -x %s", config.dns_server, ctx->socket.addr);
|
||||
FILE *dig = popen(buf, "r");
|
||||
if (dig == NULL) {
|
||||
error("Unable to start dig: %s");
|
||||
goto dig_err;
|
||||
}
|
||||
unsigned long read = fread(buf, 1, sizeof(buf), dig);
|
||||
ret = pclose(dig);
|
||||
if (ret != 0) {
|
||||
error("Dig terminated with exit code %i", ret);
|
||||
goto dig_err;
|
||||
}
|
||||
char *ptr = memchr(buf, '\n', read);
|
||||
if (ptr == buf || ptr == NULL) {
|
||||
goto dig_err;
|
||||
}
|
||||
ptr[-1] = 0;
|
||||
strncpy(ctx->host, buf, sizeof(ctx->host));
|
||||
} else {
|
||||
dig_err:
|
||||
ctx->host[0] = 0;
|
||||
}
|
||||
|
||||
ctx->cc[0] = 0;
|
||||
geoip_lookup_country(&client->_addr.sock, ctx->cc);
|
||||
|
||||
info("Connection accepted from %s %s%s%s[%s]", ctx->socket.addr, ctx->host[0] != 0 ? "(" : "",
|
||||
ctx->host[0] != 0 ? ctx->host : "", ctx->host[0] != 0 ? ") " : "",
|
||||
ctx->cc[0] != 0 ? ctx->cc : "N/A");
|
||||
|
||||
if (sock_set_socket_timeout(client, 1) != 0 || sock_set_timeout(client, CLIENT_TIMEOUT) != 0) {
|
||||
error("Unable to set timeout for socket");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (client->enc) {
|
||||
client->ssl = SSL_new(client->ctx);
|
||||
SSL_set_fd(client->ssl, client->socket);
|
||||
SSL_set_accept_state(client->ssl);
|
||||
|
||||
ret = SSL_accept(client->ssl);
|
||||
if (ret != 1) {
|
||||
error_ssl(SSL_get_error(client->ssl, ret));
|
||||
info("Unable to perform handshake");
|
||||
return - 1;
|
||||
}
|
||||
client->ts_last = clock_micros();
|
||||
}
|
||||
|
||||
ctx->req_num = 0;
|
||||
ctx->s_keep_alive = 1;
|
||||
ctx->c_keep_alive = 1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void tcp_close(client_ctx_t *ctx) {
|
||||
logger_set_prefix("[%*s]%s", INET6_ADDRSTRLEN, ctx->socket.s_addr, ctx->log_prefix);
|
||||
|
||||
sock_close(&ctx->socket);
|
||||
|
||||
ctx->cnx_e = clock_micros();
|
||||
char buf[32];
|
||||
info("Connection closed (%s)", format_duration(ctx->cnx_e - ctx->cnx_s, buf));
|
||||
|
||||
server_free_client(ctx);
|
||||
}
|
94
src/worker/ws_frame_handler.c
Normal file
94
src/worker/ws_frame_handler.c
Normal file
@ -0,0 +1,94 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief WebSocket frame handler
|
||||
* @file src/worker/ws_frame_handler.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-30
|
||||
*/
|
||||
|
||||
#include "../defs.h"
|
||||
#include "func.h"
|
||||
#include "../logger.h"
|
||||
#include "../lib/websocket.h"
|
||||
#include "../workers.h"
|
||||
#include <errno.h>
|
||||
|
||||
static int ws_frame_handler(ws_ctx_t *ctx);
|
||||
|
||||
void ws_frame_handler_func(ws_ctx_t *ctx) {
|
||||
logger_set_prefix("[%*s]%s", INET6_ADDRSTRLEN, ctx->client->socket.s_addr, ctx->client->log_prefix);
|
||||
|
||||
if (ws_frame_handler(ctx) == 0) {
|
||||
if (ctx->client->ws_close == 3) {
|
||||
ws_close(ctx);
|
||||
} else {
|
||||
ws_handle_frame(ctx);
|
||||
}
|
||||
} else {
|
||||
ws_close(ctx);
|
||||
}
|
||||
}
|
||||
|
||||
int ws_handle_connection(client_ctx_t *ctx) {
|
||||
info("Upgrading to WebSocket connection");
|
||||
sock_set_timeout(&ctx->socket, WS_TIMEOUT);
|
||||
sock_set_timeout(&ctx->proxy->proxy, WS_TIMEOUT);
|
||||
|
||||
ws_ctx_t *a = malloc(sizeof(ws_ctx_t));
|
||||
ws_ctx_t *b = malloc(sizeof(ws_ctx_t));
|
||||
|
||||
a->other = b, b->other = a;
|
||||
a->client = ctx, b->client = ctx;
|
||||
a->socket = &ctx->socket, b->socket = &ctx->proxy->proxy;
|
||||
|
||||
ws_handle_frame(a);
|
||||
ws_handle_frame(b);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int ws_frame_handler(ws_ctx_t *ctx) {
|
||||
ws_frame frame;
|
||||
char buf[CHUNK_SIZE];
|
||||
|
||||
sock *socket = ctx->socket;
|
||||
sock *other = (ctx->socket == &ctx->client->socket) ? &ctx->client->proxy->proxy : &ctx->client->socket;
|
||||
|
||||
if (ws_recv_frame_header(socket, &frame) != 0)
|
||||
return -1;
|
||||
|
||||
debug("WebSocket: Peer %s, Opcode=0x%X, Len=%li", (ctx->socket == &ctx->client->socket) ? "client" : "server", frame.opcode, frame.len);
|
||||
|
||||
if (frame.opcode == 0x8) {
|
||||
ctx->client->ws_close |= (ctx->socket == &ctx->client->socket) ? 1 : 2;
|
||||
}
|
||||
|
||||
if (ws_send_frame_header(other, &frame) != 0)
|
||||
return -1;
|
||||
|
||||
if (frame.len > 0) {
|
||||
long ret = sock_splice(other, socket, buf, sizeof(buf), frame.len);
|
||||
if (ret < 0) {
|
||||
error("Unable to forward data in WebSocket");
|
||||
return -1;
|
||||
} else if (ret != frame.len) {
|
||||
error("Unable to forward correct number of bytes in WebSocket");
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void ws_close(ws_ctx_t *ctx) {
|
||||
ws_ctx_t *other = ctx->other;
|
||||
if (other) {
|
||||
other->other = NULL;
|
||||
logger_set_prefix("[%*s]%s", INET6_ADDRSTRLEN, ctx->client->socket.s_addr, ctx->client->log_prefix);
|
||||
info("Closing WebSocket connection");
|
||||
proxy_close(ctx->client->proxy);
|
||||
tcp_close(ctx->client);
|
||||
}
|
||||
free(ctx);
|
||||
errno = 0;
|
||||
}
|
87
src/workers.c
Normal file
87
src/workers.c
Normal file
@ -0,0 +1,87 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Worker interface
|
||||
* @file src/workers.c
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-29
|
||||
*/
|
||||
|
||||
#include "workers.h"
|
||||
#include "lib/mpmc.h"
|
||||
|
||||
#include "worker/func.h"
|
||||
#include "async.h"
|
||||
|
||||
static mpmc_t tcp_acceptor_ctx, request_handler_ctx, local_handler_ctx, fastcgi_handler_cxt, proxy_handler_ctx,
|
||||
ws_frame_handler_ctx;
|
||||
|
||||
int workers_init(void) {
|
||||
mpmc_init(&tcp_acceptor_ctx, 8, 64, (void (*)(void *)) tcp_acceptor_func, "tcp");
|
||||
mpmc_init(&request_handler_ctx, 16, 64, (void (*)(void *)) request_handler_func, "req");
|
||||
mpmc_init(&local_handler_ctx, 16, 64, (void (*)(void *)) local_handler_func, "local");
|
||||
mpmc_init(&fastcgi_handler_cxt, 16, 64, (void (*)(void *)) fastcgi_handler_func, "fcgi");
|
||||
mpmc_init(&proxy_handler_ctx, 16, 64, (void (*)(void *)) proxy_handler_func, "proxy");
|
||||
mpmc_init(&ws_frame_handler_ctx, 16, 64, (void (*)(void *)) ws_frame_handler_func, "ws");
|
||||
return -1;
|
||||
}
|
||||
|
||||
void workers_stop(void) {
|
||||
mpmc_stop(&tcp_acceptor_ctx);
|
||||
mpmc_stop(&local_handler_ctx);
|
||||
mpmc_stop(&fastcgi_handler_cxt);
|
||||
mpmc_stop(&proxy_handler_ctx);
|
||||
mpmc_stop(&request_handler_ctx);
|
||||
mpmc_stop(&ws_frame_handler_ctx);
|
||||
}
|
||||
|
||||
void workers_destroy(void) {
|
||||
mpmc_destroy(&tcp_acceptor_ctx);
|
||||
mpmc_destroy(&local_handler_ctx);
|
||||
mpmc_destroy(&fastcgi_handler_cxt);
|
||||
mpmc_destroy(&proxy_handler_ctx);
|
||||
mpmc_destroy(&request_handler_ctx);
|
||||
mpmc_destroy(&ws_frame_handler_ctx);
|
||||
}
|
||||
|
||||
int tcp_accept(client_ctx_t *ctx) {
|
||||
return mpmc_queue(&tcp_acceptor_ctx, ctx);
|
||||
}
|
||||
|
||||
static int handle_request_cb(client_ctx_t *ctx) {
|
||||
return mpmc_queue(&request_handler_ctx, ctx);
|
||||
}
|
||||
|
||||
int handle_request(client_ctx_t *ctx) {
|
||||
if (ctx->c_keep_alive && ctx->s_keep_alive) {
|
||||
return async(&ctx->socket, ASYNC_WAIT_READ, 0, ctx,
|
||||
(void (*)(void *)) handle_request_cb,
|
||||
(void (*)(void *)) timeout_request,
|
||||
(void (*)(void *)) tcp_close);
|
||||
} else {
|
||||
tcp_close(ctx);
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
int local_handle(client_ctx_t *ctx) {
|
||||
return mpmc_queue(&local_handler_ctx, ctx);
|
||||
}
|
||||
|
||||
int fastcgi_handle(client_ctx_t *ctx) {
|
||||
return mpmc_queue(&fastcgi_handler_cxt, ctx);
|
||||
}
|
||||
|
||||
int proxy_handle(client_ctx_t *ctx) {
|
||||
return mpmc_queue(&proxy_handler_ctx, ctx);
|
||||
}
|
||||
|
||||
static int ws_handle_frame_cb(ws_ctx_t *ctx) {
|
||||
return mpmc_queue(&ws_frame_handler_ctx, ctx);
|
||||
}
|
||||
|
||||
int ws_handle_frame(ws_ctx_t *ctx) {
|
||||
return async(ctx->socket, ASYNC_WAIT_READ, 0, ctx,
|
||||
(void (*)(void *)) ws_handle_frame_cb,
|
||||
(void (*)(void *)) ws_close,
|
||||
(void (*)(void *)) ws_close);
|
||||
}
|
32
src/workers.h
Normal file
32
src/workers.h
Normal file
@ -0,0 +1,32 @@
|
||||
/**
|
||||
* sesimos - secure, simple, modern web server
|
||||
* @brief Worker interface (header file)
|
||||
* @file src/workers.h
|
||||
* @author Lorenz Stechauner
|
||||
* @date 2022-12-29
|
||||
*/
|
||||
|
||||
#ifndef SESIMOS_WORKERS_H
|
||||
#define SESIMOS_WORKERS_H
|
||||
|
||||
#include "worker/func.h"
|
||||
|
||||
int workers_init(void);
|
||||
|
||||
void workers_stop(void);
|
||||
|
||||
void workers_destroy(void);
|
||||
|
||||
int tcp_accept(client_ctx_t *ctx);
|
||||
|
||||
int handle_request(client_ctx_t *ctx);
|
||||
|
||||
int local_handle(client_ctx_t *ctx);
|
||||
|
||||
int fastcgi_handle(client_ctx_t *ctx);
|
||||
|
||||
int proxy_handle(client_ctx_t *ctx);
|
||||
|
||||
int ws_handle_frame(ws_ctx_t *ctx);
|
||||
|
||||
#endif //SESIMOS_WORKERS_H
|
28
test/mock_socket.c
Normal file
28
test/mock_socket.c
Normal file
@ -0,0 +1,28 @@
|
||||
|
||||
#include <stdio.h>
|
||||
#include <errno.h>
|
||||
|
||||
#include "mock_socket.h"
|
||||
|
||||
int mock_socket_send_mode;
|
||||
|
||||
static int sockets[256] = {0};
|
||||
static int n_sockets = 0;
|
||||
|
||||
int mock_socket(int domain, int type, int protocol) {
|
||||
printf("SOCKET\n");
|
||||
return (n_sockets++) + 100;
|
||||
}
|
||||
|
||||
ssize_t mock_send(int fd, const void *buf, size_t n, int flags) {
|
||||
printf("SEND\n");
|
||||
if (mock_socket_send_mode == MOCK_SOCKET_MODE_EINTR) {
|
||||
errno = EINTR;
|
||||
return rand() % ((ssize_t) n) - 1;
|
||||
} else if (mock_socket_send_mode == MOCK_SOCKET_MODE_CLOSED) {
|
||||
errno = 0; // TODO
|
||||
return -1;
|
||||
}
|
||||
|
||||
return (ssize_t) n;
|
||||
}
|
20
test/mock_socket.h
Normal file
20
test/mock_socket.h
Normal file
@ -0,0 +1,20 @@
|
||||
|
||||
#ifndef SESIMOS_MOCK_SOCKET_H
|
||||
#define SESIMOS_MOCK_SOCKET_H
|
||||
|
||||
#include <stdlib.h>
|
||||
|
||||
#define MOCK_SOCKET_MODE_SUCCESS 0
|
||||
#define MOCK_SOCKET_MODE_EINTR 1
|
||||
#define MOCK_SOCKET_MODE_CLOSED 2
|
||||
|
||||
#define socket(args...) mock_socket(args)
|
||||
#define send(args...) mock_send(args)
|
||||
|
||||
extern int mock_socket_send_mode;
|
||||
|
||||
int mock_socket(int domain, int type, int protocol);
|
||||
|
||||
ssize_t mock_send(int fd, const void *buf, size_t n, int flags);
|
||||
|
||||
#endif //SESIMOS_MOCK_SOCKET_H
|
33
test/mock_ssl.c
Normal file
33
test/mock_ssl.c
Normal file
@ -0,0 +1,33 @@
|
||||
|
||||
#include <openssl/crypto.h>
|
||||
|
||||
|
||||
int SSL_write(SSL *ssl, const void *buf, int num) {
|
||||
return num;
|
||||
}
|
||||
|
||||
int SSL_read(SSL *ssl, void *buf, int num) {
|
||||
return num;
|
||||
}
|
||||
|
||||
int SSL_peek(SSL *ssl, void *buf, int num) {
|
||||
return num;
|
||||
}
|
||||
|
||||
int SSL_get_error(const SSL *s, int ret_code) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
const char *ERR_reason_error_string(unsigned long e) {
|
||||
return "";
|
||||
}
|
||||
|
||||
int SSL_shutdown(SSL *s) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
void SSL_free(SSL *ssl) {}
|
||||
|
||||
unsigned long ERR_get_error(void) {
|
||||
return 0;
|
||||
}
|
92
test/test_list.c
Normal file
92
test/test_list.c
Normal file
@ -0,0 +1,92 @@
|
||||
|
||||
#include <criterion/criterion.h>
|
||||
#include <criterion/parameterized.h>
|
||||
|
||||
#include "../src/lib/list.h"
|
||||
|
||||
Test(list, simple) {
|
||||
int v;
|
||||
int *list = list_create(sizeof(int), 16);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 0);
|
||||
|
||||
v = 1;
|
||||
list = list_append(list, &v);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 1);
|
||||
cr_assert_eq(list[0], 1);
|
||||
|
||||
v = 3;
|
||||
list = list_append(list, &v);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 2);
|
||||
cr_assert_eq(list[0], 1);
|
||||
cr_assert_eq(list[1], 3);
|
||||
|
||||
v = 2;
|
||||
list = list_insert(list, &v, 1);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 3);
|
||||
cr_assert_eq(list[0], 1);
|
||||
cr_assert_eq(list[1], 2);
|
||||
cr_assert_eq(list[2], 3);
|
||||
|
||||
list = list_remove(list, 0);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 2);
|
||||
cr_assert_eq(list[0], 2);
|
||||
cr_assert_eq(list[1], 3);
|
||||
|
||||
list = list_remove(list, 1);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 1);
|
||||
cr_assert_eq(list[0], 2);
|
||||
|
||||
list = list_remove(list, 0);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 0);
|
||||
|
||||
list_free(list);
|
||||
}
|
||||
|
||||
Test(list, resize) {
|
||||
int v;
|
||||
int *list = list_create(sizeof(int), 4);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 0);
|
||||
|
||||
for (int i = 0; i < 4096; i++) {
|
||||
v = 9182 - i;
|
||||
list = list_append(list, &v);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), i + 1);
|
||||
for (int j = 0; j <= i; j++) {
|
||||
cr_assert_eq(list[j], 9182 - j);
|
||||
}
|
||||
}
|
||||
|
||||
for (int i = 0; i < 4096; i++) {
|
||||
list = list_remove(list, -1);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 4096 - i - 1);
|
||||
for (int j = 0; j < 4096 - i; j++) {
|
||||
cr_assert_eq(list[j], 9182 - j);
|
||||
}
|
||||
}
|
||||
|
||||
for (int i = 0; i < 4096; i++) {
|
||||
v = 47391 - i;
|
||||
list = list_append(list, &v);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), i + 1);
|
||||
for (int j = 0; j <= i; j++) {
|
||||
cr_assert_eq(list[j], 47391 - j);
|
||||
}
|
||||
}
|
||||
|
||||
list = list_clear(list);
|
||||
cr_assert_not_null(list);
|
||||
cr_assert_eq(list_size(list), 0);
|
||||
|
||||
list_free(list);
|
||||
}
|
15
test/test_sock.c
Normal file
15
test/test_sock.c
Normal file
@ -0,0 +1,15 @@
|
||||
|
||||
#include <criterion/criterion.h>
|
||||
#include "mock_socket.h"
|
||||
#include "../src/lib/sock.h"
|
||||
|
||||
|
||||
Test(sock, sock_send_1) {
|
||||
int fd = socket(AF_INET6, SOCK_STREAM, 0);
|
||||
sock s;
|
||||
s.enc = 0;
|
||||
s.socket = fd;
|
||||
|
||||
long ret = sock_send(&s, "Hello", 5, 0);
|
||||
cr_assert_eq(ret, 5);
|
||||
}
|
86
test/test_utils.c
Normal file
86
test/test_utils.c
Normal file
@ -0,0 +1,86 @@
|
||||
|
||||
#include <criterion/criterion.h>
|
||||
#include <criterion/parameterized.h>
|
||||
|
||||
#include "../src/lib/utils.h"
|
||||
|
||||
struct url_encode_t {
|
||||
long in_size;
|
||||
char in[256];
|
||||
long exp_size;
|
||||
char exp[256];
|
||||
};
|
||||
|
||||
struct format_duration_t {
|
||||
unsigned long micros;
|
||||
char exp[16];
|
||||
};
|
||||
|
||||
ParameterizedTestParameters(utils, url_encode) {
|
||||
static struct url_encode_t params[] = {
|
||||
{0, "", 0, ""},
|
||||
{9, "Test Text", 11, "Test%20Text"},
|
||||
{21, "Text\0with\0null\0bytes\0", 29, "Text%00with%00null%00bytes%00"},
|
||||
{59, "Text&with+some/strange_symbols-or#something?I%don't|know...", 59, "Text&with+some/strange_symbols-or#something?I%don't|know..."},
|
||||
{33, "Data\x12With\x13Some" "\xFF" "Control" "\xFE" "Characters", 41, "Data%12With%13Some%FFControl%FECharacters"}
|
||||
};
|
||||
return cr_make_param_array(struct url_encode_t, params, sizeof(params) / sizeof(struct url_encode_t));
|
||||
}
|
||||
|
||||
ParameterizedTest(struct url_encode_t *param, utils, url_encode) {
|
||||
char out[256];
|
||||
cr_assert_eq(url_encode(param->in, param->in_size, out, sizeof(out)), param->exp_size);
|
||||
cr_assert_arr_eq(out, param->exp, param->exp_size + 1);
|
||||
}
|
||||
|
||||
Test(utils, url_encode_bytes) {
|
||||
char out[4];
|
||||
char exp[4];
|
||||
|
||||
for (int i = 0; i < 256; i++) {
|
||||
unsigned char ch = i;
|
||||
if (ch <= 0x20 || ch >= 0x7F) {
|
||||
cr_assert_eq(url_encode(&ch, 1, out, sizeof(out)), 3);
|
||||
sprintf(exp, "%%%02X", ch);
|
||||
cr_assert_str_eq(out, exp);
|
||||
} else {
|
||||
cr_assert_eq(url_encode(&ch, 1, out, sizeof(out)), 1);
|
||||
sprintf(exp, "%c", ch);
|
||||
cr_assert_str_eq(out, exp);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Test(utils, url_encode_invalid) {
|
||||
cr_assert_eq(url_encode("Hello", 5, NULL, 0), 5);
|
||||
}
|
||||
|
||||
ParameterizedTestParameters(utils, format_duration) {
|
||||
static struct format_duration_t params[] = {
|
||||
{0, "0.0 ms"},
|
||||
{1, "0.0 ms"},
|
||||
{90, "0.1 ms"},
|
||||
{100, "0.1 ms"},
|
||||
{110, "0.1 ms"},
|
||||
{900, "0.9 ms"},
|
||||
{1000, "1.0 ms"},
|
||||
{9000, "9.0 ms"},
|
||||
{9899, "9.9 ms"},
|
||||
{9999, "10.0 ms"},
|
||||
{10000, "10 ms"},
|
||||
{11999, "12 ms"},
|
||||
{999999, "1.0 s"},
|
||||
{1000000, "1.0 s"},
|
||||
{3000000, "3.0 s"},
|
||||
{1000000 * 60, "1:00 min"},
|
||||
{1000000 * 60 * 30L - 30000000, "29:30 min"},
|
||||
{1000000 * 60 * 60L, "60:00 min"},
|
||||
{1000000 * 60 * 120L, "120 min"},
|
||||
};
|
||||
return cr_make_param_array(struct format_duration_t, params, sizeof(params) / sizeof(struct format_duration_t));
|
||||
}
|
||||
|
||||
ParameterizedTest(struct format_duration_t *param, utils, format_duration) {
|
||||
char buf[16];
|
||||
cr_assert_str_eq(format_duration(param->micros, buf), param->exp);
|
||||
}
|
Reference in New Issue
Block a user