Fix SIGINT problem in Docker image

yhirose · Sep 4, 2024 · 87fab84 · 87fab84
1 parent 4e6055f
commit 87fab84
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,8 +1,7 @@
-FROM ubuntu AS builder
+FROM yhirose4dockerhub/ubuntu-builder AS builder
 WORKDIR /build
 COPY httplib.h .
 COPY docker/main.cc .
-RUN apt update && apt install g++ -y
 RUN g++ -std=c++23 -static -o server -O2 -I. -DCPPHTTPLIB_USE_POLL main.cc && strip server
 
 FROM scratch

diff --git a/docker/main.cc b/docker/main.cc
@@ -23,6 +23,8 @@ constexpr auto error_html = R"(<html>
 </html>
 )";
 
+void sigint_handler(int s) { exit(1); }
+
 std::string time_local() {
   auto p = std::chrono::system_clock::now();
   auto t = std::chrono::system_clock::to_time_t(p);
@@ -49,6 +51,8 @@ std::string log(auto &req, auto &res) {
 }
 
 int main(int argc, const char **argv) {
+  signal(SIGINT, sigint_handler);
+
   auto base_dir = "./html";
   auto host = "0.0.0.0";
   auto port = 80;