Compare commits
	
		
			500 Commits
		
	
	
		
			v3.0
			...
			31cd2e7e73
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 31cd2e7e73 | |||
| bd8e71e83d | |||
| 9aaa28f1ca | |||
| 73e0cffa78 | |||
| 6f0371c46f | |||
| ad6ffe5425 | |||
| ab7e5cc722 | |||
| 40310faa4b | |||
| 7a3adc6ed3 | |||
| bd73061462 | |||
| 9ee0e11c86 | |||
| 240ed6bc25 | |||
| 03dd126ca7 | |||
| 17327006db | |||
| 9237c791bb | |||
| f7a6214dbc | |||
| 8053439212 | |||
| 299e726341 | |||
| e721e542f3 | |||
| df7dfb5107 | |||
| 5f3ba2b971 | |||
| 1e0a7c95da | |||
| 0394c3afb4 | |||
| 32ea03079b | |||
| 8b63d2cf32 | |||
| d461ba6f29 | |||
| 5b98fd0dab | |||
| 0b1b0bcb56 | |||
| 44fc7f2e5c | |||
| c8b605aa90 | |||
| 59b9bfd53e | |||
| e8655102b9 | |||
| add135b3ad | |||
| 2d250621c8 | |||
| f96dc46ea7 | |||
| 5138dafb8e | |||
| 4be5b0d195 | |||
| a84aa591d4 | |||
| d7db1f0be9 | |||
| 116f7035ed | |||
| 8abc0b8bfa | |||
| 410d692372 | |||
| f42ac83323 | |||
| 7d576b2466 | |||
| 5acab8144d | |||
| 4814035b62 | |||
| 15f400f376 | |||
| 32638fdd91 | |||
| 07d6280864 | |||
| 820ce8dac9 | |||
| 5bd9b3fb33 | |||
| a3dcddf69d | |||
| 487386158d | |||
| 32f00175e4 | |||
| 581b4dc5aa | |||
| 611ceabfb2 | |||
| 6b295cef95 | |||
| 764b754a6f | |||
| 7699583c5f | |||
| 8b92d4ff91 | |||
| e8d8abdc5a | |||
| 15c6b7ddaa | |||
| e9a065f8aa | |||
| 0730429041 | |||
| 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 | ||||||
| !src/** | !src/** | ||||||
| !run.sh | !res | ||||||
|  | !res/** | ||||||
|  | !doc | ||||||
|  | !doc/** | ||||||
|  | !test | ||||||
|  | !test/** | ||||||
| !Makefile | !Makefile | ||||||
| !.gitignore | !.gitignore | ||||||
| !CppNet | !*.md | ||||||
| !CppNet/** |  | ||||||
|   | |||||||
							
								
								
									
										145
									
								
								Makefile
									
									
									
									
									
								
							
							
						
						
									
										145
									
								
								Makefile
									
									
									
									
									
								
							| @@ -1,22 +1,131 @@ | |||||||
| .DEFAULT_GOAL := install |  | ||||||
|  |  | ||||||
| packages: | CC=gcc | ||||||
| 	@echo "Installing packages..." | CFLAGS=-std=gnu11 -Wno-unused-but-set-variable -D_DEFAULT_SOURCE -D_GNU_SOURCE -D_BSD_SOURCE -D_SVID_SOURCE -D_POSIX_C_SOURCE=200809L | ||||||
| 	sudo apt-get install g++ libmagic-dev libssl-dev php-cgi | LDFLAGS=-pthread -lssl -lcrypto -lmagic -lz -lmaxminddb -lbrotlienc | ||||||
| 	@echo "Finished downloading!" |  | ||||||
|  |  | ||||||
| update: | DEBIAN_OPTS=-D CACHE_MAGIC_FILE="\"/usr/share/file/magic.mgc\"" -D PHP_FPM_SOCKET="\"/var/run/php/php8.2-fpm.sock\"" | ||||||
| 	@echo "Updating imported git repos..." |  | ||||||
| 	cd CppNet |  | ||||||
| 	git pull |  | ||||||
| 	cd .. |  | ||||||
| 	@echo "Finished updating!" |  | ||||||
|  |  | ||||||
| compile: | .PHONY: all prod debug default debian permit clean test | ||||||
| 	@echo "Compiling..." | all: prod | ||||||
| 	@mkdir -p bin | default: bin bin/lib bin/worker bin/res bin/sesimos | ||||||
| 	g++ src/necronda-server.cpp -o bin/necronda-server -std=c++17 -fPIC -pthread -lz -lmagic -lssl -ldl -lcrypto |  | ||||||
| 	@echo "Finished compiling!" |  | ||||||
|  |  | ||||||
| install: | packages update compile | prod: CFLAGS += -O3 | ||||||
| 	@echo "Finished!" | 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 src/lib/error.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 $^ $@ | ||||||
|  | 	/bin/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/worker/chunk_handler.o bin/worker/fastcgi_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/worker/fastcgi_frame_handler.o: src/worker/func.h | ||||||
|  |  | ||||||
|  | bin/worker/chunk_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/* | ||||||
|   | |||||||
							
								
								
									
										49
									
								
								README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										49
									
								
								README.md
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,49 @@ | |||||||
|  |  | ||||||
|  | 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 | ||||||
|  |  | ||||||
|  |  | ||||||
|  | ### 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 |  | ||||||
							
								
								
									
										339
									
								
								src/async.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										339
									
								
								src/async.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,339 @@ | |||||||
|  | /** | ||||||
|  |  * 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) { | ||||||
|  |     while (sem_wait(&lock) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } 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; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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 | ||||||
|  |     switch (poll(fds, 1, 0)) { | ||||||
|  |         case 1: | ||||||
|  |             // fd already ready | ||||||
|  |             if (async_exec(evt, async_p2a(fds[0].revents)) == 0) | ||||||
|  |                 return 1; | ||||||
|  |             break; | ||||||
|  |         case -1: | ||||||
|  |             error("Unable to poll"); | ||||||
|  |             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; | ||||||
|  |  | ||||||
|  |     if ((local = list_create(sizeof(evt_listen_t *), 16)) == 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; | ||||||
|  |  | ||||||
|  |         // TODO timeout calculation = O(n) | ||||||
|  |         // 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 (!list_contains(local, &evt)) continue; | ||||||
|  |  | ||||||
|  |             if (async_exec(evt, async_e2a(events[i].events)) == 0) { | ||||||
|  |                 logger_set_prefix(""); | ||||||
|  |                 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); | ||||||
|  |             } | ||||||
|  |             logger_set_prefix(""); | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         // 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 | ||||||
							
								
								
									
										451
									
								
								src/cache_handler.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										451
									
								
								src/cache_handler.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,451 @@ | |||||||
|  | /** | ||||||
|  |  * 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, sem_cache, sem_magic; | ||||||
|  | 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 magic_mime_type(const char *restrict filename, char *buf) { | ||||||
|  |     while (sem_wait(&sem_magic) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } else { | ||||||
|  |             critical("Unable to lock magic semaphore"); | ||||||
|  |             return; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     magic_setflags(magic, MAGIC_MIME_TYPE); | ||||||
|  |     const char *type = magic_file(magic, filename); | ||||||
|  |  | ||||||
|  |     if (strstarts(type, "text/")) { | ||||||
|  |         if (strends(filename, ".css")) { | ||||||
|  |             strcpy(buf, "text/css"); | ||||||
|  |             sem_post(&sem_magic); | ||||||
|  |             return; | ||||||
|  |         } else if (strends(filename, ".js")) { | ||||||
|  |             strcpy(buf, "application/javascript"); | ||||||
|  |             sem_post(&sem_magic); | ||||||
|  |             return; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     strcpy(buf, type); | ||||||
|  |     sem_post(&sem_magic); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static void magic_charset(const char *restrict filename, char *buf) { | ||||||
|  |     while (sem_wait(&sem_magic) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } else { | ||||||
|  |             critical("Unable to lock magic semaphore"); | ||||||
|  |             return; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     magic_setflags(magic, MAGIC_MIME_ENCODING); | ||||||
|  |     strcpy(buf, magic_file(magic, filename)); | ||||||
|  |     sem_post(&sem_magic); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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)); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     magic_close(magic); | ||||||
|  |  | ||||||
|  |     sem_destroy(&sem_lock); | ||||||
|  |     sem_destroy(&sem_free); | ||||||
|  |     sem_destroy(&sem_used); | ||||||
|  |     sem_destroy(&sem_cache); | ||||||
|  |     sem_destroy(&sem_magic); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 (streq(entry->filename, filename)) { | ||||||
|  |             // found | ||||||
|  |             return entry; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // not found | ||||||
|  |     return NULL; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static cache_entry_t *cache_get_new_entry(cache_t *cache) { | ||||||
|  |     // globally lock cache | ||||||
|  |     while (sem_wait(&sem_cache) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } else { | ||||||
|  |             return NULL; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // search empty slot | ||||||
|  |     cache_entry_t *entry; | ||||||
|  |     for (int i = 0; i < CACHE_ENTRIES; i++) { | ||||||
|  |         entry = &cache->entries[i]; | ||||||
|  |         if (entry->filename[0] == 0) { | ||||||
|  |             // unlock cache | ||||||
|  |             sem_post(&sem_cache); | ||||||
|  |             return entry; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // unlock cache | ||||||
|  |     sem_post(&sem_cache); | ||||||
|  |  | ||||||
|  |     // 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_sha256()); | ||||||
|  |     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++) { | ||||||
|  |             buf[j] = (char) ((rel_path[j] == '/') ? '_' : rel_path[j]); | ||||||
|  |         } | ||||||
|  |         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 | (mime_is_text(entry->meta.type) ? COMPRESS_UTF8 : 0))) != 0) { | ||||||
|  |                 error("Unable to init compression"); | ||||||
|  |                 compress = 0; | ||||||
|  |                 fclose(comp_file_gz); | ||||||
|  |                 fclose(comp_file_br); | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     for (unsigned long read, avail_in, avail_out; (read = fread(buf, 1, sizeof(buf), file)) > 0;) { | ||||||
|  |         EVP_DigestUpdate(ctx, buf, read); | ||||||
|  |         if (compress) { | ||||||
|  |             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", hc->local.webroot); | ||||||
|  |         if (mkdir(buf, 0755) != 0 && errno != EEXIST) { | ||||||
|  |             critical("Unable to create directory %s", buf); | ||||||
|  |             return 1; | ||||||
|  |         } | ||||||
|  |         errno = 0; | ||||||
|  |  | ||||||
|  |         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 || | ||||||
|  |         sem_init(&sem_cache, 0, 1) != 0 || | ||||||
|  |         sem_init(&sem_magic, 0, 1) != 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; | ||||||
|  |  | ||||||
|  |     entry->flags |= CACHE_DIRTY; | ||||||
|  |     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)); | ||||||
|  |  | ||||||
|  |     while (sem_wait(&sem_free) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } else { | ||||||
|  |             error("Unable to lock semaphore"); | ||||||
|  |             errno = 0; | ||||||
|  |         } | ||||||
|  |         return; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // try to lock buffer | ||||||
|  |     while (sem_wait(&sem_lock) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } 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) { | ||||||
|  |     entry->meta.mtime = stat_mtime(filename); | ||||||
|  |     entry->webroot_len = (unsigned char) strlen(webroot); | ||||||
|  |     strcpy(entry->filename, filename); | ||||||
|  |  | ||||||
|  |     magic_mime_type(filename, entry->meta.type); | ||||||
|  |     magic_charset(filename, entry->meta.charset); | ||||||
|  |  | ||||||
|  |     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 | ||||||
|  |         if (uri->meta->mtime != stat_mtime(uri->filename)) { | ||||||
|  |             // 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(); |  | ||||||
| } |  | ||||||
|  |  | ||||||
|  |  | ||||||
							
								
								
									
										28
									
								
								src/defs.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										28
									
								
								src/defs.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,28 @@ | |||||||
|  | /** | ||||||
|  |  * 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 | ||||||
|  | #define ADDRSTRLEN 39 | ||||||
|  |  | ||||||
|  | #ifndef DEFAULT_HOST | ||||||
|  | #   define DEFAULT_HOST "www.necronda.net" | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | #ifndef SERVER_NAME | ||||||
|  | #   define SERVER_NAME DEFAULT_HOST | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | #endif //SESIMOS_DEF_H | ||||||
							
								
								
									
										88
									
								
								src/lib/compress.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										88
									
								
								src/lib/compress.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,88 @@ | |||||||
|  | /** | ||||||
|  |  * 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; | ||||||
|  |  | ||||||
|  |     if (mode & COMPRESS_GZ) { | ||||||
|  |         ctx->mode |= COMPRESS_GZ; | ||||||
|  |         ctx->gzip.zalloc = Z_NULL; | ||||||
|  |         ctx->gzip.zfree = Z_NULL; | ||||||
|  |         ctx->gzip.opaque = Z_NULL; | ||||||
|  |         ctx->gzip.data_type = (mode & COMPRESS_UTF8) ? Z_TEXT : Z_UNKNOWN; | ||||||
|  |         if (deflateInit2(&ctx->gzip, COMPRESS_LEVEL_GZIP, Z_DEFLATED, 15 + 16, 9, Z_DEFAULT_STRATEGY) != Z_OK) | ||||||
|  |             return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (mode & COMPRESS_BR) { | ||||||
|  |         ctx->mode |= COMPRESS_BR; | ||||||
|  |         if ((ctx->brotli = BrotliEncoderCreateInstance(NULL, NULL, NULL)) == NULL) | ||||||
|  |             return -1; | ||||||
|  |         BrotliEncoderSetParameter(ctx->brotli, BROTLI_PARAM_MODE, (mode & COMPRESS_UTF8) ? BROTLI_MODE_TEXT : ((mode & COMPRESS_WOFF) ? BROTLI_MODE_FONT : 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); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int compress_brotli(BrotliEncoderState *ctx, const char *in, unsigned long *in_len, char *out, unsigned long *out_len, int finish) { | ||||||
|  |     int ret = BrotliEncoderCompressStream( | ||||||
|  |             ctx, 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; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int compress_gzip(z_stream *gzip, const char *in, unsigned long *in_len, char *out, unsigned long *out_len, int finish) { | ||||||
|  |     gzip->next_in = (unsigned char*) in; | ||||||
|  |     gzip->avail_in = *in_len; | ||||||
|  |     gzip->next_out = (unsigned char*) out; | ||||||
|  |     gzip->avail_out = *out_len; | ||||||
|  |     int ret = deflate(gzip, finish ? Z_FINISH : Z_NO_FLUSH); | ||||||
|  |     *in_len = gzip->avail_in; | ||||||
|  |     *out_len = gzip->avail_out; | ||||||
|  |     return ret; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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) { | ||||||
|  |         return compress_gzip(&ctx->gzip, in, in_len, out, out_len, finish); | ||||||
|  |     } else if (mode & COMPRESS_BR) { | ||||||
|  |         return compress_brotli(ctx->brotli, in, in_len, out, out_len, finish); | ||||||
|  |     } else { | ||||||
|  |         errno = EINVAL; | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int compress_free(compress_ctx *ctx) { | ||||||
|  |     if (ctx->brotli != NULL) { | ||||||
|  |         BrotliEncoderDestroyInstance(ctx->brotli); | ||||||
|  |         ctx->brotli = NULL; | ||||||
|  |     } | ||||||
|  |     if (ctx->mode & COMPRESS_GZ) { | ||||||
|  |         deflateEnd(&ctx->gzip); | ||||||
|  |     } | ||||||
|  |     ctx->mode = 0; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
							
								
								
									
										38
									
								
								src/lib/compress.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										38
									
								
								src/lib/compress.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,38 @@ | |||||||
|  | /** | ||||||
|  |  * 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 | ||||||
|  | #define COMPRESS_UTF8 4 | ||||||
|  | #define COMPRESS_WOFF 8 | ||||||
|  |  | ||||||
|  | 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 | ||||||
							
								
								
									
										228
									
								
								src/lib/config.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										228
									
								
								src/lib/config.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,228 @@ | |||||||
|  | /** | ||||||
|  |  * 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 "utils.h" | ||||||
|  |  | ||||||
|  | #include <stdio.h> | ||||||
|  | #include <string.h> | ||||||
|  | #include <stdlib.h> | ||||||
|  |  | ||||||
|  | config_t config; | ||||||
|  |  | ||||||
|  | static int config_parse_line(char *line, char *section, int *i, int *j) { | ||||||
|  |     int mode = 0; | ||||||
|  |     char *source, *target; | ||||||
|  |  | ||||||
|  |     char *ptr = line; | ||||||
|  |     char *comment = strpbrk(ptr, "#\r\n"); | ||||||
|  |     if (comment != NULL) comment[0] = 0; | ||||||
|  |  | ||||||
|  |     unsigned long len = strlen(ptr); | ||||||
|  |     char *end_ptr = (len > 0) ? ptr + len - 1 : ptr; | ||||||
|  |     while (end_ptr[0] == ' ' || end_ptr[0] == '\t') { | ||||||
|  |         end_ptr[0] = 0; | ||||||
|  |         end_ptr--; | ||||||
|  |     } | ||||||
|  |     len = strlen(ptr); | ||||||
|  |     if (len == 0) return 0; | ||||||
|  |  | ||||||
|  |     if (ptr[0] == '[') { | ||||||
|  |         if (ptr[len - 1] != ']') return -1; | ||||||
|  |         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) return -1; | ||||||
|  |             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) return -1; | ||||||
|  |             snprintf(config.certs[*j].name, sizeof(config.certs[*j].name), "%.*s", l, ptr); | ||||||
|  |             ++(*j); | ||||||
|  |             *section = 'c'; | ||||||
|  |         } else { | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |         return 0; | ||||||
|  |     } 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 { | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |     } 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 { | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |     } 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) { | ||||||
|  |                 return -1; | ||||||
|  |             } 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) { | ||||||
|  |                 return -1; | ||||||
|  |             } 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) { | ||||||
|  |                 return -1; | ||||||
|  |             } 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) { | ||||||
|  |                 return -1; | ||||||
|  |             } else { | ||||||
|  |                 hc->type = CONFIG_TYPE_REVERSE_PROXY; | ||||||
|  |             } | ||||||
|  |         } else if (streq(ptr, "http")) { | ||||||
|  |             if (hc->type != 0 && hc->type != CONFIG_TYPE_REVERSE_PROXY) { | ||||||
|  |                 return -1; | ||||||
|  |             } else { | ||||||
|  |                 hc->type = CONFIG_TYPE_REVERSE_PROXY; | ||||||
|  |                 hc->proxy.enc = 0; | ||||||
|  |             } | ||||||
|  |             return 0; | ||||||
|  |         } else if (streq(ptr, "https")) { | ||||||
|  |             if (hc->type != 0 && hc->type != CONFIG_TYPE_REVERSE_PROXY) { | ||||||
|  |                 return -1; | ||||||
|  |             } else { | ||||||
|  |                 hc->type = CONFIG_TYPE_REVERSE_PROXY; | ||||||
|  |                 hc->proxy.enc = 1; | ||||||
|  |             } | ||||||
|  |             return 0; | ||||||
|  |         } else { | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |     } else { | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     while (source[0] == ' ' || source[0] == '\t') source++; | ||||||
|  |     if (strlen(source) == 0) return -1; | ||||||
|  |  | ||||||
|  |     if (target != NULL) { | ||||||
|  |         strcpy(target, source); | ||||||
|  |     } else if (mode == 1) { | ||||||
|  |         if (streq(source, "forbidden")) { | ||||||
|  |             config.hosts[*i - 1].local.dir_mode = URI_DIR_MODE_FORBIDDEN; | ||||||
|  |         } else if (streq(source, "info")) { | ||||||
|  |             config.hosts[*i - 1].local.dir_mode = URI_DIR_MODE_INFO; | ||||||
|  |         } else if (streq(source, "list")) { | ||||||
|  |             config.hosts[*i - 1].local.dir_mode = URI_DIR_MODE_LIST; | ||||||
|  |         } else { | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |     } else if (mode == 2) { | ||||||
|  |         config.hosts[*i - 1].proxy.port = (unsigned short) strtoul(source, NULL, 10); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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, j = 0; | ||||||
|  |     char section = 0; | ||||||
|  |  | ||||||
|  |     char *line = NULL; | ||||||
|  |     size_t line_len = 0; | ||||||
|  |     for (int line_num = 1; getline(&line, &line_len, file) != -1; line_num++) { | ||||||
|  |         if (config_parse_line(line, §ion, &i, &j) != 0) { | ||||||
|  |             critical("Unable to parse config file (line %i)", line_num); | ||||||
|  |             free(line); | ||||||
|  |             fclose(file); | ||||||
|  |             return -2; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     free(line); | ||||||
|  |     fclose(file); | ||||||
|  |  | ||||||
|  |     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; | ||||||
|  |             while (webroot[strlen(webroot) - 1] == '/') webroot[strlen(webroot) - 1] = 0; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if (hc->cert_name[0] == 0) { | ||||||
|  |             critical("Unable to parse config file: host config (%s) does not contain a certificate", hc->name); | ||||||
|  |             return -2; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         int found = 0; | ||||||
|  |         for (int m = 0; m < j; m++) { | ||||||
|  |             if (streq(config.certs[m].name, hc->cert_name)) { | ||||||
|  |                 hc->cert = m; | ||||||
|  |                 found = 1; | ||||||
|  |                 break; | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |         if (!found) { | ||||||
|  |             critical("Unable to parse config file: certificate (%s) for host config (%s) not found", hc->cert_name, hc->name); | ||||||
|  |             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 (streq(hc->name, host)) 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; | ||||||
|  | } | ||||||
							
								
								
									
										64
									
								
								src/lib/config.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										64
									
								
								src/lib/config.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,64 @@ | |||||||
|  | /** | ||||||
|  |  * 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]; | ||||||
|  | } 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 | ||||||
							
								
								
									
										87
									
								
								src/lib/error.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										87
									
								
								src/lib/error.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,87 @@ | |||||||
|  | /** | ||||||
|  |  * 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 "../logger.h" | ||||||
|  |  | ||||||
|  | #include <errno.h> | ||||||
|  | #include <string.h> | ||||||
|  | #include <netdb.h> | ||||||
|  |  | ||||||
|  | extern const char *sock_error_str(unsigned long err); | ||||||
|  | extern const char *http_error_str(int err); | ||||||
|  | extern const char *MMDB_strerror(int err); | ||||||
|  | extern const char *ERR_reason_error_string(unsigned long err); | ||||||
|  |  | ||||||
|  | static int error_compress(unsigned long err) { | ||||||
|  |     int comp = ((int) err & 0xFFFF) | (((int) err >> 8) & 0xFF0000); | ||||||
|  |     if (err & 0xFF0000) warning("Lossy error code compression! (%08lX -> %08X)", err, comp); | ||||||
|  |     return comp; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static unsigned long error_decompress(int err) { | ||||||
|  |     return (err & 0xFFFF) | ((err << 8) & 0xFF000000); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const char *error_str(int err_no, char *buf, int buf_len) { | ||||||
|  |     buf[0] = 0; | ||||||
|  |     int e = err_no & 0x00FFFFFF; | ||||||
|  |     switch (err_no >> 24) { | ||||||
|  |         case 0x00: return strerror_r(e, buf, buf_len); | ||||||
|  |         case 0x01: return sock_error_str(error_decompress(e)); | ||||||
|  |         case 0x02: return ERR_reason_error_string(error_decompress(e)); | ||||||
|  |         case 0x03: return MMDB_strerror(e); | ||||||
|  |         case 0x04: return http_error_str(e); | ||||||
|  |         case 0x05: return gai_strerror(e); | ||||||
|  |     } | ||||||
|  |     return buf; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void error_ssl(unsigned long err) { | ||||||
|  |     errno = 0x01000000 | error_compress(err); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void error_ssl_err(unsigned long err) { | ||||||
|  |     errno = 0x02000000 | error_compress(err); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void error_mmdb(int err) { | ||||||
|  |     errno = 0x03000000 | err; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void error_http(int err) { | ||||||
|  |     errno = 0x04000000 | err; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void error_gai(int err) { | ||||||
|  |     errno = 0x05000000 | err; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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_ssl_err() { | ||||||
|  |     return error_get(0x02); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int error_get_mmdb() { | ||||||
|  |     return error_get(0x03); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int error_get_http() { | ||||||
|  |     return error_get(0x04); | ||||||
|  | } | ||||||
							
								
								
									
										32
									
								
								src/lib/error.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										32
									
								
								src/lib/error.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,32 @@ | |||||||
|  | /** | ||||||
|  |  * 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(unsigned long err); | ||||||
|  |  | ||||||
|  | void error_ssl_err(unsigned long err); | ||||||
|  |  | ||||||
|  | void error_mmdb(int err); | ||||||
|  |  | ||||||
|  | void error_http(int err); | ||||||
|  |  | ||||||
|  | void error_gai(int err); | ||||||
|  |  | ||||||
|  | int error_get_sys(); | ||||||
|  |  | ||||||
|  | int error_get_ssl(); | ||||||
|  |  | ||||||
|  | int error_get_mmdb(); | ||||||
|  |  | ||||||
|  | int error_get_http(); | ||||||
|  |  | ||||||
|  | #endif //SESIMOS_ERROR_H | ||||||
							
								
								
									
										420
									
								
								src/lib/fastcgi.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										420
									
								
								src/lib/fastcgi.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,420 @@ | |||||||
|  | /** | ||||||
|  |  * sesimos - secure, simple, modern web server | ||||||
|  |  * @brief FastCGI interface implementation | ||||||
|  |  * @file src/lib/fastcgi.c | ||||||
|  |  * @author Lorenz Stechauner | ||||||
|  |  * @date 2020-12-26 | ||||||
|  |  */ | ||||||
|  |  | ||||||
|  | #include "../defs.h" | ||||||
|  | #include "fastcgi.h" | ||||||
|  | #include "utils.h" | ||||||
|  | #include "../logger.h" | ||||||
|  | #include "list.h" | ||||||
|  | #include "../workers.h" | ||||||
|  |  | ||||||
|  | #include <sys/un.h> | ||||||
|  | #include <sys/socket.h> | ||||||
|  | #include <string.h> | ||||||
|  | #include <fcntl.h> | ||||||
|  | #include <unistd.h> | ||||||
|  | #include <errno.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 { | ||||||
|  |         *((int *) ptr) = htonl(0x80000000 | key_len); | ||||||
|  |         ptr += 4; | ||||||
|  |     } | ||||||
|  |     if (val_len <= 127) { | ||||||
|  |         ptr[0] = (char) (val_len & 0x7F); | ||||||
|  |         ptr++; | ||||||
|  |     } else { | ||||||
|  |         *((int *) ptr) = htonl(0x80000000 | val_len); | ||||||
|  |         ptr += 4; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     memcpy(ptr, key, key_len); | ||||||
|  |     ptr += key_len; | ||||||
|  |     memcpy(ptr, value, val_len); | ||||||
|  |     ptr += val_len; | ||||||
|  |  | ||||||
|  |     return ptr; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_send_data(fastcgi_cnx_t *cnx, unsigned char type, unsigned short len, void *data) { | ||||||
|  |     // build header | ||||||
|  |     FCGI_Header header = { | ||||||
|  |             .version = FCGI_VERSION_1, | ||||||
|  |             .type = type, | ||||||
|  |             .requestId = htons(cnx->req_id), | ||||||
|  |             .contentLength = htons(len), | ||||||
|  |             .paddingLength = 0, | ||||||
|  |             .reserved = 0, | ||||||
|  |     }; | ||||||
|  |  | ||||||
|  |     // send FastCGI header with MSG_MORE flag | ||||||
|  |     if (sock_send_x(&cnx->socket, &header, sizeof(header), (len != 0) ? MSG_MORE : 0) == -1) { | ||||||
|  |         error("Unable to send to FastCGI socket"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // send data (if available) | ||||||
|  |     if (sock_send_x(&cnx->socket, data, len, 0) == -1) { | ||||||
|  |         error("Unable to send to FastCGI socket"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // return bytes sent totally | ||||||
|  |     return len + (int) sizeof(header); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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->webroot = uri->webroot; | ||||||
|  |     conn->err = NULL; | ||||||
|  |     conn->fd_err_bytes = 0; | ||||||
|  |     sock_init(&conn->out, 0, SOCK_PIPE); | ||||||
|  |  | ||||||
|  |     conn->socket.enc = 0; | ||||||
|  |     if ((conn->socket.socket = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) { | ||||||
|  |         error("Unable to create unix socket"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     struct sockaddr_un sock_addr = { AF_UNIX }; | ||||||
|  |     if (conn->mode == FASTCGI_BACKEND_PHP) { | ||||||
|  |         strcpy(sock_addr.sun_path, PHP_FPM_SOCKET); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (connect(conn->socket.socket, (struct sockaddr *) &sock_addr, sizeof(sock_addr)) != 0) { | ||||||
|  |         error("Unable to connect to FastCGI (unix) socket"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     FCGI_BeginRequestBody begin = { | ||||||
|  |             .role = htons(FCGI_RESPONDER), | ||||||
|  |             .flags = 0, | ||||||
|  |             .reserved = {0}, | ||||||
|  |     }; | ||||||
|  |  | ||||||
|  |     if (fastcgi_send_data(conn, FCGI_BEGIN_REQUEST, sizeof(begin), &begin) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     char param_buf[4096], buf0[256], *param_ptr = param_buf; | ||||||
|  |     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", ntohs(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", ntohs(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.fields); 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)); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (fastcgi_send_data(conn, FCGI_PARAMS, param_ptr - param_buf, param_buf) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     if (fastcgi_send_data(conn, FCGI_PARAMS, 0, NULL) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     int pipes[2][2]; | ||||||
|  |     if (pipe(pipes[0]) == -1 || pipe(pipes[1]) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     conn->fd_out = pipes[1][1]; | ||||||
|  |     conn->out.socket = pipes[1][0]; | ||||||
|  |  | ||||||
|  |     conn->fd_err = pipes[0][1]; | ||||||
|  |     conn->err = fdopen(pipes[0][0], "r"); | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_close_cnx(fastcgi_cnx_t *cnx) { | ||||||
|  |     int e = errno; | ||||||
|  |  | ||||||
|  |     if (cnx->err) fclose(cnx->err); | ||||||
|  |     cnx->err = NULL; | ||||||
|  |     if (cnx->socket.socket) sock_close(&cnx->socket); | ||||||
|  |  | ||||||
|  |     sock_close(&cnx->out); | ||||||
|  |     close(cnx->fd_err); | ||||||
|  |     close(cnx->fd_out); | ||||||
|  |  | ||||||
|  |     errno = e; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_close_stdin(fastcgi_cnx_t *cnx) { | ||||||
|  |     return (fastcgi_send_data(cnx, FCGI_STDIN, 0, NULL) == -1) ? -1 : 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_php_error(fastcgi_cnx_t *cnx, char *err_msg) { | ||||||
|  |     char *line = NULL, *line_ptr = NULL; | ||||||
|  |     size_t line_len = 0; | ||||||
|  |     int err = 0; | ||||||
|  |  | ||||||
|  |     log_lvl_t msg_type = LOG_INFO; | ||||||
|  |  | ||||||
|  |     for (long ret; cnx->fd_err_bytes > 0 && (ret = getline(&line, &line_len, cnx->err)) != -1; cnx->fd_err_bytes -= ret) { | ||||||
|  |         if (ret > 0) line[ret - 1] = 0; | ||||||
|  |         line_ptr = line; | ||||||
|  |  | ||||||
|  |         if (strstarts(line_ptr, "PHP message: ")) { | ||||||
|  |             line_ptr += 13; | ||||||
|  |             if (strstarts(line_ptr, "PHP Warning:  ")) { | ||||||
|  |                 msg_type = LOG_WARNING; | ||||||
|  |             } else if (strstarts(line_ptr, "PHP Fatal error:  ")) { | ||||||
|  |                 msg_type = LOG_ERROR; | ||||||
|  |             } else if (strstarts(line_ptr, "PHP Parse error:  ")) { | ||||||
|  |                 msg_type = LOG_ERROR; | ||||||
|  |             } else if (strstarts(line_ptr, "PHP Notice:  ")) { | ||||||
|  |                 msg_type = LOG_NOTICE; | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         logmsgf(msg_type, "%s", line_ptr); | ||||||
|  |  | ||||||
|  |         if (err_msg && msg_type <= LOG_ERROR && line_ptr != line) { | ||||||
|  |             strcpy_rem_webroot(err_msg, line_ptr, cnx->webroot); | ||||||
|  |             err = 1; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // cleanup | ||||||
|  |     free(line); | ||||||
|  |     return err; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_recv_frame(fastcgi_cnx_t *cnx) { | ||||||
|  |     FCGI_Header header; | ||||||
|  |     unsigned short req_id, content_len; | ||||||
|  |  | ||||||
|  |     if (sock_recv_x(&cnx->socket, &header, sizeof(header), 0) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     req_id = ntohs(header.requestId); | ||||||
|  |     content_len = ntohs(header.contentLength); | ||||||
|  |  | ||||||
|  |     if (req_id != cnx->req_id) { | ||||||
|  |         warning("Invalid request id from FastCGI socket"); | ||||||
|  |         char content[256 * 256]; | ||||||
|  |         sock_recv_x(&cnx->socket, content, content_len + header.paddingLength, 0); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (header.type == FCGI_STDOUT || header.type == FCGI_STDERR) { | ||||||
|  |         char buf[256]; | ||||||
|  |         if (header.type == FCGI_STDOUT) { | ||||||
|  |             uint64_t len = content_len; | ||||||
|  |             if (write(cnx->fd_out, &len, sizeof(len)) == -1) | ||||||
|  |                 return -1; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         int fd = cnx->fd_out; | ||||||
|  |         if (header.type == FCGI_STDERR) { | ||||||
|  |             fd = cnx->fd_err; | ||||||
|  |             cnx->fd_err_bytes += content_len + 1; | ||||||
|  |         } | ||||||
|  |         for (long ret, sent = 0; sent < content_len; sent += ret) { | ||||||
|  |             if ((ret = splice(cnx->socket.socket, 0, fd, 0, content_len - sent, 0)) == -1) { | ||||||
|  |                 if (errno == EINTR) { | ||||||
|  |                     errno = 0, ret = 0; | ||||||
|  |                     continue; | ||||||
|  |                 } else { | ||||||
|  |                     return -1; | ||||||
|  |                 } | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |         if (header.type == FCGI_STDERR) write(fd, "\n", 1); | ||||||
|  |  | ||||||
|  |         if (sock_recv_x(&cnx->socket, buf, header.paddingLength, 0) == -1) | ||||||
|  |             return -1; | ||||||
|  |  | ||||||
|  |         return header.type; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     char content[256 * 256]; | ||||||
|  |     if (sock_recv_x(&cnx->socket, content, content_len + header.paddingLength, 0) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     if (header.type == FCGI_END_REQUEST) { | ||||||
|  |         FCGI_EndRequestBody *body = (FCGI_EndRequestBody *) content; | ||||||
|  |         cnx->app_status = ntohl(body->appStatus); | ||||||
|  |         if (body->protocolStatus != FCGI_REQUEST_COMPLETE) | ||||||
|  |             error("FastCGI protocol error: %i", body->protocolStatus); | ||||||
|  |     } else { | ||||||
|  |         warning("Unknown FastCGI type: %i", header.type); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return header.type; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long fastcgi_send(fastcgi_cnx_t *cnx, sock *client) { | ||||||
|  |     char buf[CHUNK_SIZE]; | ||||||
|  |     return sock_splice_all(client, &cnx->out, buf, sizeof(buf)); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_header(fastcgi_cnx_t *cnx, http_res *res, char *err_msg) { | ||||||
|  |     long ret, len; | ||||||
|  |     char content[CLIENT_MAX_HEADER_SIZE]; | ||||||
|  |  | ||||||
|  |     if ((len = sock_recv_chunk_header(&cnx->out)) == -1) { | ||||||
|  |         res->status = http_get_status(500); | ||||||
|  |         sprintf(err_msg, "Unable to communicate with FastCGI socket."); | ||||||
|  |         error("Unable to receive from FastCGI socket (1)"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if ((ret = sock_recv_x(&cnx->out, content, len, 0)) == -1) { | ||||||
|  |         res->status = http_get_status(500); | ||||||
|  |         sprintf(err_msg, "Unable to communicate with FastCGI socket."); | ||||||
|  |         error("Unable to receive from FastCGI socket (2)"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |     content[ret] = 0; | ||||||
|  |  | ||||||
|  |     char *buf = content; | ||||||
|  |     char *h_pos = strstr(content, "\r\n\r\n"); | ||||||
|  |     if (h_pos == NULL) { | ||||||
|  |         error("Unable to parse header: End of header not found"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |     long header_len = h_pos - content + 4; | ||||||
|  |  | ||||||
|  |     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 -1; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (fastcgi_php_error(cnx, err_msg) != 0) { | ||||||
|  |         res->status = http_get_status(500); | ||||||
|  |         return 1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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_dump(fastcgi_cnx_t *cnx, char *buf, long len) { | ||||||
|  |     return sock_recv_x(&cnx->socket, buf, len, 0) == -1 ? -1 : 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_receive(fastcgi_cnx_t *cnx, sock *client, unsigned long len) { | ||||||
|  |     char buf[CHUNK_SIZE]; | ||||||
|  |  | ||||||
|  |     for (long to_send = (long) len, ret; to_send > 0; to_send -= ret) { | ||||||
|  |         if ((ret = sock_recv(client, buf, (to_send > sizeof(buf)) ? sizeof(buf) : to_send, 0)) <= 0) { | ||||||
|  |             error("Unable to receive"); | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if (fastcgi_send_data(cnx, FCGI_STDIN, ret, buf) == -1) | ||||||
|  |             return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_receive_chunked(fastcgi_cnx_t *cnx, sock *client) { | ||||||
|  |     for (long ret;;) { | ||||||
|  |         if ((ret = sock_recv_chunk_header(client)) < 0) { | ||||||
|  |             return (int) ret; | ||||||
|  |         } else if (ret == 0) { | ||||||
|  |             break; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if ((ret = fastcgi_receive(cnx, client, 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_TIMEOUT 3600 | ||||||
|  |  | ||||||
|  | #define FASTCGI_BACKEND_PHP 1 | ||||||
|  |  | ||||||
|  | #ifndef PHP_FPM_SOCKET | ||||||
|  | #   define PHP_FPM_SOCKET "/var/run/php-fpm/php-fpm.sock" | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | typedef struct { | ||||||
|  |     int mode; | ||||||
|  |     sock socket, out; | ||||||
|  |     int fd_err, fd_out; | ||||||
|  |     long fd_err_bytes; | ||||||
|  |     FILE *err; | ||||||
|  |     unsigned short req_id; | ||||||
|  |     int app_status; | ||||||
|  |     const char *webroot; | ||||||
|  |     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_cnx(fastcgi_cnx_t *cnx); | ||||||
|  |  | ||||||
|  | int fastcgi_close_stdin(fastcgi_cnx_t *cnx); | ||||||
|  |  | ||||||
|  | int fastcgi_php_error(fastcgi_cnx_t *cnx, char *err_msg); | ||||||
|  |  | ||||||
|  | int fastcgi_recv_frame(fastcgi_cnx_t *cnx); | ||||||
|  |  | ||||||
|  | int fastcgi_header(fastcgi_cnx_t *cnx, http_res *res, char *err_msg); | ||||||
|  |  | ||||||
|  | long fastcgi_send(fastcgi_cnx_t *cnx, sock *client); | ||||||
|  |  | ||||||
|  | int fastcgi_dump(fastcgi_cnx_t *cnx, char *buf, long len); | ||||||
|  |  | ||||||
|  | int fastcgi_receive(fastcgi_cnx_t *cnx, sock *client, unsigned long len); | ||||||
|  |  | ||||||
|  | int fastcgi_receive_chunked(fastcgi_cnx_t *cnx, sock *client); | ||||||
|  |  | ||||||
|  |  | ||||||
|  |  | ||||||
|  | #endif //SESIMOS_FASTCGI_H | ||||||
							
								
								
									
										208
									
								
								src/lib/geoip.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										208
									
								
								src/lib/geoip.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,208 @@ | |||||||
|  | /** | ||||||
|  |  * 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 "utils.h" | ||||||
|  |  | ||||||
|  | #include <memory.h> | ||||||
|  | #include <dirent.h> | ||||||
|  | #include <errno.h> | ||||||
|  |  | ||||||
|  | static MMDB_s mmdbs[GEOIP_MAX_MMDB]; | ||||||
|  |  | ||||||
|  | static void mmdb_error(int err) { | ||||||
|  |     if (err == MMDB_SUCCESS) { | ||||||
|  |         errno = 0; | ||||||
|  |     } else if (err == MMDB_IO_ERROR) { | ||||||
|  |         // errno already set | ||||||
|  |     } else { | ||||||
|  |         error_mmdb(err); | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 (!strends(entry->d_name, ".mmdb")) | ||||||
|  |             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) { | ||||||
|  |             mmdb_error(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) { | ||||||
|  |             mmdb_error(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) { | ||||||
|  |             mmdb_error(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 | ||||||
							
								
								
									
										451
									
								
								src/lib/http.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										451
									
								
								src/lib/http.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,451 @@ | |||||||
|  | /** | ||||||
|  |  * 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> | ||||||
|  | #include <errno.h> | ||||||
|  |  | ||||||
|  | static int http_error(int err) { | ||||||
|  |     if (err == 0) { | ||||||
|  |         errno = 0; | ||||||
|  |     } else if (err == HTTP_ERROR_SYSCALL) { | ||||||
|  |         // errno already set | ||||||
|  |     } else { | ||||||
|  |         error_http(err); | ||||||
|  |     } | ||||||
|  |     return -1; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const char *http_error_str(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"; | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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->last_field_num = -1; | ||||||
|  |     hdr->fields = list_create(sizeof(http_field), HTTP_INIT_HEADER_FIELD_NUM); | ||||||
|  |     if (hdr->fields == NULL) | ||||||
|  |         return http_error(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 http_error(HTTP_ERROR_GENERAL); | ||||||
|  |  | ||||||
|  |     char *pos1 = (char *) buf, *pos2 = (char *) end_ptr; | ||||||
|  |     if (buf[0] == ' ' || buf[0] == '\t') { | ||||||
|  |         if (hdr->last_field_num == -1) | ||||||
|  |             return http_error(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 http_error(HTTP_ERROR_GENERAL); | ||||||
|  |  | ||||||
|  |     long len1 = pos1 - buf; | ||||||
|  |  | ||||||
|  |     pos1++; | ||||||
|  |     str_trim_lws(&pos1, &pos2); | ||||||
|  |     long len2 = pos2 - pos1; | ||||||
|  |  | ||||||
|  |     char header_name[256]; | ||||||
|  |     sprintf(header_name, "%.*s", (int) len1, buf); | ||||||
|  |  | ||||||
|  |     int field_num = list_size(hdr->fields); | ||||||
|  |     int found = http_get_header_field_num(hdr, header_name); | ||||||
|  |     if (!(flags & HTTP_MERGE_FIELDS) || found == -1) { | ||||||
|  |         if (http_add_header_field_len(hdr, buf, len1, pos1, len2 < 0 ? 0 : len2) != 0) | ||||||
|  |             return http_error(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 http_error(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 http_error(HTTP_ERROR_HEADER_MALFORMED); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     ptr = buf; | ||||||
|  |     while (header_len > (ptr - buf + 2)) { | ||||||
|  |         pos0 = strstr(ptr, "\r\n"); | ||||||
|  |         if (pos0 == NULL) | ||||||
|  |             return http_error(HTTP_ERROR_HEADER_MALFORMED); | ||||||
|  |  | ||||||
|  |         if (req->version[0] == 0) { | ||||||
|  |             pos1 = (char *) strchr(ptr, ' ') + 1; | ||||||
|  |             if (pos1 == NULL) | ||||||
|  |                 return http_error(HTTP_ERROR_HEADER_MALFORMED); | ||||||
|  |  | ||||||
|  |             if (pos1 - ptr - 1 >= sizeof(req->method)) | ||||||
|  |                 return http_error(HTTP_ERROR_HEADER_MALFORMED); | ||||||
|  |  | ||||||
|  |             for (int i = 0; i < (pos1 - ptr - 1); i++) { | ||||||
|  |                 if (ptr[i] < 'A' || ptr[i] > 'Z') | ||||||
|  |                     return http_error(HTTP_ERROR_HEADER_MALFORMED); | ||||||
|  |             } | ||||||
|  |             snprintf(req->method, sizeof(req->method), "%.*s", (int) (pos1 - ptr - 1), ptr); | ||||||
|  |  | ||||||
|  |             pos2 = (char *) strchr(pos1, ' ') + 1; | ||||||
|  |             if (pos2 == NULL) | ||||||
|  |                 return http_error(HTTP_ERROR_HEADER_MALFORMED); | ||||||
|  |  | ||||||
|  |             if (memcmp(pos2, "HTTP/", 5) != 0 || memcmp(pos2 + 8, "\r\n", 2) != 0) | ||||||
|  |                 return http_error(HTTP_ERROR_INVALID_VERSION); | ||||||
|  |  | ||||||
|  |             len = pos2 - pos1 - 1; | ||||||
|  |             if (len >= 2048) | ||||||
|  |                 return http_error(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 http_error(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; | ||||||
|  |     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; | ||||||
|  |  | ||||||
|  |     if (sock_recv_x(client, buf, header_len, 0) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const char *http_get_header_field(const http_hdr *hdr, const char *field_name) { | ||||||
|  |     int num = http_get_header_field_num(hdr, field_name); | ||||||
|  |     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) { | ||||||
|  |     for (int i = 0; i < list_size(hdr->fields); i++) { | ||||||
|  |         if (strcasecmp(field_name, http_field_get_name(&hdr->fields[i])) == 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_x(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_x(server, buf, off, 0) != off) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const http_status *http_get_status(status_code_t 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(status_code_t status_code) { | ||||||
|  |     for (int i = 0; i < http_status_messages_size; i++) { | ||||||
|  |         if (http_status_messages[i].code == status_code) { | ||||||
|  |             return &http_status_messages[i]; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     return NULL; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const char *http_get_status_color(status_code_t status_code) { | ||||||
|  |     if (status_code == 304) return HTTP_2XX_STR; | ||||||
|  |     switch (status_code / 100) { | ||||||
|  |         case 1: return HTTP_1XX_STR; | ||||||
|  |         case 2: return HTTP_2XX_STR; | ||||||
|  |         case 3: return HTTP_3XX_STR; | ||||||
|  |         case 4: return HTTP_4XX_STR; | ||||||
|  |         case 5: return HTTP_5XX_STR; | ||||||
|  |         default: return ""; | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | char *http_format_date(time_t ts, char *buf, size_t size) { | ||||||
|  |     struct tm time_info; | ||||||
|  |     strftime(buf, size, "%a, %d %b %Y %H:%M:%S GMT", gmtime_r(&ts, &time_info)); | ||||||
|  |     return buf; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | char *http_get_date(char *buf, size_t size) { | ||||||
|  |     time_t raw_time; | ||||||
|  |     time(&raw_time); | ||||||
|  |     return http_format_date(raw_time, buf, size); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const http_doc_info *http_get_status_info(status_code_t status_code) { | ||||||
|  |     static const 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 (status_code == 304) return &info[1]; | ||||||
|  |     switch (status_code / 100) { | ||||||
|  |         case 1: return &info[0]; | ||||||
|  |         case 2: return &info[1]; | ||||||
|  |         case 3: return &info[2]; | ||||||
|  |         case 4: // see case 5 | ||||||
|  |         case 5: return &info[3]; | ||||||
|  |         default: 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 (strcontains(accept_encoding, "br")) { | ||||||
|  |             return COMPRESS_BR; | ||||||
|  |         } else if (strcontains(accept_encoding, "gzip")) { | ||||||
|  |             return COMPRESS_GZ; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
							
								
								
									
										191
									
								
								src/lib/http.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										191
									
								
								src/lib/http.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,191 @@ | |||||||
|  | /** | ||||||
|  |  * 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 | ||||||
|  |  | ||||||
|  | #ifndef SERVER_STR | ||||||
|  | #   define SERVER_STR "sesimos" | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | #ifndef SERVER_STR_HTML | ||||||
|  | #   define SERVER_STR_HTML "sesimos web server" | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | typedef unsigned short status_code_t; | ||||||
|  |  | ||||||
|  | typedef struct { | ||||||
|  |     status_code_t code:10; | ||||||
|  |     unsigned char type:3; | ||||||
|  |     char msg[64]; | ||||||
|  | } http_status; | ||||||
|  |  | ||||||
|  | typedef struct { | ||||||
|  |     status_code_t 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 { | ||||||
|  |     status_code_t 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[]; | ||||||
|  |  | ||||||
|  | const char *http_error_str(int err); | ||||||
|  |  | ||||||
|  | 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); | ||||||
|  |  | ||||||
|  | int http_get_header_field_num(const http_hdr *hdr, const char *field_name); | ||||||
|  |  | ||||||
|  | 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(status_code_t status_code); | ||||||
|  |  | ||||||
|  | const http_status_msg *http_get_error_msg(status_code_t status_code); | ||||||
|  |  | ||||||
|  | const char *http_get_status_color(status_code_t status_code); | ||||||
|  |  | ||||||
|  | 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(status_code_t status_code); | ||||||
|  |  | ||||||
|  | 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"; | ||||||
							
								
								
									
										115
									
								
								src/lib/include/fastcgi.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										115
									
								
								src/lib/include/fastcgi.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,115 @@ | |||||||
|  | /** | ||||||
|  |  * 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 short requestId; | ||||||
|  |     unsigned short contentLength; | ||||||
|  |     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 short role; | ||||||
|  |     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 int appStatus; | ||||||
|  |     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 | ||||||
							
								
								
									
										150
									
								
								src/lib/list.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										150
									
								
								src/lib/list.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,150 @@ | |||||||
|  |  | ||||||
|  | #include "list.h" | ||||||
|  |  | ||||||
|  | #include <malloc.h> | ||||||
|  | #include <memory.h> | ||||||
|  | #include <errno.h> | ||||||
|  |  | ||||||
|  | #define FACTOR 4 | ||||||
|  | #define meta(ptr) ((list_meta_t *) ((unsigned char *) (ptr) - sizeof(list_meta_t))) | ||||||
|  | #define data(ptr) ((unsigned char *) (ptr) + sizeof(list_meta_t)) | ||||||
|  |  | ||||||
|  | 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; | ||||||
|  |     } else if (new_size == list->max_size) { | ||||||
|  |         return list; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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); | ||||||
|  |     if (list_ptr == NULL) | ||||||
|  |         return NULL; | ||||||
|  |  | ||||||
|  |     memset(list_ptr, 0, 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 data(list_ptr); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int list_size(const void *list_ptr) { | ||||||
|  |     return meta(list_ptr)->size; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int list_find(void *list_ptr, void *elem) { | ||||||
|  |     list_meta_t *list = meta(list_ptr); | ||||||
|  |     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; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int list_contains(void *list_ptr, void *elem) { | ||||||
|  |     return list_find(list_ptr, elem) != -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) { | ||||||
|  |         memcpy(ptr, elem, meta(list_ptr)->elem_size); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return list_ptr; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void *list_insert_ptr(void *list_ptr, void **elem, int n) { | ||||||
|  |     list_meta_t *list = meta(list_ptr); | ||||||
|  |     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 = data(list); | ||||||
|  |  | ||||||
|  |     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; | ||||||
|  |     memset(*elem, 0, list->elem_size); | ||||||
|  |  | ||||||
|  |     list->size++; | ||||||
|  |     return array; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 = meta(list_ptr); | ||||||
|  |     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--; | ||||||
|  |     memset(array + list->size * list->elem_size, 0, list->elem_size); | ||||||
|  |  | ||||||
|  |     if (list->size <= list->max_size / FACTOR * 3 / 4 && list->max_size / FACTOR >= list->init_size) { | ||||||
|  |         if ((list = list_resize(list, list->max_size / FACTOR)) == NULL) { | ||||||
|  |             return NULL; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return data(list); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 = meta(list_ptr); | ||||||
|  |     list->size = 0; | ||||||
|  |     memset(list_ptr, 0, list->max_size * list->elem_size); | ||||||
|  |     return data(list_resize(list, list->init_size)); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void list_free(void *list_ptr) { | ||||||
|  |     free(meta(list_ptr)); | ||||||
|  | } | ||||||
							
								
								
									
										29
									
								
								src/lib/list.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								src/lib/list.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,29 @@ | |||||||
|  |  | ||||||
|  | #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); | ||||||
|  |  | ||||||
|  | int list_contains(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 | ||||||
							
								
								
									
										166
									
								
								src/lib/mpmc.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										166
									
								
								src/lib/mpmc.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,166 @@ | |||||||
|  |  | ||||||
|  | #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 | ||||||
|  |     while (sem_wait(&ctx->free) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } else { | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // lock wr field | ||||||
|  |     while (sem_wait(&ctx->lck_wr) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } 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 | ||||||
							
								
								
									
										579
									
								
								src/lib/proxy.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										579
									
								
								src/lib/proxy.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,579 @@ | |||||||
|  | /** | ||||||
|  |  * 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 "config.h" | ||||||
|  | #include "error.h" | ||||||
|  |  | ||||||
|  | #include <openssl/ssl.h> | ||||||
|  | #include <string.h> | ||||||
|  | #include <errno.h> | ||||||
|  | #include <openssl/err.h> | ||||||
|  | #include <arpa/inet.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); | ||||||
|  |             logger_set_prefix(""); | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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++; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     while (sem_wait(&available[n]) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } else { | ||||||
|  |             return NULL; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     while (sem_wait(&lock) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } 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; | ||||||
|  |     ctx->client = NULL; | ||||||
|  |     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); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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 ? "]\"" : "", | ||||||
|  |                      http_get_header_field(&req->hdr, "Host"), sock->enc ? "https" : "http"); | ||||||
|  |     if (p_len < 0 || p_len >= sizeof(buf1)) { | ||||||
|  |         error("Appended part of header field 'Forwarded' too long"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     const char *forwarded = http_get_header_field(&req->hdr, "Forwarded"); | ||||||
|  |     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"); | ||||||
|  |     forwarded = http_get_header_field(&req->hdr, "Forwarded"); | ||||||
|  |     if (xfh == NULL) { | ||||||
|  |         if (forwarded == NULL) { | ||||||
|  |             http_add_header_field(&req->hdr, "X-Forwarded-Host", http_get_header_field(&req->hdr, "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"); | ||||||
|  |     forwarded = http_get_header_field(&req->hdr, "Forwarded"); | ||||||
|  |     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}; | ||||||
|  |         int found = 0; | ||||||
|  |  | ||||||
|  |         for (int i = 0; i < sizeof(hostnames) / sizeof(hostnames[0]); i++) { | ||||||
|  |             char *hostname = hostnames[i]; | ||||||
|  |             found = 1; | ||||||
|  |  | ||||||
|  |             p_len = snprintf(buf1, sizeof(buf1), "http://%s/", hostname); | ||||||
|  |             if (strstarts(location, buf1)) break; | ||||||
|  |  | ||||||
|  |             p_len = snprintf(buf1, sizeof(buf1), "https://%s/", hostname); | ||||||
|  |             if (strstarts(location, buf1)) break; | ||||||
|  |  | ||||||
|  |             p_len = snprintf(buf1, sizeof(buf1), "http://%s:%i/", hostname, conf->proxy.port); | ||||||
|  |             if (strstarts(location, buf1)) break; | ||||||
|  |  | ||||||
|  |             p_len = snprintf(buf1, sizeof(buf1), "https://%s:%i/", hostname, conf->proxy.port); | ||||||
|  |             if (strstarts(location, buf1)) break; | ||||||
|  |  | ||||||
|  |             found = 0; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if (found) { | ||||||
|  |             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; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int proxy_connect(proxy_ctx_t *proxy, host_config_t *conf, http_res *res, http_status_ctx *ctx, char *err_msg) { | ||||||
|  |     char err_buf[256], addr_buf[1024]; | ||||||
|  |  | ||||||
|  |     info(BLUE_STR "Connecting to " BLD_STR "[%s]:%i" CLR_STR BLUE_STR "...", conf->proxy.hostname, conf->proxy.port); | ||||||
|  |  | ||||||
|  |     int fd; | ||||||
|  |     if ((fd = sock_connect(conf->proxy.hostname, conf->proxy.port, SERVER_TIMEOUT_INIT, addr_buf, sizeof(addr_buf))) == -1) { | ||||||
|  |         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", addr_buf, conf->proxy.port); | ||||||
|  |         sprintf(err_msg, "Unable to connect to server: %s.", error_str(errno, err_buf, sizeof(err_buf))); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     sock_init(&proxy->proxy, fd, 0); | ||||||
|  |  | ||||||
|  |     if (sock_set_timeout(&proxy->proxy, SERVER_TIMEOUT) != 0) { | ||||||
|  |         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))); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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); | ||||||
|  |  | ||||||
|  |         int ret; | ||||||
|  |         if ((ret = SSL_do_handshake(proxy->proxy.ssl)) != 1) { | ||||||
|  |             sock_error(&proxy->proxy, (int) ret); | ||||||
|  |             SSL_free(proxy->proxy.ssl); | ||||||
|  |             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))); | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |         proxy->proxy.enc = 1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     proxy->initialized = 1; | ||||||
|  |     proxy->cnx_s = clock_micros(); | ||||||
|  |     proxy->host = conf->name; | ||||||
|  |  | ||||||
|  |     info(BLUE_STR "Established new connection with " BLD_STR "[%s]:%i", addr_buf, conf->proxy.port); | ||||||
|  |  | ||||||
|  |     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]; | ||||||
|  |     long ret; | ||||||
|  |     int tries = 0, retry = 1; | ||||||
|  |  | ||||||
|  |     *proxy_ptr = proxy_get_by_conf(conf); | ||||||
|  |     proxy_ctx_t *proxy = *proxy_ptr; | ||||||
|  |     proxy->client = NULL; | ||||||
|  |  | ||||||
|  |     while (retry) { | ||||||
|  |         errno = 0; | ||||||
|  |  | ||||||
|  |         if (!proxy->initialized || sock_has_pending(&proxy->proxy) != 0) { | ||||||
|  |             if (proxy->initialized) | ||||||
|  |                 proxy_close(proxy); | ||||||
|  |  | ||||||
|  |             retry = 0; | ||||||
|  |             tries++; | ||||||
|  |  | ||||||
|  |             if (proxy_connect(proxy, conf, res, ctx, err_msg) != 0) | ||||||
|  |                 continue; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         const char *connection = http_get_header_field(&req->hdr, "Connection"); | ||||||
|  |         if (strcontains(connection, "upgrade") || strcontains(connection, "Upgrade")) { | ||||||
|  |             const char *upgrade = http_get_header_field(&req->hdr, "Upgrade"); | ||||||
|  |             const char *ws_version = http_get_header_field(&req->hdr, "Sec-WebSocket-Version"); | ||||||
|  |             if (streq(upgrade, "websocket") && streq(ws_version, "13")) { | ||||||
|  |                 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; | ||||||
|  |             continue; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         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 (strcontains(transfer_encoding, "chunked")) { | ||||||
|  |             ret = sock_splice_chunked(&proxy->proxy, client, buffer, sizeof(buffer), SOCK_CHUNKED); | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if (ret < 0 || (content_len != 0 && ret != content_len)) { | ||||||
|  |             if (ret == -1 && errno != EPROTO) { | ||||||
|  |                 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; | ||||||
|  |                 continue; | ||||||
|  |             } else if (ret == -1) { | ||||||
|  |                 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) - 1, 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; | ||||||
|  |             continue; | ||||||
|  |         } | ||||||
|  |         buffer[ret] = 0; | ||||||
|  |  | ||||||
|  |         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."); | ||||||
|  |             continue; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         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."); | ||||||
|  |                 continue; | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         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."); | ||||||
|  |                 continue; | ||||||
|  |             } | ||||||
|  |             if (ptr == buf) { | ||||||
|  |                 if (!strstarts(ptr, "HTTP/")) { | ||||||
|  |                     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."); | ||||||
|  |                     continue; | ||||||
|  |                 } | ||||||
|  |                 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."); | ||||||
|  |                     continue; | ||||||
|  |                 } | ||||||
|  |             } 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."); | ||||||
|  |                     continue; | ||||||
|  |                 } | ||||||
|  |             } | ||||||
|  |             if (pos0[2] == '\r' && pos0[3] == '\n') { | ||||||
|  |                 break; | ||||||
|  |             } | ||||||
|  |             ptr = pos0 + 2; | ||||||
|  |         } | ||||||
|  |         sock_recv_x(&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; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return -1; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int proxy_send(proxy_ctx_t *proxy, sock *client, unsigned long len_to_send, int flags) { | ||||||
|  |     char buffer[CHUNK_SIZE]; | ||||||
|  |     if (sock_splice(client, &proxy->proxy, buffer, sizeof(buffer), len_to_send) == -1) | ||||||
|  |         return -1; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int proxy_dump(proxy_ctx_t *proxy, char *buf, long len) { | ||||||
|  |     long ret = sock_recv(&proxy->proxy, buf, len, 0); | ||||||
|  |     if (ret == -1) return -1; | ||||||
|  |     buf[ret] = 0; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void proxy_close(proxy_ctx_t *ctx) { | ||||||
|  |     client_ctx_t *cctx = ctx->client; | ||||||
|  |     if (cctx) { | ||||||
|  |         logger_set_prefix("[%s%*s%s]%s", BLD_STR, ADDRSTRLEN, cctx->req_host, CLR_STR, cctx->log_prefix); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (ctx->initialized) { | ||||||
|  |         ctx->cnx_e = clock_micros(); | ||||||
|  |         char buf[32]; | ||||||
|  |         info(BLUE_STR "Closing proxy connection (%s)", format_duration(ctx->cnx_e - ctx->cnx_s, buf)); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     sock_close(&ctx->proxy); | ||||||
|  |  | ||||||
|  |     memset(ctx, 0, sizeof(*ctx)); | ||||||
|  |     errno = 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 | ||||||
|  |  | ||||||
|  | #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; | ||||||
|  |     long cnx_s, cnx_e; | ||||||
|  |     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); | ||||||
|  |  | ||||||
|  | void proxy_close(proxy_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | #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 | ||||||
							
								
								
									
										477
									
								
								src/lib/sock.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										477
									
								
								src/lib/sock.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,477 @@ | |||||||
|  | /** | ||||||
|  |  * 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 "../logger.h" | ||||||
|  |  | ||||||
|  | #include <errno.h> | ||||||
|  | #include <openssl/ssl.h> | ||||||
|  | #include <string.h> | ||||||
|  | #include <sys/socket.h> | ||||||
|  | #include <unistd.h> | ||||||
|  | #include <openssl/err.h> | ||||||
|  | #include <sys/ioctl.h> | ||||||
|  | #include <fcntl.h> | ||||||
|  | #include <netdb.h> | ||||||
|  |  | ||||||
|  | static void sock_ssl_error(unsigned long err) { | ||||||
|  |     if (err == SSL_ERROR_NONE) { | ||||||
|  |         errno = 0; | ||||||
|  |     } else if (err == SSL_ERROR_SYSCALL) { | ||||||
|  |         // errno already set | ||||||
|  |     } else if (err == SSL_ERROR_SSL) { | ||||||
|  |         error_ssl_err(ERR_get_error()); | ||||||
|  |     } else { | ||||||
|  |         error_ssl(err); | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void sock_error(sock *s, int ret) { | ||||||
|  |     sock_ssl_error(SSL_get_error(s->ssl, ret)); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_gai_error(int ret) { | ||||||
|  |     if (ret == 0) { | ||||||
|  |         errno = 0; | ||||||
|  |     } else if (ret == EAI_SYSTEM) { | ||||||
|  |         // errno already set | ||||||
|  |     } else { | ||||||
|  |         error_gai(ret); | ||||||
|  |     } | ||||||
|  |     return -1; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | const char *sock_error_str(unsigned long 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"; | ||||||
|  |         default: | ||||||
|  |             return "unknown error"; | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_init(sock *s, int fd, int flags) { | ||||||
|  |     if ((flags & SOCK_ENCRYPTED) && (flags & SOCK_PIPE)) { | ||||||
|  |         errno = EINVAL; | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     s->socket = fd; | ||||||
|  |     s->enc = !!(flags & SOCK_ENCRYPTED); | ||||||
|  |     s->pipe = !!(flags & SOCK_PIPE); | ||||||
|  |     s->ts_start = clock_micros(); | ||||||
|  |     s->ts_last = s->ts_start; | ||||||
|  |     s->timeout_us = -1; | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_connect(const char *hostname, unsigned short port, double timeout_sec, char *addr_buf, size_t addr_buf_size) { | ||||||
|  |     char buf[INET6_ADDRSTRLEN + 1]; | ||||||
|  |     int ret, fd, e = 0; | ||||||
|  |     long timeout_micros = (long) (timeout_sec * 1000000L); | ||||||
|  |     struct addrinfo *result, *rp, | ||||||
|  |             hints = { | ||||||
|  |                 .ai_family = AF_UNSPEC, | ||||||
|  |                 .ai_socktype = SOCK_STREAM, | ||||||
|  |                 .ai_protocol = 0, | ||||||
|  |                 .ai_flags = 0, | ||||||
|  |             }; | ||||||
|  |  | ||||||
|  |     if (addr_buf && addr_buf_size > 1) | ||||||
|  |         addr_buf[0] = 0; | ||||||
|  |  | ||||||
|  |     if ((ret = getaddrinfo(hostname, NULL, &hints, &result)) != 0) | ||||||
|  |         return sock_gai_error(ret); | ||||||
|  |  | ||||||
|  |     for (rp = result; rp != NULL; rp = rp->ai_next) { | ||||||
|  |         switch (rp->ai_family) { | ||||||
|  |             case AF_INET: | ||||||
|  |                 ((struct sockaddr_in *) rp->ai_addr)->sin_port = htons(port); | ||||||
|  |                 inet_ntop(rp->ai_family, &((struct sockaddr_in *) rp->ai_addr)->sin_addr, buf, addr_buf_size); | ||||||
|  |                 break; | ||||||
|  |             case AF_INET6: | ||||||
|  |                 ((struct sockaddr_in6 *) rp->ai_addr)->sin6_port = htons(port); | ||||||
|  |                 inet_ntop(rp->ai_family, &((struct sockaddr_in6 *) rp->ai_addr)->sin6_addr, buf, addr_buf_size); | ||||||
|  |                 break; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         debug("Trying [%s]:%i", buf, port); | ||||||
|  |  | ||||||
|  |         if ((fd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol)) == -1) { | ||||||
|  |             if (e == 0) { | ||||||
|  |                 e = errno; | ||||||
|  |             } else if (e != errno) { | ||||||
|  |                 e = -1; | ||||||
|  |             } | ||||||
|  |             continue; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if (sock_set_socket_timeout_micros(fd, timeout_micros, timeout_micros) == -1) { | ||||||
|  |             close(fd); | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if (connect(fd, rp->ai_addr, rp->ai_addrlen) == -1) { | ||||||
|  |             e = errno; | ||||||
|  |             close(fd); | ||||||
|  |             continue; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         break; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     freeaddrinfo(result); | ||||||
|  |  | ||||||
|  |     if (addr_buf && addr_buf_size > 1 && addr_buf[0] == 0) | ||||||
|  |         strncpy(addr_buf, buf, addr_buf_size); | ||||||
|  |  | ||||||
|  |     errno = e; | ||||||
|  |     return (e == 0) ? fd : -1; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_reverse_lookup(const sock *s, char *host, size_t host_size) { | ||||||
|  |     memset(host, 0, host_size); | ||||||
|  |  | ||||||
|  |     int ret; | ||||||
|  |     if ((ret = getnameinfo(&s->_addr.sock, sizeof(s->_addr), host, host_size, NULL, 0, 0)) != 0) { | ||||||
|  |         if (ret == EAI_NONAME) { | ||||||
|  |             return 0; | ||||||
|  |         } else { | ||||||
|  |             return sock_gai_error(ret); | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_set_socket_timeout_micros(int fd, 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(fd, SOL_SOCKET, SO_RCVTIMEO, &recv_to, sizeof(recv_to)) != 0) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     if (setsockopt(fd, 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->socket, (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) { | ||||||
|  |     if (s->socket == 0) { | ||||||
|  |         errno = ENOTCONN; | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     long ret; | ||||||
|  |     if (s->enc) { | ||||||
|  |         ret = SSL_write(s->ssl, buf, (int) len); | ||||||
|  |         if (ret <= 0) sock_error(s, (int) ret); | ||||||
|  |     } else if (s->pipe) { | ||||||
|  |         if (flags & ~MSG_MORE) { | ||||||
|  |             errno = EINVAL; | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |         ret = write(s->socket, buf, len); | ||||||
|  |     } else { | ||||||
|  |         ret = send(s->socket, buf, len, flags); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (ret >= 0) { | ||||||
|  |         s->ts_last = clock_micros(); | ||||||
|  |         return ret; | ||||||
|  |     } else { | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long sock_send_x(sock *s, void *buf, unsigned long len, int flags) { | ||||||
|  |     for (long ret, sent = 0; sent < len; sent += ret) { | ||||||
|  |         if ((ret = sock_send(s, (unsigned char *) buf + sent, len - sent, flags)) <= 0) { | ||||||
|  |             if (errno == EINTR) { | ||||||
|  |                 errno = 0, ret = 0; | ||||||
|  |                 continue; | ||||||
|  |             } else { | ||||||
|  |                 return -1; | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     return (long) len; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long sock_recv(sock *s, void *buf, unsigned long len, int flags) { | ||||||
|  |     if (s->socket == 0) { | ||||||
|  |         errno = ENOTCONN; | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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) sock_error(s, (int) ret); | ||||||
|  |     } else  if (s->pipe) { | ||||||
|  |         if (flags & ~MSG_WAITALL) { | ||||||
|  |             errno = EINVAL; | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |         ret = read(s->socket, buf, len); | ||||||
|  |     } else { | ||||||
|  |         ret = recv(s->socket, buf, len, flags); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (ret >= 0) { | ||||||
|  |         s->ts_last = clock_micros(); | ||||||
|  |         return ret; | ||||||
|  |     } else { | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long sock_recv_x(sock *s, void *buf, unsigned long len, int flags) { | ||||||
|  |     for (long ret, rcv = 0; rcv < len; rcv += ret) { | ||||||
|  |         if ((ret = sock_recv(s, (unsigned char *) buf + rcv, len - rcv, flags | MSG_WAITALL)) <= 0) { | ||||||
|  |             if (errno == EINTR) { | ||||||
|  |                 errno = 0, ret = 0; | ||||||
|  |                 continue; | ||||||
|  |             } else { | ||||||
|  |                 return -1; | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     return (long) len; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long sock_splice(sock *dst, sock *src, void *buf, unsigned long buf_len, unsigned long len) { | ||||||
|  |     long send_len = 0; | ||||||
|  |  | ||||||
|  |     if ((src->pipe || dst->pipe) && !src->enc && !dst->enc) { | ||||||
|  |         for (long ret; send_len < len; send_len += ret) { | ||||||
|  |             if ((ret = splice(src->socket, 0, dst->socket, 0, len, 0)) == -1) { | ||||||
|  |                 if (errno == EINTR) { | ||||||
|  |                     errno = 0, ret = 0; | ||||||
|  |                     continue; | ||||||
|  |                 } else { | ||||||
|  |                     return -1; | ||||||
|  |                 } | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } else { | ||||||
|  |         for (long ret, next_len; send_len < len; send_len += ret) { | ||||||
|  |             next_len = (long) ((buf_len < (len - send_len)) ? buf_len : (len - send_len)); | ||||||
|  |  | ||||||
|  |             if ((ret = sock_recv(src, buf, next_len, MSG_WAITALL)) <= 0) { | ||||||
|  |                 if (errno == EINTR) { | ||||||
|  |                     errno = 0, ret = 0; | ||||||
|  |                     continue; | ||||||
|  |                 } else { | ||||||
|  |                     return -1; | ||||||
|  |                 } | ||||||
|  |             } | ||||||
|  |  | ||||||
|  |             if (sock_send_x(dst, buf, ret, send_len + ret < len ? MSG_MORE : 0) == -1) | ||||||
|  |                 return -1; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return send_len; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long sock_splice_all(sock *dst, sock *src, void *buf, unsigned long buf_len) { | ||||||
|  |     long send_len = 0; | ||||||
|  |     for (long ret;; send_len += ret) { | ||||||
|  |         if ((ret = sock_recv(src, buf, buf_len, 0)) <= 0) { | ||||||
|  |             if (errno == EINTR) { | ||||||
|  |                 errno = 0, ret = 0; | ||||||
|  |                 continue; | ||||||
|  |             } else if (ret == 0) { | ||||||
|  |                 break; | ||||||
|  |             } else { | ||||||
|  |                 return -1; | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if (sock_send_x(dst, buf, ret, 0) == -1) | ||||||
|  |             return -1; | ||||||
|  |     } | ||||||
|  |     return send_len; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long sock_splice_chunked(sock *dst, sock *src, void *buf, unsigned long buf_len, int flags) { | ||||||
|  |     long ret; | ||||||
|  |     unsigned long send_len = 0, next_len; | ||||||
|  |  | ||||||
|  |     do { | ||||||
|  |         if ((ret = sock_recv_chunk_header(src)) == -1) | ||||||
|  |             return -1; | ||||||
|  |  | ||||||
|  |         next_len = ret; | ||||||
|  |  | ||||||
|  |         if (flags & SOCK_CHUNKED) | ||||||
|  |             if (sock_send_chunk_header(dst, next_len) == -1) | ||||||
|  |                 return -1; | ||||||
|  |  | ||||||
|  |         if ((ret = sock_splice(dst, src, buf, buf_len, next_len)) == -1) | ||||||
|  |             return ret; | ||||||
|  |  | ||||||
|  |         send_len += ret; | ||||||
|  |  | ||||||
|  |         if (sock_recv_chunk_trailer(src) == -1) | ||||||
|  |             return -1; | ||||||
|  |  | ||||||
|  |         if (flags & SOCK_CHUNKED) | ||||||
|  |             if (sock_send_chunk_trailer(dst) == -1) | ||||||
|  |                 return -1; | ||||||
|  |     } while (!(flags & SOCK_SINGLE_CHUNK) && next_len != 0); | ||||||
|  |  | ||||||
|  |     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, s->pipe = 0; | ||||||
|  |     errno = e; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_has_pending(sock *s) { | ||||||
|  |     int e = errno; | ||||||
|  |     long ret; | ||||||
|  |     if (s->pipe) { | ||||||
|  |         ioctl(s->socket, FIONREAD, &ret); | ||||||
|  |     } else { | ||||||
|  |         char buf[1]; | ||||||
|  |         ret = sock_recv(s, &buf, sizeof(buf), MSG_PEEK | MSG_DONTWAIT); | ||||||
|  |     } | ||||||
|  |     errno = e; | ||||||
|  |     return ret > 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long sock_recv_chunk_header(sock *s) { | ||||||
|  |     if (s->pipe) { | ||||||
|  |         uint64_t len; | ||||||
|  |         if (sock_recv_x(s, &len, sizeof(len), 0) == -1) | ||||||
|  |             return -1; | ||||||
|  |         return (long) len; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     long ret; | ||||||
|  |     size_t len = 0; | ||||||
|  |     char buf[20]; | ||||||
|  |  | ||||||
|  |     do { | ||||||
|  |         if ((ret = sock_recv(s, buf, sizeof(buf) - 1, MSG_PEEK)) <= 0) { | ||||||
|  |             if (errno == EINTR) { | ||||||
|  |                 errno = 0; | ||||||
|  |                 continue; | ||||||
|  |             } else { | ||||||
|  |                 return -1; | ||||||
|  |             } | ||||||
|  |         } else if (ret < 2) { | ||||||
|  |             continue; | ||||||
|  |         } | ||||||
|  |         buf[ret] = 0; | ||||||
|  |  | ||||||
|  |         if ((ret = parse_chunk_header(buf, ret, &len)) == -1 && errno == EPROTO) | ||||||
|  |             return -1; | ||||||
|  |     } while (ret < 0); | ||||||
|  |  | ||||||
|  |     if (sock_recv_x(s, buf, len, 0) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     return ret; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_send_chunk_header(sock *s, unsigned long size) { | ||||||
|  |     if (s->pipe) { | ||||||
|  |         uint64_t len = size; | ||||||
|  |         if (sock_send_x(s, &len, sizeof(len), 0) == -1) | ||||||
|  |             return -1; | ||||||
|  |     } else { | ||||||
|  |         char buf[20]; | ||||||
|  |         if (sock_send_x(s, buf, sprintf(buf, "%lX\r\n", size), 0) == -1) | ||||||
|  |             return -1; | ||||||
|  |     } | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_recv_chunk_trailer(sock *s) { | ||||||
|  |     if (s->pipe) return 0; | ||||||
|  |  | ||||||
|  |     char buf[2]; | ||||||
|  |     if (sock_recv_x(s, buf, sizeof(buf), MSG_PEEK) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     if (buf[0] != '\r' || buf[1] != '\n') { | ||||||
|  |         errno = EPROTO; | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (sock_recv_x(s, buf, sizeof(buf), 0) == -1) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_send_chunk_trailer(sock *s) { | ||||||
|  |     if (s->pipe) return 0; | ||||||
|  |     if (sock_send_x(s, "\r\n", 2, 0) == -1) | ||||||
|  |         return -1; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int sock_send_last_chunk(sock *s) { | ||||||
|  |     if (s->pipe) return sock_send_chunk_header(s, 0); | ||||||
|  |     if (sock_send_x(s, "0\r\n\r\n", 5, 0) == -1) | ||||||
|  |         return -1; | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
							
								
								
									
										83
									
								
								src/lib/sock.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										83
									
								
								src/lib/sock.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,83 @@ | |||||||
|  | /** | ||||||
|  |  * 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> | ||||||
|  |  | ||||||
|  | #define SOCK_CHUNKED 1 | ||||||
|  | #define SOCK_SINGLE_CHUNK 2 | ||||||
|  |  | ||||||
|  | #define SOCK_ENCRYPTED 1 | ||||||
|  | #define SOCK_PIPE 2 | ||||||
|  |  | ||||||
|  | typedef struct { | ||||||
|  |     unsigned int enc:1, pipe: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; | ||||||
|  |  | ||||||
|  | void sock_error(sock *s, int ret); | ||||||
|  |  | ||||||
|  | const char *sock_error_str(unsigned long err); | ||||||
|  |  | ||||||
|  | int sock_init(sock *s, int fd, int enc); | ||||||
|  |  | ||||||
|  | int sock_connect(const char *hostname, unsigned short port, double timeout_sec, char *addr_buf, size_t addr_buf_size); | ||||||
|  |  | ||||||
|  | int sock_reverse_lookup(const sock *s, char *host, size_t host_size); | ||||||
|  |  | ||||||
|  | int sock_init_addr_str(const sock *s, char *c_addr, size_t c_addr_size, char *s_addr, size_t s_addr_size); | ||||||
|  |  | ||||||
|  | int sock_set_socket_timeout_micros(int fd, 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_send_x(sock *s, void *buf, unsigned long len, int flags); | ||||||
|  |  | ||||||
|  | long sock_recv(sock *s, void *buf, unsigned long len, int flags); | ||||||
|  |  | ||||||
|  | long sock_recv_x(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_all(sock *dst, sock *src, void *buf, unsigned long buf_len); | ||||||
|  |  | ||||||
|  | long sock_splice_chunked(sock *dst, sock *src, void *buf, unsigned long buf_len, int flags); | ||||||
|  |  | ||||||
|  | int sock_close(sock *s); | ||||||
|  |  | ||||||
|  | int sock_has_pending(sock *s); | ||||||
|  |  | ||||||
|  | long sock_recv_chunk_header(sock *s); | ||||||
|  |  | ||||||
|  | int sock_send_chunk_header(sock *s, unsigned long size); | ||||||
|  |  | ||||||
|  | int sock_recv_chunk_trailer(sock *s); | ||||||
|  |  | ||||||
|  | int sock_send_chunk_trailer(sock *s); | ||||||
|  |  | ||||||
|  | int sock_send_last_chunk(sock *s); | ||||||
|  |  | ||||||
|  | #endif //SESIMOS_SOCK_H | ||||||
							
								
								
									
										207
									
								
								src/lib/uri.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										207
									
								
								src/lib/uri.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,207 @@ | |||||||
|  | /** | ||||||
|  |  * 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) { | ||||||
|  |     int e = errno; | ||||||
|  |     struct stat stat_buf; | ||||||
|  |     int ret = stat(path, &stat_buf); | ||||||
|  |     errno = e; | ||||||
|  |     return ret == 0 && S_ISDIR(stat_buf.st_mode) != 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int path_is_file(const char *path) { | ||||||
|  |     int e = errno; | ||||||
|  |     struct stat stat_buf; | ||||||
|  |     int ret = stat(path, &stat_buf); | ||||||
|  |     errno = e; | ||||||
|  |     return ret == 0 && S_ISDIR(stat_buf.st_mode) == 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int path_exists(const char *path) { | ||||||
|  |     int e = errno; | ||||||
|  |     struct stat stat_buf; | ||||||
|  |     int ret = stat(path, &stat_buf); | ||||||
|  |     errno = e; | ||||||
|  |     return ret == 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int uri_init(http_uri *uri, const char *webroot, const char *uri_str, int dir_mode) { | ||||||
|  |     char buf0[1024], buf1[1024], buf2[1024], buf3[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 (strcontains(uri->req_path, "/../") || strcontains(uri->req_path, "/./")) | ||||||
|  |         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.php", buf0); | ||||||
|  |         if (p_len < 0 || p_len >= sizeof(buf1)) return -1; | ||||||
|  |         p_len = snprintf(buf2, sizeof(buf2), "%s.html", buf0); | ||||||
|  |         if (p_len < 0 || p_len >= sizeof(buf2)) return -1; | ||||||
|  |  | ||||||
|  |         if (strlen(uri->path) <= 1 || path_exists(buf0) || path_is_file(buf1) || path_is_file(buf2)) | ||||||
|  |             break; | ||||||
|  |  | ||||||
|  |         char *ptr; | ||||||
|  |         parent_dir: | ||||||
|  |         ptr = strrchr(uri->path, '/'); | ||||||
|  |         size = (long) strlen(ptr); | ||||||
|  |         sprintf(buf3, "%.*s%s", (int) size, ptr, uri->pathinfo); | ||||||
|  |         strcpy(uri->pathinfo, buf3); | ||||||
|  |         ptr[0] = 0; | ||||||
|  |     } | ||||||
|  |     if (uri->pathinfo[0] != 0) { | ||||||
|  |         sprintf(buf3, "%s", uri->pathinfo + 1); | ||||||
|  |         strcpy(uri->pathinfo, buf3); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (path_is_file(buf0)) { | ||||||
|  |         uri->filename = malloc(strlen(buf0) + 1); | ||||||
|  |         strcpy(uri->filename, buf0); | ||||||
|  |         long len = (long) strlen(uri->path); | ||||||
|  |         if (strends(uri->path, ".php")) { | ||||||
|  |             uri->path[len - 4] = 0; | ||||||
|  |             uri->is_static = 0; | ||||||
|  |         } else if (strends(uri->path, ".html")) { | ||||||
|  |             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 { | ||||||
|  |         uri->is_dir = 1; | ||||||
|  |         strcpy(uri->path + strlen(uri->path), "/"); | ||||||
|  |         sprintf(buf1, "%s%s" "index.php", uri->webroot, uri->path); | ||||||
|  |         sprintf(buf2, "%s%s" "index.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); | ||||||
|  |         } 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 (strends(uri->path + strlen(uri->path), "index")) | ||||||
|  |         uri->path[strlen(uri->path) - 5] = 0; | ||||||
|  |  | ||||||
|  |     if (streq(uri->pathinfo, "index.php") || streq(uri->pathinfo, "index.html")) | ||||||
|  |         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; | ||||||
|  | } | ||||||
							
								
								
									
										51
									
								
								src/lib/uri.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										51
									
								
								src/lib/uri.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,51 @@ | |||||||
|  | /** | ||||||
|  |  * 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]; | ||||||
|  |     long mtime; | ||||||
|  | } 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, 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 | ||||||
							
								
								
									
										400
									
								
								src/lib/utils.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										400
									
								
								src/lib/utils.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,400 @@ | |||||||
|  | /** | ||||||
|  |  * 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> | ||||||
|  | #include <sys/stat.h> | ||||||
|  | #include <errno.h> | ||||||
|  | #include <unistd.h> | ||||||
|  | #include <dirent.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 *restrict 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 | ||||||
|  |         mime_is_text(type) || | ||||||
|  |         streq(type_parsed, "application/vnd.ms-fontobject") || | ||||||
|  |         streq(type_parsed, "application/x-font-ttf") || | ||||||
|  |         streq(type_parsed, "font/eot") || | ||||||
|  |         streq(type_parsed, "font/opentype") || | ||||||
|  |         streq(type_parsed, "image/bmp") || | ||||||
|  |         streq(type_parsed, "image/gif") || | ||||||
|  |         streq(type_parsed, "image/vnd.microsoft.icon") || | ||||||
|  |         streq(type_parsed, "image/vnd.microsoft.iconbinary") || | ||||||
|  |         streq(type_parsed, "image/x-icon"); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int mime_is_text(const char *restrict 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 | ||||||
|  |         strstarts(type_parsed, "text/") || | ||||||
|  |         strstarts(type_parsed, "message/") || | ||||||
|  |         strends(type_parsed, "+xml") || | ||||||
|  |         strends(type_parsed, "+json") || | ||||||
|  |         streq(type_parsed, "application/javascript") || | ||||||
|  |         streq(type_parsed, "application/json") || | ||||||
|  |         streq(type_parsed, "application/xml") || | ||||||
|  |         streq(type_parsed, "application/x-www-form-urlencoded") || | ||||||
|  |         streq(type_parsed, "application/x-tex") || | ||||||
|  |         streq(type_parsed, "application/x-httpd-php") || | ||||||
|  |         streq(type_parsed, "application/x-latex") || | ||||||
|  |         streq(type_parsed, "application/x-javascript"); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int strcpy_rem_webroot(char *dst, const char *src, const char *webroot) { | ||||||
|  |     strcpy(dst, src); | ||||||
|  |     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 streq(const char *restrict str1, const char *restrict str2) { | ||||||
|  |     return str1 != NULL && str2 != NULL && strcmp(str1, str2) == 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int strcontains(const char *restrict haystack, const char *restrict needle) { | ||||||
|  |     return haystack != NULL && needle != NULL && strstr(haystack, needle) != NULL; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int strstarts(const char *restrict str, const char *restrict prefix) { | ||||||
|  |     if (str == NULL || prefix == NULL) return 0; | ||||||
|  |     unsigned long l1 = strlen(str), l2 = strlen(prefix); | ||||||
|  |     return l2 <= l1 && strncmp(str, prefix, l2) == 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int strends(const char *restrict str, const char *restrict suffix) { | ||||||
|  |     if (str == NULL || suffix == NULL) return 0; | ||||||
|  |     unsigned long l1 = strlen(str), l2 = strlen(suffix); | ||||||
|  |     return l2 <= l1 && strcmp(str + l1 - l2, suffix) == 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; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long stat_mtime(const char *filename) { | ||||||
|  |     struct stat stat_buf; | ||||||
|  |     stat(filename, &stat_buf); | ||||||
|  |     return stat_buf.st_mtime; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int rm_rf(const char *path) { | ||||||
|  |     struct stat stat_buf; | ||||||
|  |     if (lstat(path, &stat_buf) != 0) | ||||||
|  |         return (errno == ENOENT) ? 0 : -1; | ||||||
|  |  | ||||||
|  |     if (S_ISREG(stat_buf.st_mode)) { | ||||||
|  |         // regular file | ||||||
|  |         return unlink(path); | ||||||
|  |     } else if (S_ISLNK(stat_buf.st_mode)) { | ||||||
|  |         // link | ||||||
|  |         return unlink(path); | ||||||
|  |     } else if (S_ISDIR(stat_buf.st_mode)) { | ||||||
|  |         // directory | ||||||
|  |         char buf[FILENAME_MAX]; | ||||||
|  |         DIR *dir; | ||||||
|  |  | ||||||
|  |         // open directory | ||||||
|  |         if ((dir = opendir(path)) == NULL) | ||||||
|  |             return -1; | ||||||
|  |  | ||||||
|  |         // read directory | ||||||
|  |         for (struct dirent *ent; (ent = readdir(dir)) != NULL;) { | ||||||
|  |             if (streq(ent->d_name, ".") || streq(ent->d_name, "..")) | ||||||
|  |                 continue; | ||||||
|  |  | ||||||
|  |             snprintf(buf, sizeof(buf), "%s/%s", path, ent->d_name); | ||||||
|  |             if (rm_rf(buf) != 0) { | ||||||
|  |                 closedir(dir); | ||||||
|  |                 return -1; | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         // close and remove directory | ||||||
|  |         closedir(dir); | ||||||
|  |         return rmdir(path); | ||||||
|  |     } else { | ||||||
|  |         // other - not supported | ||||||
|  |         errno = ENOTSUP; | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long fsize(FILE *file) { | ||||||
|  |     long cur_pos, len; | ||||||
|  |     if ((cur_pos = ftell(file)) == -1) | ||||||
|  |         return -1; | ||||||
|  |     if (fseek(file, 0, SEEK_END) != 0) | ||||||
|  |         return -1; | ||||||
|  |     if ((len = ftell(file)) == -1) | ||||||
|  |         return -1; | ||||||
|  |     if (fseek(file, cur_pos, SEEK_SET) != 0) | ||||||
|  |         return -1; | ||||||
|  |     return len; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long flines(FILE *file) { | ||||||
|  |     long cur_pos, lines = 0; | ||||||
|  |     if ((cur_pos = ftell(file)) == -1) | ||||||
|  |         return -1; | ||||||
|  |     if (fseek(file, 0, SEEK_SET) != 0) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     for (int ch; (ch = fgetc(file)) != EOF;) { | ||||||
|  |         if (ch == '\n') lines++; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (fseek(file, cur_pos, SEEK_SET) != 0) | ||||||
|  |         return -1; | ||||||
|  |     return lines + 1; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long file_get_line_pos(FILE *file, long line_num) { | ||||||
|  |     if (line_num < 1) { | ||||||
|  |         errno = EINVAL; | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     long cur_pos; | ||||||
|  |     if ((cur_pos = ftell(file)) == -1) | ||||||
|  |         return -1; | ||||||
|  |     if (fseek(file, 0, SEEK_SET) != 0) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     long lines = 0, pos = 0; | ||||||
|  |     for (int ch; lines < line_num - 1 && (ch = fgetc(file)) != EOF; pos++) { | ||||||
|  |         if (ch == '\n') lines++; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (fseek(file, cur_pos, SEEK_SET) != 0) | ||||||
|  |         return -1; | ||||||
|  |     return pos; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fseekl(FILE *file, long line_num) { | ||||||
|  |     if (line_num < 1) { | ||||||
|  |         errno = EINVAL; | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (fseek(file, 0, SEEK_SET) != 0) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     long lines = 0; | ||||||
|  |     for (int ch; lines < line_num - 1 && (ch = fgetc(file)) != EOF;) { | ||||||
|  |         if (ch == '\n') lines++; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long ftelll(FILE *file) { | ||||||
|  |     long cur_pos; | ||||||
|  |     if ((cur_pos = ftell(file)) == -1) | ||||||
|  |         return -1; | ||||||
|  |     if (fseek(file, 0, SEEK_SET) != 0) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     long lines = 0, pos = 0; | ||||||
|  |     for (int ch; pos < cur_pos && (ch = fgetc(file)) != EOF; pos++) { | ||||||
|  |         if (ch == '\n') lines++; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (fseek(file, cur_pos, SEEK_SET) != 0) | ||||||
|  |         return -1; | ||||||
|  |     return lines + 1; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | long parse_chunk_header(const char *buf, size_t len, size_t *ret_len) { | ||||||
|  |     for (int i = 0; i < len; i++) { | ||||||
|  |         char ch = buf[i]; | ||||||
|  |         if (ch == '\r') { | ||||||
|  |             continue; | ||||||
|  |         } else if (ch == '\n' && i > 1 && buf[i - 1] == '\r') { | ||||||
|  |             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'))) { | ||||||
|  |             errno = EPROTO; | ||||||
|  |             return -1; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     return -1; | ||||||
|  | } | ||||||
							
								
								
									
										70
									
								
								src/lib/utils.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										70
									
								
								src/lib/utils.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,70 @@ | |||||||
|  | /** | ||||||
|  |  * 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 *restrict type); | ||||||
|  |  | ||||||
|  | int mime_is_text(const char *restrict type); | ||||||
|  |  | ||||||
|  | int strcpy_rem_webroot(char *dst, const char *str, const char *webroot); | ||||||
|  |  | ||||||
|  | int str_trim(char **start, char **end); | ||||||
|  |  | ||||||
|  | int str_trim_lws(char **start, char **end); | ||||||
|  |  | ||||||
|  | int streq(const char *restrict str1, const char *restrict str2); | ||||||
|  |  | ||||||
|  | int strcontains(const char *restrict haystack, const char *restrict needle); | ||||||
|  |  | ||||||
|  | int strstarts(const char *restrict str, const char *restrict prefix); | ||||||
|  |  | ||||||
|  | int strends(const char *restrict str, const char *restrict suffix); | ||||||
|  |  | ||||||
|  | int base64_encode(void *data, unsigned long data_len, char *output, unsigned long *output_len); | ||||||
|  |  | ||||||
|  | long clock_micros(void); | ||||||
|  |  | ||||||
|  | long clock_cpu(void); | ||||||
|  |  | ||||||
|  | long stat_mtime(const char *filename); | ||||||
|  |  | ||||||
|  | int rm_rf(const char *path); | ||||||
|  |  | ||||||
|  | long fsize(FILE *file); | ||||||
|  |  | ||||||
|  | long flines(FILE *file); | ||||||
|  |  | ||||||
|  | long file_get_line_pos(FILE *file, long line_num); | ||||||
|  |  | ||||||
|  | int fseekl(FILE *file, long line_num); | ||||||
|  |  | ||||||
|  | long ftelll(FILE *file); | ||||||
|  |  | ||||||
|  | long parse_chunk_header(const char *buf, size_t len, size_t *ret_len); | ||||||
|  |  | ||||||
|  | #endif //SESIMOS_UTILS_H | ||||||
							
								
								
									
										130
									
								
								src/lib/websocket.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										130
									
								
								src/lib/websocket.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,130 @@ | |||||||
|  | /** | ||||||
|  |  * 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]; | ||||||
|  |  | ||||||
|  |     if (sock_recv_x(s, buf, 2, 0) == -1) { | ||||||
|  |         error("Unable to receive from socket"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (sock_recv_x(s, buf, remaining, 0) == -1) { | ||||||
|  |         error("Unable to receive from socket"); | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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_x(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 | ||||||
							
								
								
									
										305
									
								
								src/logger.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										305
									
								
								src/logger.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,305 @@ | |||||||
|  | /** | ||||||
|  |  * 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][%-8s][%-6s]" | ||||||
|  | #define LOG_TIME_BUF_SIZE 9 | ||||||
|  | #define LOG_TIME_FMT "%H:%M:%S" | ||||||
|  |  | ||||||
|  | typedef struct { | ||||||
|  |     log_lvl_t lvl; | ||||||
|  |     long time; | ||||||
|  |     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 const char *timestr(time_t ts, char *buf) { | ||||||
|  |     struct tm time_info; | ||||||
|  |     strftime(buf, LOG_TIME_BUF_SIZE, LOG_TIME_FMT, localtime_r(&ts, &time_info)); | ||||||
|  |     return buf; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static void err(const char *restrict msg) { | ||||||
|  |     char err_buf[64], time_buf[LOG_TIME_BUF_SIZE]; | ||||||
|  |     fprintf(stderr, ERR_STR LOG_PREFIX " %s: %s" CLR_STR "\n", timestr(time(NULL), time_buf), "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], err_buf[256], time_buf[LOG_TIME_BUF_SIZE]; | ||||||
|  |     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, | ||||||
|  |                timestr(time(NULL), time_buf), | ||||||
|  |                (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 | ||||||
|  |         while (sem_wait(&sem_buf_free) != 0) { | ||||||
|  |             if (errno == EINTR) { | ||||||
|  |                 errno = 0; | ||||||
|  |                 continue; | ||||||
|  |             } else { | ||||||
|  |                 err("Unable to lock semaphore"); | ||||||
|  |                 errno = 0; | ||||||
|  |             } | ||||||
|  |             // cleanup | ||||||
|  |             va_end(args); | ||||||
|  |             return; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         // try to lock buffer | ||||||
|  |         while (sem_wait(&sem_buf) != 0) { | ||||||
|  |             if (errno == EINTR) { | ||||||
|  |                 errno = 0; | ||||||
|  |                 continue; | ||||||
|  |             } 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; | ||||||
|  |         msg->time = time(NULL); | ||||||
|  |  | ||||||
|  |         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; | ||||||
|  |  | ||||||
|  |     void *ptr; | ||||||
|  |     if (key_name == -1) { | ||||||
|  |         // not initialized | ||||||
|  |         va_start(args, format); | ||||||
|  |         vsnprintf(global_name, sizeof(global_name), format, args); | ||||||
|  |         ptr = global_name; | ||||||
|  |     } else { | ||||||
|  |         int ret; | ||||||
|  |         if ((ptr = pthread_getspecific(key_name)) == NULL) { | ||||||
|  |             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); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // set thread name | ||||||
|  |     // warning: max length is 16 (incl. terminating null byte) | ||||||
|  |     pthread_setname_np(pthread_self(), ptr); | ||||||
|  |  | ||||||
|  |     // 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) { | ||||||
|  |     char time_buf[LOG_TIME_BUF_SIZE]; | ||||||
|  |  | ||||||
|  |     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 : ""), | ||||||
|  |                (timestr(msg->time, time_buf)), | ||||||
|  |                (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; | ||||||
|  |     pthread_kill(thread, SIGUSR1); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int logger_join(void) { | ||||||
|  |     return pthread_join(thread, NULL); | ||||||
|  | } | ||||||
							
								
								
									
										42
									
								
								src/logger.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										42
									
								
								src/logger.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,42 @@ | |||||||
|  | /** | ||||||
|  |  * 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); | ||||||
|  |  | ||||||
|  | int logger_join(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 |  | ||||||
							
								
								
									
										433
									
								
								src/server.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										433
									
								
								src/server.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,433 @@ | |||||||
|  | /** | ||||||
|  |  * 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> | ||||||
|  | #include <semaphore.h> | ||||||
|  |  | ||||||
|  | const char *config_file; | ||||||
|  | static int sockets[NUM_SOCKETS]; | ||||||
|  | static SSL_CTX *contexts[CONFIG_MAX_CERT_CONFIG]; | ||||||
|  | static client_ctx_t **clients; | ||||||
|  | static sem_t sem_clients_lock; | ||||||
|  |  | ||||||
|  | static void clean(void) { | ||||||
|  |     notice("Cleaning sesimos cache and metadata files..."); | ||||||
|  |  | ||||||
|  |     // remove legacy files | ||||||
|  |     //     /.../server/, /.../server/cache | ||||||
|  |     if (rm_rf("/var/sesimos/server") != 0) { | ||||||
|  |         error("Unable to remove /var/sesimos/server/"); | ||||||
|  |     } else if (!errno) { | ||||||
|  |         notice("Successfully removed /var/sesimos/server/"); | ||||||
|  |     } | ||||||
|  |     errno = 0; | ||||||
|  |  | ||||||
|  |     // remove cache and metadata files | ||||||
|  |     char buf[512]; | ||||||
|  |     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; | ||||||
|  |  | ||||||
|  |         snprintf(buf, sizeof(buf), "%s/.sesimos", hc->local.webroot); | ||||||
|  |         if (rm_rf(buf) != 0) { | ||||||
|  |             error("Unable to remove %s/", buf); | ||||||
|  |         } else if (!errno) { | ||||||
|  |             notice("Successfully removed %s/", buf); | ||||||
|  |         } | ||||||
|  |         errno = 0; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     notice("Cleaned all sesimos cache and metadata files!"); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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) { | ||||||
|  |     // try to lock clients list | ||||||
|  |     while (sem_wait(&sem_clients_lock) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } else { | ||||||
|  |             critical("Unable to lock clients list"); | ||||||
|  |             return; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // delete from list | ||||||
|  |     clients = list_delete(clients, &ctx); | ||||||
|  |     if (clients == NULL) { | ||||||
|  |         critical("Unable to delete context from list"); | ||||||
|  |         return; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // unlock clients list | ||||||
|  |     sem_post(&sem_clients_lock); | ||||||
|  |  | ||||||
|  |     // free data | ||||||
|  |     free(ctx); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static void ssl_free() { | ||||||
|  |     for (int i = 0; i < CONFIG_MAX_CERT_CONFIG; i++) { | ||||||
|  |         const cert_config_t *conf = &config.certs[i]; | ||||||
|  |         if (conf->name[0] == 0) break; | ||||||
|  |         SSL_CTX_free(contexts[i]); | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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; | ||||||
|  |     } | ||||||
|  |     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"); | ||||||
|  |         free(client_ctx); | ||||||
|  |         return; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     sock_init(client, client_fd, (i == 1) ? SOCK_ENCRYPTED : 0); | ||||||
|  |     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; | ||||||
|  |  | ||||||
|  |     // try to lock clients list | ||||||
|  |     while (sem_wait(&sem_clients_lock) != 0) { | ||||||
|  |         if (errno == EINTR) { | ||||||
|  |             errno = 0; | ||||||
|  |             continue; | ||||||
|  |         } else { | ||||||
|  |             critical("Unable to lock clients list"); | ||||||
|  |             return; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // append to list | ||||||
|  |     clients = list_append(clients, &client_ctx); | ||||||
|  |     if (clients == NULL) { | ||||||
|  |         critical("Unable to add client context to list"); | ||||||
|  |         free(client_ctx); | ||||||
|  |         errno = 0; | ||||||
|  |         return; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     // unlock clients list | ||||||
|  |     sem_post(&sem_clients_lock); | ||||||
|  |  | ||||||
|  |     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) { | ||||||
|  |     if (sig != 0) 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(); | ||||||
|  |  | ||||||
|  |     logger_set_prefix(""); | ||||||
|  |     proxy_close_all(); | ||||||
|  |  | ||||||
|  |     while (list_size(clients) > 0) | ||||||
|  |         tcp_close(clients[0]); | ||||||
|  |     logger_set_prefix(""); | ||||||
|  |  | ||||||
|  |     async_stop(); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static void nothing(int sig) {} | ||||||
|  |  | ||||||
|  | int main(int argc, char *const argv[]) { | ||||||
|  |     const int YES = 1; | ||||||
|  |     int ret; | ||||||
|  |     int mode = 0; | ||||||
|  |  | ||||||
|  |     memset(sockets, 0, sizeof(sockets)); | ||||||
|  |  | ||||||
|  |     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_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'}, | ||||||
|  |             {"clean",   no_argument,        0, 'C'}, | ||||||
|  |             {"config",  required_argument,  0, 'c'}, | ||||||
|  |             { 0,        0,                  0,  0 } | ||||||
|  |     }; | ||||||
|  |  | ||||||
|  |     config_file = NULL; | ||||||
|  |     for (int c, opt_idx; (c = getopt_long(argc, argv, "hCc:", 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" | ||||||
|  |                         "  -C, --clean                 clear cached files and other metadata\n" | ||||||
|  |                         "  -h, --help                  print this dialogue\n"); | ||||||
|  |                 return 0; | ||||||
|  |             case 'c': | ||||||
|  |                 config_file = optarg; | ||||||
|  |                 break; | ||||||
|  |             case 'C': | ||||||
|  |                 mode = 1; | ||||||
|  |                 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 (mode == 1) { | ||||||
|  |         clean(); | ||||||
|  |         return 0; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     clients = list_create(sizeof(client_ctx_t *), 64); | ||||||
|  |     if (clients == NULL) { | ||||||
|  |         critical("Unable to initialize client list"); | ||||||
|  |         ssl_free(); | ||||||
|  |         return 1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (sem_init(&sem_clients_lock, 0, 1) != 0) { | ||||||
|  |         critical("Unable to create clients lock semaphore"); | ||||||
|  |         ssl_free(); | ||||||
|  |         list_free(clients); | ||||||
|  |         return 1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (async_init() != 0) { | ||||||
|  |         critical("Unable to initialize async thread"); | ||||||
|  |         ssl_free(); | ||||||
|  |         geoip_free(); | ||||||
|  |         list_free(clients); | ||||||
|  |         sem_destroy(&sem_clients_lock); | ||||||
|  |         return 1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (proxy_preload() != 0) { | ||||||
|  |         critical("Unable to initialize proxy"); | ||||||
|  |         ssl_free(); | ||||||
|  |         geoip_free(); | ||||||
|  |         list_free(clients); | ||||||
|  |         sem_destroy(&sem_clients_lock); | ||||||
|  |         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); | ||||||
|  |             ssl_free(); | ||||||
|  |             geoip_free(); | ||||||
|  |             list_free(clients); | ||||||
|  |             sem_destroy(&sem_clients_lock); | ||||||
|  |             async_free(); | ||||||
|  |             proxy_unload(); | ||||||
|  |             return 1; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     logger_init(); | ||||||
|  |  | ||||||
|  |     if ((ret = cache_init()) != 0) { | ||||||
|  |         if (ret == -1) critical("Unable to initialize cache"); | ||||||
|  |         ssl_free(); | ||||||
|  |         geoip_free(); | ||||||
|  |         list_free(clients); | ||||||
|  |         sem_destroy(&sem_clients_lock); | ||||||
|  |         async_free(); | ||||||
|  |         proxy_unload(); | ||||||
|  |         logger_stop(); | ||||||
|  |         logger_join(); | ||||||
|  |         return 1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     logger_set_name("main"); | ||||||
|  |  | ||||||
|  |     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"); | ||||||
|  |  | ||||||
|  |     int error = 0; | ||||||
|  |     async_thread(); | ||||||
|  |     if (errno != 0) { | ||||||
|  |         errno = 0; | ||||||
|  |         error = 2; | ||||||
|  |         terminate_gracefully(0); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     notice("Goodbye!"); | ||||||
|  |  | ||||||
|  |     // cleanup | ||||||
|  |     ssl_free(); | ||||||
|  |     list_free(clients); | ||||||
|  |     sem_destroy(&sem_clients_lock); | ||||||
|  |     geoip_free(); | ||||||
|  |     proxy_unload(); | ||||||
|  |     cache_join(); | ||||||
|  |     async_free(); | ||||||
|  |     logger_stop(); | ||||||
|  |     logger_join(); | ||||||
|  |  | ||||||
|  |     return error; | ||||||
|  | } | ||||||
							
								
								
									
										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 | ||||||
							
								
								
									
										50
									
								
								src/worker/chunk_handler.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										50
									
								
								src/worker/chunk_handler.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,50 @@ | |||||||
|  | /** | ||||||
|  |  * sesimos - secure, simple, modern web server | ||||||
|  |  * @brief FastCGI frame handler | ||||||
|  |  * @file src/worker/fcti_frame_handler.c | ||||||
|  |  * @author Lorenz Stechauner | ||||||
|  |  * @date 2023-01-22 | ||||||
|  |  */ | ||||||
|  |  | ||||||
|  | #include "func.h" | ||||||
|  | #include "../logger.h" | ||||||
|  | #include "../workers.h" | ||||||
|  |  | ||||||
|  | #include <errno.h> | ||||||
|  |  | ||||||
|  | void chunk_handler_func(chunk_ctx_t *ctx) { | ||||||
|  |     logger_set_prefix("[%*s]%s", ADDRSTRLEN, ctx->client->socket.s_addr, ctx->client->log_prefix); | ||||||
|  |  | ||||||
|  |     char buf[CHUNK_SIZE]; | ||||||
|  |     long sent = sock_splice_chunked(&ctx->client->socket, ctx->socket, buf, sizeof(buf), ctx->flags | SOCK_SINGLE_CHUNK); | ||||||
|  |     if (sent < 0) { | ||||||
|  |         // error | ||||||
|  |         error("Unable to splice chunk"); | ||||||
|  |         errno = 0; | ||||||
|  |         ctx->err_cb(ctx); | ||||||
|  |     } else if (sent == 0) { | ||||||
|  |         // last chunk | ||||||
|  |         ctx->client->chunks_transferred = 1; | ||||||
|  |         ctx->next_cb(ctx); | ||||||
|  |     } else { | ||||||
|  |         // next chunk | ||||||
|  |         handle_chunk(ctx); | ||||||
|  |         return; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     free(ctx); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int handle_chunks(client_ctx_t *ctx, sock *socket, int flags, void (*next_cb)(chunk_ctx_t *), void (*err_cb)(chunk_ctx_t *)) { | ||||||
|  |     chunk_ctx_t *a = malloc(sizeof(chunk_ctx_t)); | ||||||
|  |  | ||||||
|  |     a->client = ctx; | ||||||
|  |     a->socket = socket; | ||||||
|  |     a->flags = flags; | ||||||
|  |     a->next_cb = (void (*)(void *)) next_cb; | ||||||
|  |     a->err_cb  = (void (*)(void *)) err_cb; | ||||||
|  |  | ||||||
|  |     handle_chunk(a); | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
							
								
								
									
										72
									
								
								src/worker/fastcgi_frame_handler.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										72
									
								
								src/worker/fastcgi_frame_handler.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,72 @@ | |||||||
|  | /** | ||||||
|  |  * sesimos - secure, simple, modern web server | ||||||
|  |  * @brief FastCGI frame handler | ||||||
|  |  * @file src/worker/fcti_frame_handler.c | ||||||
|  |  * @author Lorenz Stechauner | ||||||
|  |  * @date 2023-01-22 | ||||||
|  |  */ | ||||||
|  |  | ||||||
|  | #include "func.h" | ||||||
|  | #include "../lib/fastcgi.h" | ||||||
|  | #include "../logger.h" | ||||||
|  | #include "../workers.h" | ||||||
|  |  | ||||||
|  | #include <errno.h> | ||||||
|  | #include <memory.h> | ||||||
|  | #include <unistd.h> | ||||||
|  |  | ||||||
|  | void fastcgi_frame_handler_func(fastcgi_ctx_t *ctx) { | ||||||
|  |     logger_set_prefix("[%*s]%s", ADDRSTRLEN, ctx->client->socket.s_addr, ctx->client->log_prefix); | ||||||
|  |  | ||||||
|  |     switch (fastcgi_recv_frame(&ctx->cnx)) { | ||||||
|  |         case FCGI_STDOUT: | ||||||
|  |         case FCGI_STDERR: | ||||||
|  |             fastcgi_handle_frame(ctx); | ||||||
|  |             break; | ||||||
|  |         case -1: | ||||||
|  |             error("Unable to receive FastCGI frame"); | ||||||
|  |             ctx->client->s_keep_alive = 0; | ||||||
|  |             fastcgi_close(ctx); | ||||||
|  |             break; | ||||||
|  |         default: | ||||||
|  |             // end of request received | ||||||
|  |             write(ctx->cnx.fd_out, "\0\0\0\0\0\0\0\0\r\n", 10); | ||||||
|  |             fastcgi_close(ctx); | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_handle_connection(client_ctx_t *ctx, fastcgi_cnx_t **cnx) { | ||||||
|  |     sock_set_timeout(&(*cnx)->socket, FASTCGI_TIMEOUT); | ||||||
|  |     sock_set_socket_timeout(&(*cnx)->socket, 1); | ||||||
|  |  | ||||||
|  |     fastcgi_ctx_t *a = malloc(sizeof(fastcgi_ctx_t)); | ||||||
|  |     a->closed = 0; | ||||||
|  |     a->client = ctx; | ||||||
|  |     memcpy(&a->cnx, *cnx, sizeof(fastcgi_cnx_t)); | ||||||
|  |     ctx->fcgi_ctx = a; | ||||||
|  |     fastcgi_handle_frame(a); | ||||||
|  |     *cnx = &a->cnx; | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void fastcgi_close(fastcgi_ctx_t *ctx) { | ||||||
|  |     if (ctx->closed == 0) { | ||||||
|  |         ctx->closed++; | ||||||
|  |         return; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     logger_set_prefix("[%*s]%s", ADDRSTRLEN, ctx->client->socket.s_addr, ctx->client->log_prefix); | ||||||
|  |  | ||||||
|  |     fastcgi_php_error(&ctx->cnx, NULL); | ||||||
|  |  | ||||||
|  |     if (ctx->cnx.app_status != 0) | ||||||
|  |         error("FastCGI app terminated with exit code %i", ctx->cnx.app_status); | ||||||
|  |  | ||||||
|  |     debug("Closing FastCGI connection"); | ||||||
|  |  | ||||||
|  |     fastcgi_close_cnx(&ctx->cnx); | ||||||
|  |     ctx->client->fcgi_ctx = NULL; | ||||||
|  |     free(ctx); | ||||||
|  |     errno = 0; | ||||||
|  | } | ||||||
							
								
								
									
										189
									
								
								src/worker/fastcgi_handler.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										189
									
								
								src/worker/fastcgi_handler.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,189 @@ | |||||||
|  | /** | ||||||
|  |  * 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 "../workers.h" | ||||||
|  | #include "../lib/fastcgi.h" | ||||||
|  |  | ||||||
|  | #include <string.h> | ||||||
|  | #include <errno.h> | ||||||
|  | #include <unistd.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, ADDRSTRLEN, ctx->req_host, CLR_STR, ctx->log_prefix); | ||||||
|  |  | ||||||
|  |     if (!ctx->chunks_transferred) { | ||||||
|  |         fastcgi_cnx_t *fcgi_cnx = NULL; | ||||||
|  |         int ret = fastcgi_handler_1(ctx, &fcgi_cnx); | ||||||
|  |         respond(ctx); | ||||||
|  |         if (ret == 0) { | ||||||
|  |             switch (fastcgi_handler_2(ctx, fcgi_cnx)) { | ||||||
|  |                 case 1: return; | ||||||
|  |                 case 2: break; | ||||||
|  |             } | ||||||
|  |         } else if (ctx->fcgi_ctx != NULL) { | ||||||
|  |             fastcgi_close(ctx->fcgi_ctx); | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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; | ||||||
|  |     char buf[1024]; | ||||||
|  |  | ||||||
|  |     int mode, ret; | ||||||
|  |     if (strends(uri->filename, ".php")) { | ||||||
|  |         mode = FASTCGI_BACKEND_PHP; | ||||||
|  |     } else { | ||||||
|  |         res->status = http_get_status(500); | ||||||
|  |         error("Invalid FastCGI extension: %s", uri->filename); | ||||||
|  |         return 3; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     fastcgi_cnx_t fcgi_cnx_buf; | ||||||
|  |     sock_init(&fcgi_cnx_buf.socket, 0, 0); | ||||||
|  |     fcgi_cnx_buf.req_id = 0; | ||||||
|  |     fcgi_cnx_buf.r_addr = ctx->socket.addr; | ||||||
|  |     fcgi_cnx_buf.r_host = (ctx->host[0] != 0) ? ctx->host : NULL; | ||||||
|  |  | ||||||
|  |     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_buf, mode, ctx->req_num, client, req, uri) != 0) { | ||||||
|  |         fastcgi_close_cnx(&fcgi_cnx_buf); | ||||||
|  |         res->status = http_get_status(503); | ||||||
|  |         sprintf(err_msg, "Unable to communicate with FastCGI socket."); | ||||||
|  |         return 3; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     (*fcgi_cnx) = &fcgi_cnx_buf; | ||||||
|  |     fastcgi_handle_connection(ctx, fcgi_cnx); | ||||||
|  |  | ||||||
|  |     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 (strcontains(client_transfer_encoding, "chunked")) { | ||||||
|  |         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); | ||||||
|  |  | ||||||
|  |     if ((ret = fastcgi_header(*fcgi_cnx, res, err_msg)) != 0) { | ||||||
|  |         if (ret == -1) res->status = http_get_status(502); | ||||||
|  |         return ret; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     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 && | ||||||
|  |         strstarts(content_type, "text/html") && | ||||||
|  |         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; | ||||||
|  |     ctx->content_length = -1; | ||||||
|  |  | ||||||
|  |     if (http_get_header_field(&res->hdr, "Content-Length") == NULL) { | ||||||
|  |         http_add_header_field(&res->hdr, "Transfer-Encoding", "chunked"); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static void fastcgi_next_cb(chunk_ctx_t *ctx) { | ||||||
|  |     if(ctx->client->fcgi_ctx) { | ||||||
|  |         fastcgi_close(ctx->client->fcgi_ctx); | ||||||
|  |         ctx->client->fcgi_ctx = NULL; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     fastcgi_handle(ctx->client); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static void fastcgi_error_cb(chunk_ctx_t *ctx) { | ||||||
|  |     if (ctx->client->chunks_transferred) | ||||||
|  |         return; | ||||||
|  |  | ||||||
|  |     logger_set_prefix("[%s%*s%s]%s", BLD_STR, ADDRSTRLEN, ctx->client->req_host, CLR_STR, ctx->client->log_prefix); | ||||||
|  |  | ||||||
|  |     warning("Closing connection due to FastCGI error"); | ||||||
|  |     if(ctx->client->fcgi_ctx) { | ||||||
|  |         fastcgi_close(ctx->client->fcgi_ctx); | ||||||
|  |         ctx->client->fcgi_ctx = NULL; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     tcp_close(ctx->client); | ||||||
|  |  | ||||||
|  |     errno = 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int fastcgi_handler_2(client_ctx_t *ctx, fastcgi_cnx_t *fcgi_cnx) { | ||||||
|  |     int chunked = strcontains(http_get_header_field(&ctx->res.hdr, "Transfer-Encoding"), "chunked"); | ||||||
|  |  | ||||||
|  |     if (chunked) { | ||||||
|  |         handle_chunks(ctx, &fcgi_cnx->out, SOCK_CHUNKED, fastcgi_next_cb, fastcgi_error_cb); | ||||||
|  |         return 1; | ||||||
|  |     } else { | ||||||
|  |         fastcgi_send(fcgi_cnx, &ctx->socket); | ||||||
|  |         fastcgi_close(ctx->fcgi_ctx); | ||||||
|  |         ctx->fcgi_ctx = NULL; | ||||||
|  |         fastcgi_handle(ctx); | ||||||
|  |         return 2; | ||||||
|  |     } | ||||||
|  | } | ||||||
							
								
								
									
										96
									
								
								src/worker/func.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										96
									
								
								src/worker/func.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,96 @@ | |||||||
|  | /** | ||||||
|  |  * 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 "../defs.h" | ||||||
|  | #include "../lib/sock.h" | ||||||
|  | #include "../lib/http.h" | ||||||
|  | #include "../lib/uri.h" | ||||||
|  | #include "../lib/config.h" | ||||||
|  | #include "../lib/proxy.h" | ||||||
|  | #include "../lib/fastcgi.h" | ||||||
|  |  | ||||||
|  | typedef struct { | ||||||
|  |     sock socket; | ||||||
|  |     int req_num; | ||||||
|  |     unsigned char s_keep_alive:1, c_keep_alive:1, use_fastcgi:4, use_proxy:2, ws_close:2, chunks_transferred:1; | ||||||
|  |     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; | ||||||
|  |     void *fcgi_ctx; | ||||||
|  | } client_ctx_t; | ||||||
|  |  | ||||||
|  | typedef struct { | ||||||
|  |     client_ctx_t *client; | ||||||
|  |     sock *socket; | ||||||
|  |     void *other; | ||||||
|  | } ws_ctx_t; | ||||||
|  |  | ||||||
|  | typedef struct { | ||||||
|  |     int closed:2; | ||||||
|  |     client_ctx_t *client; | ||||||
|  |     fastcgi_cnx_t cnx; | ||||||
|  | } fastcgi_ctx_t; | ||||||
|  |  | ||||||
|  | typedef struct { | ||||||
|  |     client_ctx_t *client; | ||||||
|  |     sock *socket; | ||||||
|  |     int flags; | ||||||
|  |     void (*next_cb)(void *); | ||||||
|  |     void (*err_cb)(void *); | ||||||
|  | } chunk_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 fastcgi_frame_handler_func(fastcgi_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | void proxy_handler_func(client_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | void ws_frame_handler_func(ws_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | void chunk_handler_func(chunk_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); | ||||||
|  |  | ||||||
|  | int ws_handle_connection(client_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | void ws_close(ws_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | int handle_chunks(client_ctx_t *ctx, sock *socket, int flags, void (*next_cb)(chunk_ctx_t *), void (*err_cb)(chunk_ctx_t *)); | ||||||
|  |  | ||||||
|  | int fastcgi_handle_connection(client_ctx_t *ctx, fastcgi_cnx_t **cnx); | ||||||
|  |  | ||||||
|  | void fastcgi_close(fastcgi_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | #endif //SESIMOS_FUNC_H | ||||||
							
								
								
									
										251
									
								
								src/worker/local_handler.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										251
									
								
								src/worker/local_handler.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,251 @@ | |||||||
|  | /** | ||||||
|  |  * 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, 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 range_handler(client_ctx_t *ctx) { | ||||||
|  |     char buf[64]; | ||||||
|  |     long num0, num1, num2; | ||||||
|  |     char *ptr; | ||||||
|  |     int mode; | ||||||
|  |     const char *range = http_get_header_field(&ctx->req.hdr, "Range"); | ||||||
|  |  | ||||||
|  |     if (strcontains(range, ",")) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     ctx->file = fopen(ctx->uri.filename, "rb"); | ||||||
|  |  | ||||||
|  |     if (strstarts(range, "bytes=")) { | ||||||
|  |         mode = 0; | ||||||
|  |         range += 6; | ||||||
|  |         num0 = fsize(ctx->file), num1 = 0, num2 = num0 - 1; | ||||||
|  |         sprintf(buf, "bytes */%li", num0); | ||||||
|  |     } else if (strstarts(range, "lines=") && mime_is_text(ctx->uri.meta->type)) { | ||||||
|  |         mode = 1; | ||||||
|  |         range += 6; | ||||||
|  |         num0 = flines(ctx->file), num1 = 1, num2 = num0; | ||||||
|  |         sprintf(buf, "lines */%li", num0); | ||||||
|  |     } else { | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     http_add_header_field(&ctx->res.hdr, "Content-Range", buf); | ||||||
|  |  | ||||||
|  |     if ((ptr = strchr(range, '-')) == NULL) | ||||||
|  |         return -1; | ||||||
|  |  | ||||||
|  |     if (num0 == 0) { | ||||||
|  |         ctx->content_length = 0; | ||||||
|  |         return 0; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if (ptr[1] != 0) num2 = (long) strtoul(ptr + 1, NULL, 10); | ||||||
|  |     if (ptr != range) { | ||||||
|  |         num1 = (long) strtoul(range, NULL, 10); | ||||||
|  |     } else { | ||||||
|  |         if (mode == 0) { | ||||||
|  |             num1 = num0 - num2 - 1; | ||||||
|  |             num2 = num0 - 1; | ||||||
|  |         } else if (mode == 1) { | ||||||
|  |             num1 = num0 - num2 + 1; | ||||||
|  |             num2 = num0; | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if ((mode == 0 && (num1 >= num0 || num2 >= num0 || num1 > num2 || num1 < 0 || num2 < 0)) || | ||||||
|  |         (mode == 1 && (num1 > num0 || num2 > num0 || num1 > num2 || num1 <= 0 || num2 <= 0))) | ||||||
|  |     { | ||||||
|  |         return -1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     sprintf(buf, "%s %li-%li/%li", (mode == 0) ? "bytes" : "lines", num1, num2, num0); | ||||||
|  |     http_remove_header_field(&ctx->res.hdr, "Content-Range", HTTP_REMOVE_ALL); | ||||||
|  |     http_add_header_field(&ctx->res.hdr, "Content-Range", buf); | ||||||
|  |  | ||||||
|  |     if (mode == 0) { | ||||||
|  |         fseek(ctx->file, num1, SEEK_SET); | ||||||
|  |         ctx->content_length = num2 - num1 + 1; | ||||||
|  |     } else if (mode == 1) { | ||||||
|  |         fseekl(ctx->file, num1); | ||||||
|  |         ctx->content_length = file_get_line_pos(ctx->file, num2 + 1) - file_get_line_pos(ctx->file, num1); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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 (streq(req->method, "TRACE")) { | ||||||
|  |         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.fields); 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 (strstarts(uri->req_path, "/.well-known/")) { | ||||||
|  |         http_add_header_field(&res->hdr, "Access-Control-Allow-Origin", "*"); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     if ((!strstarts(uri->req_path, "/.well-known/") && strcontains(uri->path, "/.")) || strends(uri->filename, ".inc") || strends(uri->filename, ".inc.php")) { | ||||||
|  |         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); | ||||||
|  |         cache_init_uri(ctx->conf->cache, uri); | ||||||
|  |  | ||||||
|  |         http_add_header_field(&res->hdr, "Accept-Ranges", mime_is_text(uri->meta->type) ? "bytes, lines" : "bytes"); | ||||||
|  |  | ||||||
|  |         if (!streq(req->method, "GET") && !streq(req->method, "HEAD")) { | ||||||
|  |             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; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         const char *last_modified = http_format_date(uri->meta->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 && strcontains(accept_encoding, "br")) { | ||||||
|  |                 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 && strcontains(accept_encoding, "gzip")) { | ||||||
|  |                 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) { | ||||||
|  |             strcpy(buf1, uri->meta->etag); | ||||||
|  |             if (enc) { | ||||||
|  |                 strcat(buf1, "-"); | ||||||
|  |                 strcat(buf1, (enc & COMPRESS_BR) ? "br" : (enc & COMPRESS_GZ) ? "gzip" : ""); | ||||||
|  |             } | ||||||
|  |             http_add_header_field(&res->hdr, "ETag", buf1); | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         http_add_header_field(&res->hdr, "Cache-Control", mime_is_text(uri->meta->type) ? "public, max-age=3600" : "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 && !strcontains(if_none_match, uri->meta->etag)) || | ||||||
|  |             (accept_if_modified_since && streq(if_modified_since, last_modified))) | ||||||
|  |         { | ||||||
|  |             res->status = http_get_status(304); | ||||||
|  |             return 0; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if (http_get_header_field(&req->hdr, "Range") != NULL) { | ||||||
|  |             if (range_handler(ctx) == 0) { | ||||||
|  |                 res->status = http_get_status(206); | ||||||
|  |             } else { | ||||||
|  |                 if (ctx->file) { | ||||||
|  |                     fclose(ctx->file); | ||||||
|  |                     ctx->file = NULL; | ||||||
|  |                 } | ||||||
|  |                 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); | ||||||
|  |                 res->status = http_get_status(416); | ||||||
|  |             } | ||||||
|  |             return 0; | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         if (ctx->file == NULL) ctx->file = fopen(uri->filename, "rb"); | ||||||
|  |         ctx->content_length = fsize(ctx->file); | ||||||
|  |     } else { | ||||||
|  |         return 1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
							
								
								
									
										142
									
								
								src/worker/proxy_handler.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										142
									
								
								src/worker/proxy_handler.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,142 @@ | |||||||
|  | /** | ||||||
|  |  * 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> | ||||||
|  |  | ||||||
|  | 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, 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) { | ||||||
|  |         if (proxy_handler_2(ctx) == 1) { | ||||||
|  |             // chunked | ||||||
|  |             return; | ||||||
|  |         } | ||||||
|  |         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 && | ||||||
|  |             (strcontains(connection, "upgrade") || strcontains(connection, "Upgrade")) && | ||||||
|  |             streq(upgrade, "websocket")) | ||||||
|  |         { | ||||||
|  |             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 = streq(buf, ws_accept) ? 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 || | ||||||
|  |                 streq(content_length_f, "0") || | ||||||
|  |                 (content_length_f != NULL && strstarts(content_type, "text/html")))) | ||||||
|  |         { | ||||||
|  |             long content_len = (!streq(ctx->req.method, "HEAD") && content_length_f != NULL) ? strtol(content_length_f, NULL, 10) : 0; | ||||||
|  |             if (content_len < sizeof(ctx->msg_content)) { | ||||||
|  |                 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; | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return streq(ctx->req.method, "HEAD") ? 1 : 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static void proxy_chunk_next_cb(chunk_ctx_t *ctx) { | ||||||
|  |     proxy_unlock_ctx(ctx->client->proxy); | ||||||
|  |     ctx->client->proxy = NULL; | ||||||
|  |  | ||||||
|  |     request_complete(ctx->client); | ||||||
|  |     handle_request(ctx->client); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static void proxy_chunk_err_cb(chunk_ctx_t *ctx) { | ||||||
|  |     ctx->client->c_keep_alive = 0; | ||||||
|  |     proxy_chunk_next_cb(ctx); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int proxy_handler_2(client_ctx_t *ctx) { | ||||||
|  |     const char *transfer_encoding = http_get_header_field(&ctx->res.hdr, "Transfer-Encoding"); | ||||||
|  |     int chunked = strcontains(transfer_encoding, "chunked"); | ||||||
|  |  | ||||||
|  |     const char *content_len = http_get_header_field(&ctx->res.hdr, "Content-Length"); | ||||||
|  |     unsigned long len_to_send = (content_len != NULL) ? strtol(content_len, NULL, 10) : 0; | ||||||
|  |  | ||||||
|  |     if (chunked) { | ||||||
|  |         handle_chunks(ctx, &ctx->proxy->proxy, SOCK_CHUNKED, proxy_chunk_next_cb, proxy_chunk_err_cb); | ||||||
|  |         return 1; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     int ret; | ||||||
|  |     if ((ret = proxy_send(ctx->proxy, &ctx->socket, len_to_send, 0)) == -1) { | ||||||
|  |         ctx->c_keep_alive = 0; | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return ret; | ||||||
|  | } | ||||||
							
								
								
									
										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", 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->chunks_transferred = 0; | ||||||
|  |     ctx->fcgi_ctx = NULL; | ||||||
|  |     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); | ||||||
|  |     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 = (strcontains(hdr_connection, "keep-alive") || strcontains(hdr_connection, "Keep-Alive")); | ||||||
|  |     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, ADDRSTRLEN, ctx->req_host, CLR_STR, ctx->log_prefix); | ||||||
|  |     info(BLD_STR "%s %s", req->method, req->uri); | ||||||
|  |  | ||||||
|  |     if (strstarts(req->uri, "/.sesimos/res/")) { | ||||||
|  |         if (!streq(req->method, "GET") && !streq(req->method, "HEAD")) { | ||||||
|  |             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 (streq(req->uri + 14, r->name)) { | ||||||
|  |                 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 && streq(req->method, "TRACE")) { | ||||||
|  |         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 && !strstarts(uri->uri, "/.well-known/")); | ||||||
|  |         if (!streq(uri->uri, buf0) || 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->code); | ||||||
|  |             const http_status_msg *http_msg = http_get_error_msg(res->status->code); | ||||||
|  |  | ||||||
|  |             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 (strstarts(ctx->msg_content, "<!DOCTYPE html>") || strstarts(ctx->msg_content, "<html")) { | ||||||
|  |                 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); | ||||||
|  |             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, 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->code), 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 (!streq(req->method, "HEAD")) { | ||||||
|  |         // default response | ||||||
|  |         if (ctx->msg_buf != NULL) { | ||||||
|  |             ret = sock_send_x(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_x(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", 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); | ||||||
|  | } | ||||||
							
								
								
									
										108
									
								
								src/worker/tcp_acceptor.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										108
									
								
								src/worker/tcp_acceptor.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,108 @@ | |||||||
|  | /** | ||||||
|  |  * 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> | ||||||
|  |  | ||||||
|  | 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; | ||||||
|  |  | ||||||
|  |     memset(ctx->_c_addr, 0, sizeof(ctx->_c_addr)); | ||||||
|  |     memset(ctx->_s_addr, 0, sizeof(ctx->_s_addr)); | ||||||
|  |     inet_ntop(ctx->socket._addr.ipv6.sin6_family, &ctx->socket._addr.ipv6.sin6_addr, ctx->_c_addr, sizeof(ctx->_c_addr)); | ||||||
|  |     if (strstarts(ctx->_c_addr, "::ffff:")) { | ||||||
|  |         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 (strstarts(ctx->_s_addr, "::ffff:")) { | ||||||
|  |         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]*/ "", ADDRSTRLEN, ctx->socket.addr, | ||||||
|  |             ntohs(ctx->socket._addr.ipv6.sin6_port), CLR_STR); | ||||||
|  |  | ||||||
|  |     logger_set_prefix("[%*s]%s", ADDRSTRLEN, ctx->socket.s_addr, ctx->log_prefix); | ||||||
|  |  | ||||||
|  |     sock *client = &ctx->socket; | ||||||
|  |     ctx->cnx_s = clock_micros(); | ||||||
|  |  | ||||||
|  |     sock_reverse_lookup(&ctx->socket, ctx->host, sizeof(ctx->host)); | ||||||
|  |  | ||||||
|  |     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); | ||||||
|  |  | ||||||
|  |         int ret; | ||||||
|  |         if ((ret = SSL_accept(client->ssl)) != 1) { | ||||||
|  |             sock_error(client, 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; | ||||||
|  |     ctx->chunks_transferred = 0; | ||||||
|  |  | ||||||
|  |     return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void tcp_close(client_ctx_t *ctx) { | ||||||
|  |     errno = 0; | ||||||
|  |     logger_set_prefix("[%*s]%s", 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); | ||||||
|  | } | ||||||
							
								
								
									
										95
									
								
								src/worker/ws_frame_handler.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										95
									
								
								src/worker/ws_frame_handler.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,95 @@ | |||||||
|  | /** | ||||||
|  |  * 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", 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", 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; | ||||||
|  | } | ||||||
							
								
								
									
										115
									
								
								src/workers.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										115
									
								
								src/workers.c
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,115 @@ | |||||||
|  | /** | ||||||
|  |  * 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_ctx, proxy_handler_ctx, | ||||||
|  |               ws_frame_handler_ctx, chunk_handler_ctx, fastcgi_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,       8, 64, (void (*)(void *)) request_handler_func,       "req"); | ||||||
|  |     mpmc_init(&local_handler_ctx,         8, 64, (void (*)(void *)) local_handler_func,         "local"); | ||||||
|  |     mpmc_init(&fastcgi_handler_ctx,       8, 64, (void (*)(void *)) fastcgi_handler_func,       "fcgi"); | ||||||
|  |     mpmc_init(&proxy_handler_ctx,         8, 64, (void (*)(void *)) proxy_handler_func,         "proxy"); | ||||||
|  |     mpmc_init(&ws_frame_handler_ctx,      8, 64, (void (*)(void *)) ws_frame_handler_func,      "ws"); | ||||||
|  |     mpmc_init(&chunk_handler_ctx,         8, 64, (void (*)(void *)) chunk_handler_func,         "chunk"); | ||||||
|  |     mpmc_init(&fastcgi_frame_handler_ctx, 8, 64, (void (*)(void *)) fastcgi_frame_handler_func, "fcgi_f"); | ||||||
|  |     return -1; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void workers_stop(void) { | ||||||
|  |     mpmc_stop(&tcp_acceptor_ctx); | ||||||
|  |     mpmc_stop(&local_handler_ctx); | ||||||
|  |     mpmc_stop(&fastcgi_handler_ctx); | ||||||
|  |     mpmc_stop(&proxy_handler_ctx); | ||||||
|  |     mpmc_stop(&request_handler_ctx); | ||||||
|  |     mpmc_stop(&ws_frame_handler_ctx); | ||||||
|  |     mpmc_stop(&chunk_handler_ctx); | ||||||
|  |     mpmc_stop(&fastcgi_frame_handler_ctx); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | void workers_destroy(void) { | ||||||
|  |     mpmc_destroy(&tcp_acceptor_ctx); | ||||||
|  |     mpmc_destroy(&local_handler_ctx); | ||||||
|  |     mpmc_destroy(&fastcgi_handler_ctx); | ||||||
|  |     mpmc_destroy(&proxy_handler_ctx); | ||||||
|  |     mpmc_destroy(&request_handler_ctx); | ||||||
|  |     mpmc_destroy(&ws_frame_handler_ctx); | ||||||
|  |     mpmc_destroy(&chunk_handler_ctx); | ||||||
|  |     mpmc_destroy(&fastcgi_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_ctx, ctx); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int fastcgi_handle_frame_cb(fastcgi_ctx_t *ctx) { | ||||||
|  |     return mpmc_queue(&fastcgi_frame_handler_ctx, ctx); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int fastcgi_handle_frame(fastcgi_ctx_t *ctx) { | ||||||
|  |     return async(&ctx->cnx.socket, ASYNC_WAIT_READ, 0, ctx, | ||||||
|  |                  (void (*)(void *)) fastcgi_handle_frame_cb, | ||||||
|  |                  (void (*)(void *)) fastcgi_close, | ||||||
|  |                  (void (*)(void *)) fastcgi_close); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | 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); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int handle_chunk_cb(chunk_ctx_t *ctx) { | ||||||
|  |     return mpmc_queue(&chunk_handler_ctx, ctx); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int handle_chunk(chunk_ctx_t *ctx) { | ||||||
|  |     return async(ctx->socket, ASYNC_WAIT_READ, 0, ctx, | ||||||
|  |                  (void (*)(void *)) handle_chunk_cb, | ||||||
|  |                  (void (*)(void *)) ctx->err_cb, | ||||||
|  |                  (void (*)(void *)) ctx->err_cb); | ||||||
|  | } | ||||||
							
								
								
									
										36
									
								
								src/workers.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										36
									
								
								src/workers.h
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,36 @@ | |||||||
|  | /** | ||||||
|  |  * 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 fastcgi_handle_frame(fastcgi_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | int proxy_handle(client_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | int ws_handle_frame(ws_ctx_t *ctx); | ||||||
|  |  | ||||||
|  | int handle_chunk(chunk_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