package server import "bytes" import "context" import "crypto/tls" import "flag" import "fmt" import "io" import "io/ioutil" import "log" import "net" import "net/http" import "os" import "os/signal" import "runtime" import "runtime/debug" import "runtime/pprof" import "strconv" import "strings" import "sync" import "sync/atomic" import "syscall" import "time" import "unsafe" import "github.com/colinmarc/cdb" import "github.com/valyala/fasthttp" import "github.com/lucas-clemente/quic-go" import "github.com/lucas-clemente/quic-go/http3" import . "github.com/volution/kawipiko/lib/common" import . "github.com/volution/kawipiko/lib/server" import "github.com/volution/kawipiko/cmd/version" import "github.com/volution/kawipiko/embedded" import _ "embed" type server struct { httpPlain1Server *fasthttp.Server httpPlain2Server *http.Server httpTls1Server *fasthttp.Server httpTls2Server *http.Server httpQuicServer *http3.Server cdbReader *cdb.CDB cachedReferences map[string][2]uint64 cachedDataMeta map[uint64][]byte cachedDataContent map[uint64][]byte hostsDisabled bool specialPagesEnabled bool securityHeadersEnabled bool securityHeadersTls bool http1Disabled bool http2Disabled bool http3AltSvc string quiet bool debug bool dummy bool dummyEmpty bool dummyDelay time.Duration } func (_server *server) ServeUnwrapped (_context *fasthttp.RequestCtx) () { if _server.dummy { if !_server.dummyEmpty { _server.ServeDummy (_context) } else { _context.Response.SetStatusCode (fasthttp.StatusOK) } if _server.dummyDelay != 0 { time.Sleep (_server.dummyDelay) } return } _request := (*fasthttp.Request) (NoEscape (unsafe.Pointer (&_context.Request))) _requestHeaders := (*fasthttp.RequestHeader) (NoEscape (unsafe.Pointer (&_request.Header))) _response := (*fasthttp.Response) (NoEscape (unsafe.Pointer (&_context.Response))) _responseHeaders := (*fasthttp.ResponseHeader) (NoEscape (unsafe.Pointer (&_response.Header))) _requestMethod := _requestHeaders.Method () _requestHost := _requestHeaders.Host () _requestUri := _requestHeaders.RequestURI () _requestUriString_0 := BytesToString (_requestUri) _requestUriString := NoEscapeString (&_requestUriString_0) _pathBuffer := [256]byte {} _keyBufferLarge := [256 + 16]byte {} _keyBufferSmall := [8]byte {} _path := _pathBuffer[:0] _path = append (_path, _requestUri ...) if _pathLimit := bytes.IndexByte (_path, '?'); _pathLimit > 0 { _path = _path[: _pathLimit] } // FIXME: Decode path according to `decodeArgAppendNoPlus`! _pathLen := len (_path) if ! bytes.Equal (StringToBytes (http.MethodGet), _requestMethod) { if !_server.quiet { log.Printf ("[ww] [bce7a75b] [http-x..] invalid method `%s` for `%s`!\n", BytesToString (_requestHeaders.Method ()), *_requestUriString) } _server.ServeError (_context, http.StatusMethodNotAllowed, nil, true) return } if (_pathLen == 0) || (_path[0] != '/') { if !_server.quiet { log.Printf ("[ww] [fa6b1923] [http-x..] invalid path `%s`!\n", *_requestUriString) } _server.ServeError (_context, http.StatusBadRequest, nil, true) return } if bytes.HasPrefix (_path, StringToBytes ("/__/")) { if !_server.specialPagesEnabled { _server.ServeError (_context, http.StatusForbidden, nil, false) return } if bytes.Equal (_path, StringToBytes ("/__/heartbeat")) || bytes.HasPrefix (_path, StringToBytes ("/__/heartbeat/")) { _server.ServeStatic (_context, http.StatusOK, HeartbeatDataOk, HeartbeatContentType, HeartbeatContentEncoding, false) return } else if bytes.Equal (_path, StringToBytes ("/__/kawipiko/about")) || bytes.Equal (_path, StringToBytes ("/__/kawipiko/about/")) { _server.ServeStatic (_context, http.StatusOK, AboutBannerData, AboutBannerContentType, AboutBannerContentEncoding, true) return } else if bytes.Equal (_path, StringToBytes ("/__/kawipiko/version")) || bytes.Equal (_path, StringToBytes ("/__/kawipiko/version/")) { _server.ServeSpecial (_context, "version", true) return } else if bytes.Equal (_path, StringToBytes ("/__/kawipiko/manual.txt")) { _server.ServeSpecial (_context, "manual.txt", true) return } else if bytes.Equal (_path, StringToBytes ("/__/kawipiko/manual.html")) { _server.ServeSpecial (_context, "manual.html", true) return } else if bytes.Equal (_path, StringToBytes ("/__/kawipiko/sources.md5")) { _server.ServeSpecial (_context, "sources.md5", true) return } else if bytes.Equal (_path, StringToBytes ("/__/kawipiko/sources.cpio")) { _server.ServeSpecial (_context, "sources.cpio", true) return } else if bytes.Equal (_path, StringToBytes ("/__/kawipiko/sbom.txt")) { _server.ServeSpecial (_context, "sbom.txt", true) return } else if bytes.Equal (_path, StringToBytes ("/__/kawipiko/sbom.json")) { _server.ServeSpecial (_context, "sbom.json", true) return } else if bytes.HasPrefix (_path, StringToBytes ("/__/kawipiko/banners/errors/")) { _code := _path[len ("/__/kawipiko/banners/errors/") :] if _code, _error := strconv.Atoi (BytesToString (*NoEscapeBytes (&_code))); _error == nil { _banner, _bannerFound := ErrorBannersData[uint (_code)] if (_code > 0) && _bannerFound { _server.ServeStatic (_context, http.StatusOK, _banner, ErrorBannerContentType, ErrorBannerContentEncoding, true) return } } _server.ServeError (_context, http.StatusBadRequest, nil, true) return } else { _server.ServeError (_context, http.StatusBadRequest, nil, true) return } } var _referencesFound bool var _referencesValues [2]uint64 var _referencesBuffer []byte for _hostIdx := 0; _hostIdx < 2; _hostIdx += 1 { if (_hostIdx == 0) && _server.hostsDisabled { continue } if !_referencesFound { for _, _namespace := range _namespaces_a_static { if _server.cachedReferences != nil { _key := _keyBufferLarge[:0] _key = append (_key, KeyNamespacePrefix (_namespace), ':') if _hostIdx == 0 { _key = append (_key, "://" ...) _key = append (_key, _requestHost ...) } _key = append (_key, _path ...) _referencesValues, _referencesFound = _server.cachedReferences[BytesToString (*NoEscapeBytes (&_key))] } else { _key := _keyBufferLarge[:0] _key = append (_key, KeyNamespacePrefix (_namespace), ':') if _hostIdx == 0 { _key = append (_key, "://" ...) _key = append (_key, _requestHost ...) } _key = append (_key, _path ...) if _value, _error := _server.cdbReader.GetWithCdbHash (_key); _error == nil { _referencesBuffer = _value _referencesFound = _value != nil } else { _server.ServeError (_context, http.StatusInternalServerError, _error, false) return } } if _referencesFound { break } } } if !_referencesFound { if bytes.Equal (StringToBytes ("/favicon.ico"), _path) { if _hostIdx != 0 { _server.ServeStatic (_context, http.StatusOK, FaviconData, FaviconContentType, FaviconContentEncoding, true) return } } } if !_referencesFound { for _pathLimit := bytes.LastIndexByte (_path, '/'); _pathLimit >= 0; _pathLimit = bytes.LastIndexByte (_path[: _pathLimit], '/') { for _, _namespace := range _namespaces_b_static { if _server.cachedReferences != nil { _key := _keyBufferLarge[:0] _key = append (_key, KeyNamespacePrefix (_namespace), ':') if _hostIdx == 0 { _key = append (_key, "://" ...) _key = append (_key, _requestHost ...) } _key = append (_key, _path[: _pathLimit] ...) _key = append (_key, "/*" ...) _referencesValues, _referencesFound = _server.cachedReferences[BytesToString (*NoEscapeBytes (&_key))] } else { _key := _keyBufferLarge[:0] _key = append (_key, KeyNamespacePrefix (_namespace), ':') if _hostIdx == 0 { _key = append (_key, "://" ...) _key = append (_key, _requestHost ...) } _key = append (_key, _path[: _pathLimit] ...) _key = append (_key, "/*" ...) if _value, _error := _server.cdbReader.GetWithCdbHash (_key); _error == nil { _referencesBuffer = _value _referencesFound = _value != nil } else { _server.ServeError (_context, http.StatusInternalServerError, _error, false) return } } if _referencesFound { break } } if _referencesFound { break } } } if _referencesFound { break } } if !_referencesFound { if !_server.quiet { log.Printf ("[ww] [7416f61d] [http-x..] not found `%s`!\n", *_requestUriString) } _server.ServeError (_context, http.StatusNotFound, nil, false) return } var _keyMeta, _keyData uint64 if _referencesBuffer != nil { if _keyMeta_0, _keyData_0, _error := DecodeKeysPair (_referencesBuffer); _error == nil { _keyMeta = _keyMeta_0 _keyData = _keyData_0 } else { log.Printf ("[ee] [7ee6c981] [cdb.....] invalid data fingerprints for `%s`!\n", *_requestUriString) _server.ServeError (_context, http.StatusInternalServerError, nil, false) return } } else { _keyMeta = _referencesValues[0] _keyData = _referencesValues[1] } var _data []byte if _server.cachedDataContent != nil { _data, _ = _server.cachedDataContent[_keyData] } else { _key := _keyBufferSmall[:8] if _error := EncodeKeyToBytes_0 (NamespaceDataContent, _keyData, _key); _error != nil { _server.ServeError (_context, http.StatusInternalServerError, _error, false) return } if _value, _error := _server.cdbReader.GetWithCdbHash (_key); _error == nil { _data = _value } else { _server.ServeError (_context, http.StatusInternalServerError, _error, false) return } } if _data == nil { if !_server.quiet { log.Printf ("[ee] [0165c193] [cdb.....] missing data content for `%s`!\n", *_requestUriString) } _server.ServeError (_context, http.StatusInternalServerError, nil, false) return } var _dataMetaRaw []byte if _server.cachedDataMeta != nil { _dataMetaRaw, _ = _server.cachedDataMeta[_keyMeta] } else { _key := _keyBufferSmall[:8] if _error := EncodeKeyToBytes_0 (NamespaceDataMetadata, _keyMeta, _key); _error != nil { _server.ServeError (_context, http.StatusInternalServerError, _error, false) return } if _value, _error := _server.cdbReader.GetWithCdbHash (_key); _error == nil { _dataMetaRaw = _value } else { _server.ServeError (_context, http.StatusInternalServerError, _error, false) return } } if _dataMetaRaw == nil { if !_server.quiet { log.Printf ("[ee] [e8702411] [cdb.....] missing data metadata for `%s`!\n", *_requestUriString) } _server.ServeError (_context, http.StatusInternalServerError, nil, false) return } _responseStatus := http.StatusOK _headersInvalid := false _handleHeader := func (_name []byte, _value []byte) () { if _name[0] != '!' { _responseHeaders.AddBytesKV (_name, _value) } else { switch BytesToString (_name) { case "!Status" : if _value, _error := strconv.Atoi (BytesToString (_value)); _error == nil { if (_value >= 200) && (_value <= 599) { _responseStatus = _value } else { if !_server.quiet { log.Printf ("[ee] [c2f7ec36] [cdb.....] invalid data metadata for `%s`!\n", *_requestUriString) } _headersInvalid = true } } else { if !_server.quiet { log.Printf ("[ee] [beedae55] [cdb.....] invalid data metadata for `%s`!\n", *_requestUriString) } _headersInvalid = true } default : if !_server.quiet { log.Printf ("[ee] [7acc7d90] [cdb.....] invalid data metadata for `%s`!\n", *_requestUriString) } _headersInvalid = true } } } if _error := MetadataDecodeBinaryIterate (_dataMetaRaw, _handleHeader); _error != nil { _server.ServeError (_context, http.StatusInternalServerError, _error, false) return } if _headersInvalid { _server.ServeError (_context, http.StatusInternalServerError, nil, false) return } if _server.securityHeadersTls { _responseHeaders.AddBytesKV (StringToBytes ("Strict-Transport-Security"), StringToBytes ("max-age=31536000")) _responseHeaders.AddBytesKV (StringToBytes ("Content-Security-Policy"), StringToBytes ("upgrade-insecure-requests")) } if _server.securityHeadersEnabled { _responseHeaders.AddBytesKV (StringToBytes ("Referrer-Policy"), StringToBytes ("strict-origin-when-cross-origin")) _responseHeaders.AddBytesKV (StringToBytes ("X-Content-Type-Options"), StringToBytes ("nosniff")) _responseHeaders.AddBytesKV (StringToBytes ("X-XSS-Protection"), StringToBytes ("1; mode=block")) _responseHeaders.AddBytesKV (StringToBytes ("X-Frame-Options"), StringToBytes ("sameorigin")) } if _server.http3AltSvc != "" { _responseHeaders.AddBytesKV (StringToBytes ("Alt-Svc"), StringToBytes (_server.http3AltSvc)) } if _server.debug { log.Printf ("[dd] [b15f3cad] [http-x..] serving for `%s`...\n", *_requestUriString) } if _server.dummyDelay != 0 { time.Sleep (_server.dummyDelay) } _response.SetStatusCode (_responseStatus) _response.SetBodyRaw (_data) atomic.AddUint64 (&_statsRequestsBody, uint64 (len (_data))) } var _namespaces_a_static = []string { NamespaceFilesContent, NamespaceRedirectsContent, NamespaceFoldersContent, } var _namespaces_b_static = []string { NamespaceFilesContent, NamespaceRedirectsContent, } func (_server *server) ServeStatic (_context *fasthttp.RequestCtx, _status uint, _data []byte, _contentType string, _contentEncoding string, _cache bool) () { _response := (*fasthttp.Response) (NoEscape (unsafe.Pointer (&_context.Response))) _responseHeaders := (*fasthttp.ResponseHeader) (NoEscape (unsafe.Pointer (&_response.Header))) _responseHeaders.AddBytesKV (StringToBytes ("Content-Type"), StringToBytes (_contentType)) _responseHeaders.AddBytesKV (StringToBytes ("Content-Encoding"), StringToBytes (_contentEncoding)) if _cache { _responseHeaders.AddBytesKV (StringToBytes ("Cache-Control"), StringToBytes ("public, immutable, max-age=3600")) } else { _responseHeaders.AddBytesKV (StringToBytes ("Cache-Control"), StringToBytes ("no-store, max-age=0")) } if _server.http3AltSvc != "" { _responseHeaders.AddBytesKV (StringToBytes ("Alt-Svc"), StringToBytes (_server.http3AltSvc)) } _response.SetStatusCode (int (_status)) _response.SetBodyRaw (_data) atomic.AddUint64 (&_statsRequestsBody, uint64 (len (_data))) } func (_server *server) ServeRedirect (_context *fasthttp.RequestCtx, _status uint, _path []byte, _cache bool) () { _response := (*fasthttp.Response) (NoEscape (unsafe.Pointer (&_context.Response))) _responseHeaders := (*fasthttp.ResponseHeader) (NoEscape (unsafe.Pointer (&_response.Header))) _responseHeaders.SetCanonical (StringToBytes ("Location"), _path) if _cache { _responseHeaders.AddBytesKV (StringToBytes ("Cache-Control"), StringToBytes ("public, immutable, max-age=3600")) } else { _responseHeaders.AddBytesKV (StringToBytes ("Cache-Control"), StringToBytes ("no-store, max-age=0")) } if _server.http3AltSvc != "" { _responseHeaders.AddBytesKV (StringToBytes ("Alt-Svc"), StringToBytes (_server.http3AltSvc)) } _response.SetStatusCode (int (_status)) } func (_server *server) ServeError (_context *fasthttp.RequestCtx, _status uint, _error error, _cache bool) () { _response := (*fasthttp.Response) (NoEscape (unsafe.Pointer (&_context.Response))) _responseHeaders := (*fasthttp.ResponseHeader) (NoEscape (unsafe.Pointer (&_response.Header))) _responseHeaders.AddBytesKV (StringToBytes ("Content-Type"), StringToBytes (ErrorBannerContentType)) _responseHeaders.AddBytesKV (StringToBytes ("Content-Encoding"), StringToBytes (ErrorBannerContentEncoding)) if _cache { _responseHeaders.AddBytesKV (StringToBytes ("Cache-Control"), StringToBytes ("public, immutable, max-age=3600")) } else { _responseHeaders.AddBytesKV (StringToBytes ("Cache-Control"), StringToBytes ("no-store, max-age=0")) } if _banner, _bannerFound := ErrorBannersData[_status]; _bannerFound { _response.SetBodyRaw (_banner) atomic.AddUint64 (&_statsRequestsBody, uint64 (len (_banner))) } _response.SetStatusCode (int (_status)) if (_error != nil) && !_server.quiet { LogError (_error, "[23d6cb35] [http-x..] failed handling request!") } } func (_server *server) ServeSpecial (_context *fasthttp.RequestCtx, _special string, _cache bool) () { var _data []byte var _contentType string var _contentEncoding string switch _special { case "version" : _buffer := bytes.NewBuffer (nil) if _error := version.Version ("kawipiko-server", "", _buffer); _error != nil { _server.ServeError (_context, http.StatusInternalServerError, _error, false) return } _data = _buffer.Bytes () _contentType = MimeTypeText _contentEncoding = "identity" case "manual.txt" : _data = StringToBytes (manualText) _contentType = MimeTypeText _contentEncoding = "identity" case "manual.html" : _data = StringToBytes (manualHtml) _contentType = MimeTypeHtml _contentEncoding = "identity" case "sources.md5" : _data = StringToBytes (embedded.BuildSourcesMd5) _contentType = MimeTypeText _contentEncoding = "identity" case "sources.cpio" : _data = embedded.BuildSourcesCpioGz _contentType = MimeTypeText _contentEncoding = "application/x-cpio" case "sbom.txt" : _data = StringToBytes (embedded.SbomTxt) _contentType = MimeTypeText _contentEncoding = "identity" case "sbom.json" : _data = StringToBytes (embedded.SbomJson) _contentType = MimeTypeJson _contentEncoding = "identity" default : panic ("[8546f2bd]") } _server.ServeStatic (_context, http.StatusOK, _data, _contentType, _contentEncoding, _cache) } func (_server *server) ServeDummy (_context *fasthttp.RequestCtx) () { _server.ServeStatic (_context, http.StatusOK, DummyData, DummyContentType, DummyContentEncoding, false) } func (_server *server) ServeWrapped (_context *fasthttp.RequestCtx) () { _context.Response.SetStatusCode (-1) _server.ServeUnwrapped (_context) _status := _context.Response.StatusCode () _invalid := false switch { case _status < 100 : _invalid = true case _status < 200 : atomic.AddUint64 (&_statsResponses1xx, 1) case _status < 300 : atomic.AddUint64 (&_statsResponses2xx, 1) case _status < 400 : atomic.AddUint64 (&_statsResponses3xx, 1) case _status < 500 : atomic.AddUint64 (&_statsResponses4xx, 1) case _status < 600 : atomic.AddUint64 (&_statsResponses5xx, 1) default : _invalid = true } if _invalid { if !_server.quiet { log.Printf ("[ee] [3db6b217] [http-x..] invalid status code `%d`!\n", _status) } _context.Response.Reset () _context.Response.SetStatusCode (http.StatusInternalServerError) atomic.AddUint64 (&_statsResponses5xx, 1) } } func (_server *server) ServeFast (_context *fasthttp.RequestCtx) () { atomic.AddUint64 (&_statsRequestsFast, 1) atomic.AddUint64 (&_statsRequestsTotal, 1) _server.ServeWrapped (_context) } func (_server *server) ServeHTTP (_response http.ResponseWriter, _request *http.Request) () { atomic.AddUint64 (&_statsRequestsSlow, 1) atomic.AddUint64 (&_statsRequestsTotal, 1) _requestProtoUnsupported := false switch _request.ProtoMajor { case 1 : _requestProtoUnsupported = _server.http1Disabled || (_request.ProtoMinor < 0) || (_request.ProtoMinor > 1) if _server.debug && !_requestProtoUnsupported { log.Printf ("[dd] [670e36d4] [go-http.] using Go HTTP/1 for `%s`...", _request.URL.Path) } case 2 : _requestProtoUnsupported = _server.http2Disabled || (_request.ProtoMinor != 0) if _server.debug && !_requestProtoUnsupported { log.Printf ("[dd] [524cd64b] [go-http.] using Go HTTP/2 for `%s`...", _request.URL.Path) } case 3 : _requestProtoUnsupported = (_server.httpQuicServer == nil) || (_request.ProtoMinor != 0) if _server.debug && !_requestProtoUnsupported { log.Printf ("[dd] [be95da51] [go-http.] using QUIC HTTP/3 for `%s`...", _request.URL.Path) } default : _requestProtoUnsupported = true } if _requestProtoUnsupported { _request.Close = true _response.WriteHeader (http.StatusHTTPVersionNotSupported) if !_server.quiet { log.Printf ("[ww] [4c44e3c0] [go-http.] protocol HTTP/%d not supported for `%s`!", _request.ProtoMajor, _request.URL.Path) } atomic.AddUint64 (&_statsResponses5xx, 1) return } if _server.dummy { if !_server.dummyEmpty { _responseHeaders := NewHttpResponseWriterHeadersBuffer (http.StatusOK) _responseHeaders.IncludeString ("Content-Type", DummyContentType) _responseHeaders.IncludeString ("Content-Encoding", DummyContentEncoding) _responseHeaders.IncludeString ("Cache-Control", "no-store, max-age=0") _responseHeaders.WriteTo (_response) _response.Write (DummyData) atomic.AddUint64 (&_statsRequestsBody, uint64 (len (DummyData))) } if _server.dummyDelay != 0 { time.Sleep (_server.dummyDelay) } atomic.AddUint64 (&_statsResponses2xx, 1) return } // FIXME: Reimplemnet this to eliminate the HTTP-encode-followed-by-HTTP-decode! var _context *fasthttp.RequestCtx if _context_0 := _requestContextsPool.Get (); _context_0 != nil { _context = _context_0.(*fasthttp.RequestCtx) } else { _context = new (fasthttp.RequestCtx) } defer _requestContextsPool.Put (_context) _context.Request.Reset () _context.Request.Header.SetMethod (_request.Method) _context.Request.Header.SetRequestURI (_request.URL.Path) _context.Response.Reset () _server.ServeWrapped (_context) _responseHeaders := NewHttpResponseWriterHeadersBuffer (_context.Response.Header.StatusCode ()) _context.Response.Header.VisitAll ( func (_key []byte, _value []byte) () { switch BytesToString (_key) { case "Connection", "Content-Length", "Date" : // NOP default : _responseHeaders.IncludeBytes (_key, _value) } }) _responseHeaders.WriteTo (_response) _response.Write (_context.Response.Body ()) } var _requestContextsPool sync.Pool func (_server *server) Printf (_format string, _arguments ... interface{}) () { if !_server.quiet { log.Printf ("[ee] [47765179] [fasthttp] | " + _format, _arguments ...) } } func Main () () { if len (os.Args) == 2 { switch os.Args[1] { case "version", "--version", "-v" : version.Main ("kawipiko-server", "version") return case "--sources-md5" : version.Main ("kawipiko-server", "sources.md5") return case "--sources-cpio" : version.Main ("kawipiko-server", "sources.cpio") return case "--sbom-text", "--sbom-txt", "--sbom" : version.Main ("kawipiko-server", "sbom.txt") return case "--sbom-json" : version.Main ("kawipiko-server", "sbom.json") return case "--help", "-h" : if _, _error := os.Stdout.WriteString (usageText); _error == nil { return } else { AbortError (_error, "[68e52113] unexpected error!") } case "--man" : if _, _error := os.Stdout.WriteString (manualText); _error == nil { return } else { AbortError (_error, "[78817f08] unexpected error!") } } } log.SetPrefix (fmt.Sprintf ("[%8d] ", os.Getpid ())) Main_0 (main_0) } func main_0 () (error) { // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- var _bindPlain1 string var _bindPlain2 string var _bindTls1 string var _bindTls2 string var _bindQuic string var _http1Disabled bool var _http2Disabled bool var _http3AltSvc string var _tlsPrivate string var _tlsPublic string var _tlsEmbeddedRsa bool var _tlsEmbeddedEd bool var _archivePath string var _archiveInmem bool var _archiveMmap bool var _archivePreload bool var _indexPaths bool var _indexDataMeta bool var _indexDataContent bool var _hostsDisabled bool var _specialPagesEnabled bool var _securityHeadersEnabled bool var _securityHeadersTls bool var _timeoutDisabled bool var _processes uint var _threads uint var _slave uint var _quiet bool var _report bool var _debug bool var _dummy bool var _dummyEmpty bool var _dummyDelay time.Duration var _profileCpu string var _profileMem string var _limitMemory uint var _seccompEnabled bool var _isFirst bool var _isMaster bool { _flags := flag.NewFlagSet ("kawipiko-server", flag.ContinueOnError) _flags.Usage = func () () { fmt.Fprintf (os.Stderr, "%s", usageText) } _bindPlain1_0 := _flags.String ("bind", "", "") _bindPlain2_0 := _flags.String ("bind-2", "", "") _bindTls1_0 := _flags.String ("bind-tls", "", "") _bindTls2_0 := _flags.String ("bind-tls-2", "", "") _bindQuic_0 := _flags.String ("bind-quic", "", "") _http1Disabled_0 := _flags.Bool ("http1-disable", false, "") _http2Disabled_0 := _flags.Bool ("http2-disable", false, "") _http3AltSvc_0 := _flags.String ("http3-alt-svc", "", "") _archivePath_0 := _flags.String ("archive", "", "") _archiveInmem_0 := _flags.Bool ("archive-inmem", false, "") _archiveMmap_0 := _flags.Bool ("archive-mmap", false, "") _archivePreload_0 := _flags.Bool ("archive-preload", false, "") _indexAll_0 := _flags.Bool ("index-all", false, "") _indexPaths_0 := _flags.Bool ("index-paths", false, "") _indexDataMeta_0 := _flags.Bool ("index-data-meta", false, "") _indexDataContent_0 := _flags.Bool ("index-data-content", false, "") _timeoutDisabled_0 := _flags.Bool ("timeout-disable", false, "") _hostsDisabled_0 := _flags.Bool ("hosts-disable", false, "") _specialPagesDisabled_0 := _flags.Bool ("special-pages-disable", false, "") _securityHeadersDisabled_0 := _flags.Bool ("security-headers-disable", false, "") _securityHeadersTls_0 := _flags.Bool ("security-headers-tls", false, "") _tlsPrivate_0 := _flags.String ("tls-private", "", "") _tlsPublic_0 := _flags.String ("tls-public", "", "") _tlsBundle_0 := _flags.String ("tls-bundle", "", "") _tlsEmbeddedRsa_0 := _flags.Bool ("tls-self-rsa", false, "") _tlsEmbeddedEd_0 := _flags.Bool ("tls-self-ed25519", false, "") _processes_0 := _flags.Uint ("processes", 0, "") _threads_0 := _flags.Uint ("threads", 0, "") _slave_0 := _flags.Uint ("slave", 0, "") _quiet_0 := _flags.Bool ("quiet", false, "") _report_0 := _flags.Bool ("report", false, "") _debug_0 := _flags.Bool ("debug", false, "") _dummy_0 := _flags.Bool ("dummy", false, "") _dummyEmpty_0 := _flags.Bool ("dummy-empty", false, "") _dummyDelay_0 := _flags.Duration ("dummy-delay", 0, "") _profileCpu_0 := _flags.String ("profile-cpu", "", "") _profileMem_0 := _flags.String ("profile-mem", "", "") _limitMemory_0 := _flags.Uint ("limit-memory", 0, "") _seccompEnabled_0 := _flags.Bool ("seccomp-enable", false, "") FlagsParse (_flags, 0, 0) _bindPlain1 = *_bindPlain1_0 _bindPlain2 = *_bindPlain2_0 _bindTls1 = *_bindTls1_0 _bindTls2 = *_bindTls2_0 _bindQuic = *_bindQuic_0 _http1Disabled = *_http1Disabled_0 _http2Disabled = *_http2Disabled_0 _http3AltSvc = *_http3AltSvc_0 _archivePath = *_archivePath_0 _archiveInmem = *_archiveInmem_0 _archiveMmap = *_archiveMmap_0 _archivePreload = *_archivePreload_0 _indexAll := *_indexAll_0 _indexPaths = _indexAll || *_indexPaths_0 _indexDataMeta = _indexAll || *_indexDataMeta_0 _indexDataContent = _indexAll || *_indexDataContent_0 _hostsDisabled = *_hostsDisabled_0 _specialPagesEnabled = ! *_specialPagesDisabled_0 _securityHeadersEnabled = ! *_securityHeadersDisabled_0 _securityHeadersTls = *_securityHeadersTls_0 _timeoutDisabled = *_timeoutDisabled_0 _processes = *_processes_0 _threads = *_threads_0 _slave = *_slave_0 _quiet = *_quiet_0 && !*_debug_0 _report = *_report_0 _debug = *_debug_0 _dummy = *_dummy_0 _dummyEmpty = *_dummyEmpty_0 _dummyDelay = *_dummyDelay_0 _profileCpu = *_profileCpu_0 _profileMem = *_profileMem_0 _limitMemory = *_limitMemory_0 _seccompEnabled = *_seccompEnabled_0 if _slave == 0 { _isMaster = true } if _slave <= 1 { _isFirst = true } if (_bindPlain1 == "") && (_bindPlain2 == "") && (_bindTls1 == "") && (_bindTls2 == "") && (_bindQuic == "") { AbortError (nil, "[6edd9512] expected bind address argument!") } if (*_tlsBundle_0 != "") && ((*_tlsPrivate_0 != "") || (*_tlsPublic_0 != "")) { AbortError (nil, "[717f5f84] TLS bundle and TLS private/public are mutually exclusive!") } if (*_tlsBundle_0 != "") { _tlsPrivate = *_tlsBundle_0 _tlsPublic = *_tlsBundle_0 } else { _tlsPrivate = *_tlsPrivate_0 _tlsPublic = *_tlsPublic_0 } _tlsEmbeddedRsa = *_tlsEmbeddedRsa_0 _tlsEmbeddedEd = *_tlsEmbeddedEd_0 if ((_tlsPrivate != "") && (_tlsPublic == "")) || ((_tlsPublic != "") && (_tlsPrivate == "")) { AbortError (nil, "[6e5b42e4] TLS private/public must be specified together!") } if (_tlsPrivate != "") && (_tlsEmbeddedRsa || _tlsEmbeddedEd) { AbortError (nil, "[3de098d3] TLS self-signed and TLS bundle or TLS private/public are mutually exclusive!") } if ((_tlsPrivate != "") || (_tlsPublic != "") || _tlsEmbeddedRsa || _tlsEmbeddedEd) && ((_bindTls1 == "") && (_bindTls2 == "") && (_bindQuic == "")) { AbortError (nil, "[4e31f251] TLS certificate specified, but TLS not enabled!") } if _http1Disabled && ((_bindPlain1 != "") || (_bindPlain2 != "")) { AbortError (nil, "[bd16d596] HTTP/1 is mandatory with `--bind` or `--bind-2`!") } if _http1Disabled && (_bindTls1 != "") { AbortError (nil, "[f498816a] HTTP/1 is mandatory with `--bind-tls`!") } if _http1Disabled && (_bindPlain1 == "") && (_bindPlain2 == "") && (_bindTls1 == "") && (_bindTls2 == "") { log.Printf ("[ww] [6bc56c8e] HTTP/1 is not available!\n") } if _http2Disabled && (_bindTls1 == "") && (_bindTls2 == "") { log.Printf ("[ww] [1ed4864c] HTTP/2 is not available!\n") } if (_http3AltSvc != "") && (_bindQuic == "") { log.Printf ("[ww] [93510d2a] HTTP/3 Alt-Svc is not available!\n") } if (_http3AltSvc == "") && (_bindQuic != "") { log.Printf ("[ww] [225bda04] HTTP/3 Alt-Svc is mandatory with QUIC!\n") } if (_http3AltSvc != "") { if strings.HasPrefix (_http3AltSvc, "h3=") { // NOP } else if _host, _port, _error := net.SplitHostPort (_http3AltSvc); _error == nil { _endpoint := net.JoinHostPort (_host, _port) _http3AltSvc = fmt.Sprintf ("h3=\"%s\", h3-29=\"%s\"", _endpoint, _endpoint) } else { AbortError (nil, "[1a5476b1] HTTP/3 Alt-Svc is invalid!") } CanonicalHeaderValueRegister (_http3AltSvc) } _dummy = _dummy || _dummyEmpty if !_dummy { if _archivePath == "" { AbortError (nil, "[eefe1a38] expected archive file argument!") } if _archiveInmem && _archiveMmap { AbortError (nil, "[a2101041] archive 'memory-loaded' and 'memory-mapped' are mutually exclusive!") } if _archiveInmem && _archivePreload { log.Printf ("[ww] [3e8a40e4] archive 'memory-loaded' implies preloading!\n") _archivePreload = false } } else { if _isMaster { log.Printf ("[ww] [8e014192] running in dummy mode; all archive related arguments are ignored!\n") } _archivePath = "" _archiveInmem = false _archiveMmap = false _archivePreload = false _indexPaths = false _indexDataMeta = false _indexDataContent = false } if !_dummy && (_dummyDelay != 0) { if _isMaster { log.Printf ("[ww] [e9296c03] running with a response delay of `%s`!\n", _dummyDelay) } } if _processes < 1 { _processes = 1 } if _threads < 1 { _threads = 1 } if _processes > 1024 { AbortError (nil, "[45736c1d] maximum number of allowed processes is 1024!") } if _threads > 1024 { AbortError (nil, "[c5df3c8d] maximum number of allowed threads is 1024!") } if (_processes * _threads) > 1024 { AbortError (nil, "[b0177488] maximum number of allowed threads in total is 1024!") } if (_limitMemory != 0) && ((_limitMemory > (16 * 1024)) || (_limitMemory < 128)) { AbortError (nil, "[2781f54c] maximum memory limit is between 128 and 16384 MiB!") } if _seccompEnabled && !seccompSupported { AbortError (nil, "[d4d22d4e] Linux seccomp is not supported with this build!") } if _seccompEnabled && (_processes > 1) { AbortError (nil, "[69c06e0c] Linux seccomp is not supported with multiple processes!") } if _seccompEnabled && ((_profilingCpu != "") || (_profilingMem != "")) { AbortError (nil, "[1fb06ca1] Linux seccomp is not supported with profiling!") } if (_processes > 1) && ((_profileCpu != "") || (_profileMem != "")) { AbortError (nil, "[cd18d250] multi-process and profiling are mutually exclusive!") } if (_processes > 1) && (_bindQuic != "") { AbortError (nil, "[d6db77ba] QUIC is only available with a single process!") } } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- if _seccompEnabled { seccompApplyPhase1 () } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- runtime.GOMAXPROCS (int (_threads)) debug.SetGCPercent (50) debug.SetMaxThreads (int (128 * (_threads / 64 + 1))) debug.SetMaxStack (32 * 1024) _httpReduceMemory := false if _limitMemory > 0 { if !_quiet && _isMaster { log.Printf ("[ii] [2c130d70] limiting memory to %d MiB;\n", _limitMemory) } if _error := SysSetrlimit (_limitMemory); _error != nil { AbortError (_error, "[4da96378] failed to configure memory limit!") } } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- if _processes > 1 { log.Printf ("[ii] [06f8c944] [master..] sub-processes starting (`%d` processes with `%d` threads each)...\n", _processes, _threads) _processesJoin := & sync.WaitGroup {} _processesPid := make ([]*os.Process, _processes) _processName := os.Args[0] _processArguments := make ([]string, 0, len (os.Args)) if _bindPlain1 != "" { _processArguments = append (_processArguments, "--bind", _bindPlain1) } if _bindPlain2 != "" { _processArguments = append (_processArguments, "--bind-2", _bindPlain2) } if _bindTls1 != "" { _processArguments = append (_processArguments, "--bind-tls", _bindTls1) } if _bindTls2 != "" { _processArguments = append (_processArguments, "--bind-tls-2", _bindTls2) } if _http1Disabled { _processArguments = append (_processArguments, "--http1-disabled") } if _http2Disabled { _processArguments = append (_processArguments, "--http2-disabled") } if _http3AltSvc != "" { _processArguments = append (_processArguments, "--http3-alt-svc", _http3AltSvc) } if _archivePath != "" { _processArguments = append (_processArguments, "--archive", _archivePath) } if _archiveInmem { _processArguments = append (_processArguments, "--archive-inmem") } if _archiveMmap { _processArguments = append (_processArguments, "--archive-mmap") } if _archivePreload { _processArguments = append (_processArguments, "--archive-preload") } if _indexPaths { _processArguments = append (_processArguments, "--index-paths") } if _indexDataMeta { _processArguments = append (_processArguments, "--index-data-meta") } if _indexDataContent { _processArguments = append (_processArguments, "--index-data-content") } if _hostsDisabled { _processArguments = append (_processArguments, "--hosts-disabled") } if !_specialPagesEnabled { _processArguments = append (_processArguments, "--special-pages-disable") } if !_securityHeadersEnabled { _processArguments = append (_processArguments, "--security-headers-disable") } if _securityHeadersTls { _processArguments = append (_processArguments, "--security-headers-tls") } if _tlsPrivate != "" { _processArguments = append (_processArguments, "--tls-private", _tlsPrivate) } if _tlsPublic != "" { _processArguments = append (_processArguments, "--tls-public", _tlsPublic) } if _tlsEmbeddedRsa { _processArguments = append (_processArguments, "--tls-self-rsa") } if _tlsEmbeddedEd { _processArguments = append (_processArguments, "--tls-self-ed25519") } if _timeoutDisabled { _processArguments = append (_processArguments, "--timeout-disable") } if _limitMemory != 0 { _processArguments = append (_processArguments, "--limit-memory", fmt.Sprintf ("%d", _limitMemory)) } if _seccompEnabled { _processArguments = append (_processArguments, "--seccomp-enable") } if _quiet { _processArguments = append (_processArguments, "--quiet") } if _report { _processArguments = append (_processArguments, "--report") } if _debug { _processArguments = append (_processArguments, "--debug") } if _dummy { _processArguments = append (_processArguments, "--dummy") } if _dummyEmpty { _processArguments = append (_processArguments, "--dummy-empty") } if _dummyDelay != 0 { _processArguments = append (_processArguments, "--dummy-delay", fmt.Sprintf ("%s", _dummyDelay)) } _processArguments = append (_processArguments, "--threads", fmt.Sprintf ("%d", _threads)) _processAttributes := & os.ProcAttr { Env : []string {}, Files : []*os.File { os.Stdin, os.Stdout, os.Stderr, }, Sys : nil, } for _processIndex, _ := range _processesPid { _processArguments := append ([]string { _processName, "--slave", fmt.Sprintf ("%d", _processIndex + 1) }, _processArguments ...) if _processPid, _error := os.StartProcess (_processName, _processArguments, _processAttributes); _error == nil { _processesJoin.Add (1) _processesPid[_processIndex] = _processPid if !_quiet { log.Printf ("[ii] [63cb22f8] [master..] sub-process `%d` started (with `%d` threads);\n", _processPid.Pid, _threads) } go func (_index int, _processPid *os.Process) () { if _processStatus, _error := _processPid.Wait (); _error == nil { if _processStatus.Success () { if _debug { log.Printf ("[ii] [66b60b81] [master..] sub-process `%d` succeeded;\n", _processPid.Pid) } } else { log.Printf ("[ww] [5d25046b] [master..] sub-process `%d` failed: `%s`; ignoring!\n", _processPid.Pid, _processStatus) } } else { LogError (_error, fmt.Sprintf ("[f1bfc927] [master..] failed waiting for sub-process `%d`; ignoring!", _processPid.Pid)) } _processesPid[_processIndex] = nil _processesJoin.Done () } (_processIndex, _processPid) } else { LogError (_error, "[8892b34d] [master..] failed starting sub-process; ignoring!") } } { _signals := make (chan os.Signal, 32) signal.Notify (_signals, syscall.SIGINT, syscall.SIGTERM) go func () () { for { _signal := <- _signals if _debug { log.Printf ("[ii] [a9243ecb] [master..] signaling sub-processes...\n") } for _, _processPid := range _processesPid { if _processPid != nil { if _error := _processPid.Signal (_signal); _error != nil { LogError (_error, fmt.Sprintf ("[ab681164] [master..] failed signaling sub-process `%d`; ignoring!", _processPid.Pid)) } } } } } () } _processesJoin.Wait () if !_quiet { log.Printf ("[ii] [b949bafc] [master..] sub-processes terminated;\n") } return nil } if _isMaster { log.Printf ("[ii] [6602a54a] [master..] starting (with `%d` threads)...\n", _threads) } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- var _cdbReader *cdb.CDB if _archivePath != "" { if !_quiet && (_debug || _isFirst) { log.Printf ("[ii] [3b788396] [cdb.....] opening archive file `%s`...\n", _archivePath) } var _cdbFile *os.File if _cdbFile_0, _error := os.Open (_archivePath); _error == nil { _cdbFile = _cdbFile_0 } else { AbortError (_error, "[9e0b5ed3] [cdb.....] failed opening archive file!") } var _cdbFileSize int { var _cdbFileSize_0 int64 if _cdbFileStat, _error := _cdbFile.Stat (); _error == nil { _cdbFileSize_0 = _cdbFileStat.Size () } else { AbortError (_error, "[0ccf0a3b] [cdb.....] failed opening archive file!") } if _cdbFileSize_0 < 1024 { AbortError (nil, "[6635a2a8] [cdb.....] failed opening archive: file is too small (or empty)!") } if _cdbFileSize_0 >= (4 * 1024 * 1024 * 1024) { AbortError (nil, "[545bf6ce] [cdb.....] failed opening archive: file is too large!") } _cdbFileSize = int (_cdbFileSize_0) } if _archivePreload { if !_quiet { log.Printf ("[ii] [13f4ebf7] [cdb.....] preloading archive file...\n") } _buffer := [16 * 1024]byte {} _loop : for { switch _, _error := _cdbFile.Read (_buffer[:]); _error { case io.EOF : break _loop case nil : continue _loop default : AbortError (_error, "[a1c3b922] [cdb.....] failed preloading archive file...") } } } if _archiveInmem || _archiveMmap { var _cdbData []byte if _archiveInmem { if _debug { log.Printf ("[ii] [216e584b] [cdb.....] opening memory-loaded archive...\n") } _cdbData = make ([]byte, _cdbFileSize) if _, _error := io.ReadFull (_cdbFile, _cdbData); _error != nil { AbortError (_error, "[73039784] [cdb.....] failed loading archive file!") } } else if _archiveMmap { if _debug { log.Printf ("[ii] [f47fae8a] [cdb.....] opening memory-mapped archive...\n") } if _cdbData_0, _error := syscall.Mmap (int (_cdbFile.Fd ()), 0, int (_cdbFileSize), syscall.PROT_READ, syscall.MAP_SHARED); _error == nil { _cdbData = _cdbData_0 } else { AbortError (_error, "[c0e2632c] [cdb.....] failed mapping archive file!") } if _archivePreload { if _debug { log.Printf ("[ii] [d96b06c9] [cdb.....] preloading memory-loaded archive...\n") } _buffer := [16 * 1024]byte {} _bufferOffset := 0 for { if _bufferOffset == _cdbFileSize { break } _bufferOffset += copy (_buffer[:], _cdbData[_bufferOffset:]) } } } else { panic ("[e4fffcd8]") } if _error := _cdbFile.Close (); _error != nil { AbortError (_error, "[5e0449c2] [cdb.....] failed closing archive file!") } if _cdbReader_0, _error := cdb.NewFromBufferWithHasher (_cdbData, nil); _error == nil { _cdbReader = _cdbReader_0 } else { AbortError (_error, "[27e4813e] [cdb.....] failed opening archive!") } } else { if !_quiet && (_debug || _isFirst) { log.Printf ("[ww] [dd697a66] [cdb.....] using `read`-based archive (with significant performance impact)!\n") } if _cdbReader_0, _error := cdb.NewFromReaderWithHasher (_cdbFile, nil); _error == nil { _cdbReader = _cdbReader_0 } else { AbortError (_error, "[35832022] [cdb.....] failed opening archive!") } } } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- if _seccompEnabled { seccompApplyPhase2 () } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- if _schemaVersion, _error := _cdbReader.GetWithCdbHash ([]byte (NamespaceSchemaVersion)); _error == nil { if _schemaVersion == nil { AbortError (nil, "[09316866] [cdb.....] missing archive schema version!") } else if string (_schemaVersion) != CurrentSchemaVersion { AbortError (nil, "[e6482cf7] [cdb.....] invalid archive schema version!") } } else { AbortError (_error, "[87cae197] [cdb.....] failed opening archive!") } var _cachedReferences map[string][2]uint64 if _indexPaths { _cachedReferences = make (map[string][2]uint64, 128 * 1024) } var _cachedDataMeta map[uint64][]byte if _indexDataMeta { _cachedDataMeta = make (map[uint64][]byte, 128 * 1024) } var _cachedDataContent map[uint64][]byte if _indexDataContent { _cachedDataContent = make (map[uint64][]byte, 128 * 1024) } if _indexPaths || _indexDataMeta || _indexDataContent { if !_quiet { log.Printf ("[ii] [fa5338fd] [cdb.....] indexing archive...\n") } for _, _namespaceContents := range []string { NamespaceFilesContent, NamespaceFoldersContent, NamespaceRedirectsContent } { _namespaceIndex := "" switch _namespaceContents { case NamespaceFilesContent : _namespaceIndex = NamespaceFilesIndex case NamespaceFoldersContent : _namespaceIndex = NamespaceFoldersIndex case NamespaceRedirectsContent : _namespaceIndex = NamespaceRedirectsIndex } var _indexKey string if _key_0, _error := PrepareKeyToString (_namespaceIndex, 1); _error == nil { _indexKey = _key_0 } else { AbortError (_error, "[5289ee67] [cdb.....] failed indexing archive!") } var _index []byte if _index_0, _error := _cdbReader.GetWithCdbHash (StringToBytes (_indexKey)); _error == nil { _index = _index_0 } else { AbortError (_error, "[82299b3d] [cdb.....] failed indexing arcdive!") } if _index == nil { log.Printf ("[ww] [30314f31] [cdb.....] missing archive index `%s`; ignoring!\n", _namespaceIndex) continue } _keyBuffer := [1024]byte {} _keysCount := 0 for { _offset := bytes.IndexByte (_index, '\n') if _offset == 0 { continue } if _offset == -1 { break } _path := _index[: _offset] _index = _index[_offset + 1 :] var _references []byte { _key := _keyBuffer[:0] _key = append (_key, KeyNamespacePrefix (_namespaceContents), ':') _key = append (_key, _path ...) if _references_0, _error := _cdbReader.GetWithCdbHash (_key); _error == nil { if _references_0 != nil { _references = _references_0 } else { AbortError (_error, "[460b3cf1] [cdb.....] failed indexing archive!") } } else { AbortError (_error, "[216f2075] [cdb.....] failed indexing archive!") } } var _keyDataMeta, _keyDataContent uint64 if _keyDataMeta_0, _keyDataContent_0, _error := DecodeKeysPair (_references); _error == nil { _keyDataMeta = _keyDataMeta_0 _keyDataContent = _keyDataContent_0 } else { AbortError (_error, "[7d1a366f] [cdb.....] failed indexing archive!") } if _indexPaths { _key := _keyBuffer[:0] _key = append (_key, KeyNamespacePrefix (_namespaceContents), ':') _key = append (_key, _path ...) _cachedReferences[string (_key)] = [2]uint64 { _keyDataMeta, _keyDataContent } } if _indexDataMeta { if _, _wasCached := _cachedDataMeta[_keyDataMeta]; !_wasCached { _key := _keyBuffer[:8] if _error := EncodeKeyToBytes_0 (NamespaceDataMetadata, _keyDataMeta, _key); _error != nil { AbortError (_error, "[b8cd07f4] [cdb.....] failed indexing archive!") } if _dataMeta, _error := _cdbReader.GetWithCdbHash (_key); _error == nil { if _dataMeta != nil { _cachedDataMeta[_keyDataMeta] = _dataMeta } else { AbortError (_error, "[6df556bf] [cdb.....] failed indexing archive!") } } else { AbortError (_error, "[0d730134] [cdb.....] failed indexing archive!") } } } if _indexDataContent { if _, _wasCached := _cachedDataContent[_keyDataContent]; !_wasCached { _key := _keyBuffer[:8] if _error := EncodeKeyToBytes_0 (NamespaceDataContent, _keyDataContent, _key); _error != nil { AbortError (_error, "[580e387e] [cdb.....] failed indexing archive!") } if _dataContent, _error := _cdbReader.GetWithCdbHash (_key); _error == nil { if _dataContent != nil { _cachedDataContent[_keyDataContent] = _dataContent } else { AbortError (_error, "[4e27fe46] [cdb.....] failed indexing archive!") } } else { AbortError (_error, "[532845ad] [cdb.....] failed indexing archive!") } } } _keysCount += 1 } if !_quiet { log.Printf ("[ii] [0c2abafe] [cdb.....] indexed %d references in `%s`;\n", _keysCount, _namespaceIndex) } } if !_quiet { if _indexPaths { log.Printf ("[ii] [6b7ec5d9] [cdb.....] cached %d references;\n", len (_cachedReferences)) } if _indexDataMeta { log.Printf ("[ii] [5ec4f113] [cdb.....] cached %d meta-data blocks;\n", len (_cachedDataMeta)) } if _indexDataContent { log.Printf ("[ii] [d9680a2f] [cdb.....] cached %d content blocks;\n", len (_cachedDataContent)) } } if len (_cachedReferences) == 0 { _indexPaths = false _cachedReferences = nil } if len (_cachedDataMeta) == 0 { _indexDataMeta = false _cachedDataMeta = nil } if len (_cachedDataContent) == 0 { _indexDataContent = false _cachedDataContent = nil } } if _indexPaths && _indexDataMeta && _indexDataContent { if _error := _cdbReader.Close (); _error == nil { _cdbReader = nil } else { AbortError (_error, "[d7aa79e1] [cdb.....] failed closing archive!") } } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- if _profileCpu != "" { log.Printf ("[ii] [70c210f3] [pprof...] profiling CPU to `%s`...\n", _profileCpu) _stream, _error := os.Create (_profileCpu) if _error != nil { AbortError (_error, "[fd4e0009] [pprof...] failed opening CPU profile!") } _error = pprof.StartCPUProfile (_stream) if _error != nil { AbortError (_error, "[ac721629] [pprof...] failed starting CPU profile!") } defer pprof.StopCPUProfile () } if _profileMem != "" { log.Printf ("[ii] [9196ee90] [pprof...] profiling MEM to `%s`...\n", _profileMem) _stream, _error := os.Create (_profileMem) if _error != nil { AbortError (_error, "[907d08b5] [pprof...] failed opening MEM profile!") } _profile := pprof.Lookup ("heap") defer func () () { runtime.GC () if _profile != nil { if _error := _profile.WriteTo (_stream, 0); _error != nil { AbortError (_error, "[4b1e5112] [pprof...] failed writing MEM profile!") } } else { AbortError (nil, "[385dc8f0] [pprof...] failed loading MEM profile!") } _stream.Close () } () } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- _server := & server { cdbReader : _cdbReader, cachedReferences : _cachedReferences, cachedDataMeta : _cachedDataMeta, cachedDataContent : _cachedDataContent, hostsDisabled : _hostsDisabled, specialPagesEnabled : _specialPagesEnabled, securityHeadersEnabled : _securityHeadersEnabled, securityHeadersTls : _securityHeadersTls, http1Disabled : _http1Disabled, http2Disabled : _http2Disabled, http3AltSvc : _http3AltSvc, quiet : _quiet, debug : _debug, dummy : _dummy, dummyEmpty : _dummyEmpty, dummyDelay : _dummyDelay, } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- _tls1Config := & tls.Config { Certificates : nil, MinVersion : tls.VersionTLS12, MaxVersion : tls.VersionTLS13, CipherSuites : []uint16 { // NOTE: https://wiki.mozilla.org/Security/Server_Side_TLS#Modern_compatibility // NOTE: https://github.com/golang/go/issues/29349 // NOTE: TLSv1.3 tls.TLS_CHACHA20_POLY1305_SHA256, tls.TLS_AES_128_GCM_SHA256, tls.TLS_AES_256_GCM_SHA384, // NOTE: TLSv1.2 // NOTE: https://datatracker.ietf.org/doc/html/rfc7540#section-9.2.2 tls.TLS_ECDHE_RSA_WITH_CHACHA20_POLY1305_SHA256, // -- ECDHE-RSA-CHACHA20-POLY1305 tls.TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256, // -- ECDHE-RSA-AES128-GCM-SHA256 tls.TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384, // -- ECDHE-RSA-AES256-GCM-SHA384 }, Renegotiation : tls.RenegotiateNever, SessionTicketsDisabled : true, DynamicRecordSizingDisabled : true, NextProtos : []string { "http/1.1", "http/1.0" }, } if (_bindTls1 != "") || (_bindTls2 != "") { if _tlsPrivate != "" { if _certificate, _error := tls.LoadX509KeyPair (_tlsPublic, _tlsPrivate); _error == nil { _tls1Config.Certificates = append (_tls1Config.Certificates, _certificate) } else { AbortError (_error, "[ecdf443d] [tls.....] failed loading TLS certificate!") } } if len (_tls1Config.Certificates) == 0 { var _tlsPublic, _tlsPrivate []byte if _tlsEmbeddedRsa { if !_quiet { log.Printf ("[ii] [46ad23d6] [tls.....] using self-signed RSA certificate!\n") } _tlsPublic = DefaultTlsRsaCertificatePublic _tlsPrivate = DefaultTlsRsaCertificatePrivate } else if _tlsEmbeddedEd { if !_quiet { log.Printf ("[ii] [80c7d1b7] [tls.....] using self-signed Ed25519 certificate!\n") } _tlsPublic = DefaultTlsEd25519CertificatePublic _tlsPrivate = DefaultTlsEd25519CertificatePrivate } else { if !_quiet { log.Printf ("[ii] [344ba198] [tls.....] no TLS certificate specified; using self-signed RSA certificate!\n") } _tlsPublic = DefaultTlsRsaCertificatePublic _tlsPrivate = DefaultTlsRsaCertificatePrivate } if _certificate, _error := tls.X509KeyPair (_tlsPublic, _tlsPrivate); _error == nil { _tls1Config.Certificates = append (_tls1Config.Certificates, _certificate) } else { AbortError (_error, "[98ba6d23] [tls.....] failed parsing TLS certificate!") } } } _httpPlain1Server := & fasthttp.Server { Name : "kawipiko", Handler : _server.ServeFast, GetOnly : true, NoDefaultServerHeader : true, NoDefaultContentType : true, NoDefaultDate : true, DisableHeaderNamesNormalizing : true, Concurrency : 16 * 1024 + 128, MaxRequestsPerConn : 256 * 1024, ReadBufferSize : 16 * 1024, WriteBufferSize : 16 * 1024, MaxRequestBodySize : 16 * 1024, ReadTimeout : 30 * time.Second, WriteTimeout : 30 * time.Second, IdleTimeout : 360 * time.Second, TCPKeepalive : true, TCPKeepalivePeriod : 60 * time.Second, ReduceMemoryUsage : _httpReduceMemory, CloseOnShutdown : true, DisableKeepalive : false, Logger : _server, } _httpTls1Server := & fasthttp.Server {} *_httpTls1Server = *_httpPlain1Server _httpPlain2Server := & http.Server { Handler : _server, MaxHeaderBytes : _httpPlain1Server.ReadBufferSize, ReadTimeout : _httpPlain1Server.ReadTimeout, ReadHeaderTimeout : _httpPlain1Server.ReadTimeout, WriteTimeout : _httpPlain1Server.WriteTimeout, IdleTimeout : _httpPlain1Server.IdleTimeout, } _httpTls2Server := & http.Server {} *_httpTls2Server = *_httpPlain2Server _tls2Config := _tls1Config.Clone () if !_http1Disabled && !_http2Disabled { _tls2Config.NextProtos = []string { "h2", "http/1.1", "http/1.0" } } else if !_http1Disabled { _tls2Config.NextProtos = []string { "http/1.1", "http/1.0" } } else if !_http2Disabled { _tls2Config.NextProtos = []string { "h2" } } else if _bindQuic != "" { // NOP } else { panic ("[1b618ffe]") } if !_quiet { _httpPlain2Server.ErrorLog = log.New (os.Stderr, log.Prefix () + "[ee] [efe8cf82] [go-http.] | ", 0) } else { _httpPlain2Server.ErrorLog = log.New (ioutil.Discard, "", 0) } if !_quiet { _httpTls2Server.ErrorLog = log.New (os.Stderr, log.Prefix () + "[ee] [f734edc4] [go-http.] | ", 0) } else { _httpTls2Server.ErrorLog = log.New (ioutil.Discard, "", 0) } _httpQuicServer := & http3.Server { Handler : _server, TLSConfig : nil, } _tls3Config := _tls1Config.Clone () _tls3Config.NextProtos = []string { "h3", "h3-29" } _httpQuicServer.TLSConfig = _tls3Config _httpQuicServer.QuicConfig = & quic.Config { Versions : []quic.VersionNumber { quic.Version1, quic.VersionDraft29, }, HandshakeIdleTimeout : 6 * time.Second, MaxIdleTimeout : _httpTls1Server.IdleTimeout, MaxIncomingStreams : 1024, MaxIncomingUniStreams : 1024, InitialConnectionReceiveWindow : 1 * 1024 * 1024, MaxConnectionReceiveWindow : 4 * 1024 * 1024, InitialStreamReceiveWindow : 512 * 1024, MaxStreamReceiveWindow : 2 * 1024 * 1024, KeepAlivePeriod : 60 * time.Second, } // FIXME: Control the error logging! //if !_quiet { // _httpQuicServer.ErrorLog = log.New (os.Stderr, log.Prefix () + "[ee] [a6af7354] [quic-h3.] | ", 0) //} else { // _httpQuicServer.ErrorLog = log.New (ioutil.Discard, "", 0) //} if _timeoutDisabled { if !_quiet { log.Printf ("[ii] [5e901455] [http-x..] disabling HTTP timeouts!\n") } _httpPlain1Server.ReadTimeout = 0 _httpPlain1Server.WriteTimeout = 0 _httpPlain1Server.IdleTimeout = 0 _httpTls1Server.ReadTimeout = 0 _httpTls1Server.WriteTimeout = 0 _httpTls1Server.IdleTimeout = 0 _httpPlain2Server.ReadTimeout = 0 _httpPlain2Server.ReadHeaderTimeout = 0 _httpPlain2Server.WriteTimeout = 0 _httpPlain2Server.IdleTimeout = 0 _httpTls2Server.ReadTimeout = 0 _httpTls2Server.ReadHeaderTimeout = 0 _httpTls2Server.WriteTimeout = 0 _httpTls2Server.IdleTimeout = 0 } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- if !_quiet && (_debug || _isFirst) { if _bindPlain1 != "" { log.Printf ("[ii] [f11e4e37] [bind-0a.] listening on `http://%s/` (using FastHTTP supporting HTTP/1.1, HTTP/1.0);\n", _bindPlain1) } if _bindPlain2 != "" { log.Printf ("[ii] [9a05dbb7] [bind-0b.] listening on `http://%s/` (using Go HTTP supporting HTTP/1.1, HTTP/1.0);\n", _bindPlain2) } if _bindTls1 != "" { if !_http1Disabled && (!_http2Disabled && _bindTls2 == "") { log.Printf ("[ii] [21f050c3] [bind-1..] listening on `https://%s/` (using FastHTTP supporting TLS with HTTP/1.1, HTTP/1.0, and HTTP/2 split);\n", _bindTls1) } else if !_http1Disabled { log.Printf ("[ii] [8e41f2df] [bind-1..] listening on `https://%s/` (using FastHTTP supporting TLS with HTTP/1.1, HTTP/1.0);\n", _bindTls1) } else { panic ("[fc754170]") } } if _bindTls2 != "" { if !_http1Disabled && !_http2Disabled { log.Printf ("[ii] [e7f03c99] [bind-2..] listening on `https://%s/` (using Go HTTP supporting TLS with HTTP/2, HTTP/1.1, HTTP/1.0);\n", _bindTls2) } else if !_http1Disabled { log.Printf ("[ii] [477583ad] [bind-2..] listening on `https://%s/` (using Go HTTP supporting TLS with HTTP/1.1, HTTP/1.0 only);\n", _bindTls2) } else if !_http2Disabled { log.Printf ("[ii] [7d2c7ddb] [bind-2..] listening on `https://%s/` (using Go HTTP supporting TLS with HTTP/2 only);\n", _bindTls2) } else { panic ("[d784a82c]") } } if _bindQuic != "" { log.Printf ("[ii] [b958617a] [bind-3..] listening on `https://%s/` (using QUIC supporting TLS with HTTP/3 only);", _bindQuic) } } var _httpPlain1Listener net.Listener if _bindPlain1 != "" { if _listener_0, _error := listenTcp (_bindPlain1); _error == nil { _httpPlain1Listener = _listener_0 } else { AbortError (_error, "[d5f51e9f] [bind-0a.] failed creating TCP listener!") } } var _httpPlain2Listener net.Listener if _bindPlain2 != "" { if _listener_0, _error := listenTcp (_bindPlain2); _error == nil { _httpPlain2Listener = _listener_0 } else { AbortError (_error, "[546075c2] [bind-0b.] failed creating TCP listener!") } } var _httpTls1Listener net.Listener if _bindTls1 != "" { if _listener_0, _error := listenTcp (_bindTls1); _error == nil { _httpTls1Listener = _listener_0 } else { AbortError (_error, "[e35cc693] [bind-1..] failed creating TCP listener!") } } var _httpTls2Listener net.Listener if _bindTls2 != "" { if _listener_0, _error := listenTcp (_bindTls2); _error == nil { _httpTls2Listener = _listener_0 } else { AbortError (_error, "[63567445] [bind-2..] failed creating TCP listener!") } } var _httpQuicListener net.PacketConn if _bindQuic != "" { if _listener_0, _error := listenUdp (_bindQuic); _error == nil { _httpQuicListener = _listener_0 } else { AbortError (_error, "[3b1bfc15] [bind-3..] failed creating UDP listener!") } } var _splitListenerClose func () () if (_httpTls1Listener != nil) && (_httpTls2Listener == nil) && !_http2Disabled { log.Printf ("[ii] [1098a405] [bind-1..] listening on `https://%s/` (using Go HTTP supporting only HTTP/2 split);\n", _bindTls1) _tls1Config.NextProtos = append ([]string { "h2" }, _tls1Config.NextProtos ...) if !_quiet { log.Printf ("[ii] [ba970bbb] [bind-1..] advertising TLS next protocols: %s", _tls1Config.NextProtos) } _tlsListener := tls.NewListener (_httpTls1Listener, _tls1Config) _httpTls1Listener_0 := & splitListener { listener : _tlsListener, queue : make (chan net.Conn), } _httpTls2Listener_0 := & splitListener { listener : _tlsListener, queue : make (chan net.Conn), } _splitListenerClose = func () () { if _error := _tlsListener.Close (); _error != nil { LogError (_error, "[a5bce477] [bind-1..] failed closing TLS listener!") } } go func () () { for { if _connection_0, _error := _tlsListener.Accept (); _error == nil { go func () () { _connection := _connection_0.(*tls.Conn) if _error := _connection.Handshake (); _error != nil { if !_quiet { log.Printf ("[ww] [d1c3dba3] [bind-1..] failed negotiating TLS connection!\n") } } _protocol := _connection.ConnectionState () .NegotiatedProtocol if _protocol == "h2" { if _debug { log.Printf ("[dd] [df9f3e7e] [bind-1..] dispatching HTTP/2 TLS connection!\n") } _httpTls2Listener_0.queue <- _connection } else if (_protocol == "http/1.1") || (_protocol == "http/1.0") || (_protocol == "") { if _debug { log.Printf ("[dd] [d534c361] [bind-1..] dispatching HTTP/1.x TLS connection!\n") } _httpTls1Listener_0.queue <- _connection } else { if !_quiet { log.Printf ("[ww] [5cc0ebde] [bind-1..] unknown TLS protocol `%s`!\n", _protocol) } _connection.Close () } } () } else if strings.Contains (_error.Error (), "use of closed network connection") { break } else { LogError (_error, "[04b6637f] [bind-1..] failed accepting TLS connection!") break } } } () _httpTls1Listener = _httpTls1Listener_0 _httpTls2Listener = _httpTls2Listener_0 } else { if _httpTls1Listener != nil { if !_quiet { log.Printf ("[ii] [ceed854a] [bind-1..] advertising TLS next protocols: %s", _tls1Config.NextProtos) } _httpTls1Listener = tls.NewListener (_httpTls1Listener, _tls1Config) } if _httpTls2Listener != nil { if !_quiet { log.Printf ("[ii] [8b97c977] [bind-2..] advertising TLS next protocols: %s", _tls2Config.NextProtos) } _httpTls2Listener = tls.NewListener (_httpTls2Listener, _tls2Config) } } if _httpQuicListener != nil { if !_quiet { log.Printf ("[ii] [22feb826] [bind-3..] advertising TLS next protocols: %s", _tls3Config.NextProtos) } } if _httpPlain1Listener != nil { _server.httpPlain1Server = _httpPlain1Server } if _httpPlain2Listener != nil { _server.httpPlain2Server = _httpPlain2Server } if _httpTls1Listener != nil { _server.httpTls1Server = _httpTls1Server } if _httpTls2Listener != nil { _server.httpTls2Server = _httpTls2Server } if _httpQuicListener != nil { _server.httpQuicServer = _httpQuicServer } _httpPlain1Server = nil _httpPlain2Server = nil _httpTls1Server = nil _httpTls2Server = nil _httpQuicServer = nil // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- if _seccompEnabled { seccompApplyPhase3 () } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- var _waiter sync.WaitGroup if _server.httpPlain1Server != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [f2061f1b] [fasthttp] starting FastHTTP server...\n") } if _error := _server.httpPlain1Server.Serve (_httpPlain1Listener); _error != nil { AbortError (_error, "[44f45c67] [fasthttp] failed executing server!") } if !_quiet { log.Printf ("[ii] [aca4a14f] [fasthttp] stopped FastHTTP server;\n") } } () } if _server.httpTls1Server != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [83cb1f6f] [fasthttp] starting FastHTTP server (for TLS)...\n") } if _error := _server.httpTls1Server.Serve (_httpTls1Listener); _error != nil { AbortError (_error, "[b2d50852] [fasthttp] failed executing server!") } if !_quiet { log.Printf ("[ii] [ee4180b7] [fasthttp] stopped FastHTTP server (for TLS);\n") } } () } if _server.httpPlain2Server != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [4e579931] [go-http.] starting Go HTTP server...\n") } if _error := _server.httpPlain2Server.Serve (_httpPlain2Listener); (_error != nil) && (_error != http.ErrServerClosed) { AbortError (_error, "[99e9abba] [go-http.] failed executing server!") } if !_quiet { log.Printf ("[ii] [af0e9811] [go-http.] stopped Go HTTP server;\n") } } () } if _server.httpTls2Server != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [46ec2e41] [go-http.] starting Go HTTP server (for TLS)...\n") } if _error := _server.httpTls2Server.Serve (_httpTls2Listener); (_error != nil) && (_error != http.ErrServerClosed) { AbortError (_error, "[9f6d28f4] [go-http.] failed executing server (for TLS)!") } if !_quiet { log.Printf ("[ii] [9a487770] [go-http.] stopped Go HTTP server (for TLS);\n") } } () } if _server.httpQuicServer != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [4cf834b0] [quic-h3.] starting QUIC server...\n") } if _error := _server.httpQuicServer.Serve (_httpQuicListener); (_error != nil) && (_error.Error () != "server closed") { AbortError (_error, "[73e700c5] [quic-h3.] failed executing server!") } if !_quiet { log.Printf ("[ii] [0a9d72e9] [quic-h3.] stopped QUIC server;\n") } } () } if _report && _reportStatsEnabled { _reportStatsQuiet = _quiet go func () () { reportStatsLoop () } () } if !_quiet { time.Sleep (100 * time.Millisecond) log.Printf ("[--] [ ]\n") } { _waiter.Add (1) _signals := make (chan os.Signal, 32) signal.Notify (_signals, syscall.SIGINT, syscall.SIGTERM) go func () () { defer _waiter.Done () <- _signals _reportStatsStop = true if !_quiet { log.Printf ("[--] [ ]\n") } if !_quiet { log.Printf ("[ii] [691cb695] [shutdown] terminating...\n") } if _server.httpPlain1Server != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [8eea3f63] [fasthttp] stopping FastHTTP server...\n") } _server.httpPlain1Server.Shutdown () } () } if _splitListenerClose != nil { _waiter.Add (1) go func () () { defer _waiter.Done () _splitListenerClose () } () } if _server.httpTls1Server != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [ff651007] [fasthttp] stopping FastHTTP server (for TLS)...\n") } _server.httpTls1Server.Shutdown () } () } if _server.httpPlain2Server != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [befe966d] [go-http.] stopping Go HTTP server...\n") } _server.httpPlain2Server.Shutdown (context.TODO ()) } () } if _server.httpTls2Server != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [9ae5a25b] [go-http.] stopping Go HTTP server (for TLS)...\n") } _server.httpTls2Server.Shutdown (context.TODO ()) } () } if _server.httpQuicServer != nil { _waiter.Add (1) go func () () { defer _waiter.Done () if !_quiet { log.Printf ("[ii] [41dab8c2] [quic-h3.] stopping QUIC server...\n") } _server.httpQuicServer.CloseGracefully (1 * time.Second) time.Sleep (1 * time.Second) _server.httpQuicServer.Close () } () } if true { go func () () { time.Sleep (6 * time.Second) AbortError (nil, "[827c672c] [shutdown] forced terminated!") } () } } () } _waiter.Wait () if !_quiet { defer log.Printf ("[ii] [a49175db] [shutdown] terminated!\n") } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- if _cdbReader != nil { _server.cdbReader = nil if _error := _cdbReader.Close (); _error != nil { AbortError (_error, "[a1031c39] [cdb.....] failed closing archive!") } } // -------------------------------------------------------------------------------- // -------------------------------------------------------------------------------- return nil } type splitListener struct { listener net.Listener queue chan net.Conn } func (_listener *splitListener) Accept () (net.Conn, error) { if _connection, _ok := <- _listener.queue; _ok { return _connection, nil } else { return nil, io.EOF } } func (_listener *splitListener) Close () (error) { close (_listener.queue) return nil } func (_listener *splitListener) Addr () (net.Addr) { if _listener.listener != nil { return _listener.listener.Addr () } else { return nil } } var _statsRequestsTotal uint64 var _statsRequestsBody uint64 var _statsRequestsFast uint64 var _statsRequestsSlow uint64 var _statsResponses1xx uint64 var _statsResponses2xx uint64 var _statsResponses3xx uint64 var _statsResponses4xx uint64 var _statsResponses5xx uint64 var _statsUsageCpuTotal uint64 var _statsUsageCpuUser uint64 var _statsUsageCpuSys uint64 var _statsUsageSwitchPreempted uint64 var _statsUsageSwitchVoluntary uint64 var _statsUsageFaultsMajor uint64 var _statsUsageFaultsMinor uint64 var _statsUsageIoReads uint64 var _statsUsageIoWrites uint64 func reportStatsLoop () () { for { reportUpdateUsage () reportUpdateStats () time.Sleep (1000 * time.Millisecond) if _reportStatsStop { break } } } func reportUpdateStats () () { _timestamp := RuntimeNanoseconds () _reportHeartbeatCounter += 1 _shouldLog := false if !_reportStatsQuiet { _shouldLog = _shouldLog || ((_reportHeartbeatCounter % 30) == 0) } else { _shouldLog = _shouldLog || ((_reportHeartbeatCounter % 360) == 0) } _reportHeartbeat.Update (_timestamp) if _reportHeartbeat.Speed1paLast < 0.001 { _reportHeartbeat.Changed = false } _invalid := false _changed := false _reportRequestsTotal.Update2 (_timestamp, &_changed, &_invalid) _reportRequestsBody.Update2 (_timestamp, &_changed, &_invalid) _reportRequestsFast.Update2 (_timestamp, &_changed, &_invalid) _reportRequestsSlow.Update2 (_timestamp, &_changed, &_invalid) _reportResponses1xx.Update2 (_timestamp, &_changed, &_invalid) _reportResponses2xx.Update2 (_timestamp, &_changed, &_invalid) _reportResponses3xx.Update2 (_timestamp, &_changed, &_invalid) _reportResponses4xx.Update2 (_timestamp, &_changed, &_invalid) _reportResponses5xx.Update2 (_timestamp, &_changed, &_invalid) _reportUsageCpuTotal.Update2 (_timestamp, &_changed, &_invalid) _reportUsageCpuUser.Update2 (_timestamp, &_changed, &_invalid) _reportUsageCpuSys.Update2 (_timestamp, &_changed, &_invalid) _reportUsageSwitchPreempted.Update2 (_timestamp, &_changed, &_invalid) _reportUsageSwitchVoluntary.Update2 (_timestamp, &_changed, &_invalid) _reportUsageFaultsMajor.Update2 (_timestamp, &_changed, &_invalid) _reportUsageFaultsMinor.Update2 (_timestamp, &_changed, &_invalid) _reportUsageIoReads.Update2 (_timestamp, &_changed, &_invalid) _reportUsageIoWrites.Update2 (_timestamp, &_changed, &_invalid) if _invalid || (!_shouldLog && !_changed) { return } if _shouldLog && !_reportStatsQuiet { log.Printf ("[--] [ ]\n") } if (_shouldLog || _reportHeartbeat.Changed) && _reportHeartbeat.Touched { log.Printf ("[ii] [addc4553] [stats...] uptime %7.2f h | tps %7.2f %6.1f%%\n", _reportHeartbeat.ValueLast / 3600, _reportHeartbeat.Speed1Last, _reportHeartbeat.Speed1prLast) } if (_shouldLog || _reportRequestsTotal.Changed) && _reportRequestsTotal.Touched { log.Printf ("[ii] [870f4146] [stats...] requests %7.2f M | kps %7.2f %6.1f%% (%+.1f%%)\n", _reportRequestsTotal.ValueLast, _reportRequestsTotal.Speed1Last, _reportRequestsTotal.Speed1prLast, _reportRequestsTotal.Speed1prWindow) } if (_shouldLog || _reportResponses1xx.Changed) && _reportResponses1xx.Touched { log.Printf ("[ii] [d12ebda3] [stats...] resp-1xx %7.2f M | kps %7.2f %6.1f%% (%+.1f%%)\n", _reportResponses1xx.ValueLast, _reportResponses1xx.Speed1Last, _reportResponses1xx.Speed1prLast, _reportResponses1xx.Speed1prWindow) } if (_shouldLog || _reportResponses2xx.Changed) && _reportResponses2xx.Touched { log.Printf ("[ii] [2464e4c2] [stats...] resp-2xx %7.2f M | kps %7.2f %6.1f%% (%+.1f%%)\n", _reportResponses2xx.ValueLast, _reportResponses2xx.Speed1Last, _reportResponses2xx.Speed1prLast, _reportResponses2xx.Speed1prWindow) } if (_shouldLog || _reportResponses3xx.Changed) && _reportResponses3xx.Touched { log.Printf ("[ii] [59bea970] [stats...] resp-3xx %7.2f M | kps %7.2f %6.1f%% (%+.1f%%)\n", _reportResponses3xx.ValueLast, _reportResponses3xx.Speed1Last, _reportResponses3xx.Speed1prLast, _reportResponses3xx.Speed1prWindow) } if (_shouldLog || _reportResponses4xx.Changed) && _reportResponses4xx.Touched { log.Printf ("[ii] [babb043c] [stats...] resp-4xx %7.2f M | kps %7.2f %6.1f%% (%+.1f%%)\n", _reportResponses4xx.ValueLast, _reportResponses4xx.Speed1Last, _reportResponses4xx.Speed1prLast, _reportResponses4xx.Speed1prWindow) } if (_shouldLog || _reportResponses5xx.Changed) && _reportResponses5xx.Touched { log.Printf ("[ii] [047ba05b] [stats...] resp-5xx %7.2f M | kps %7.2f %6.1f%% (%+.1f%%)\n", _reportResponses5xx.ValueLast, _reportResponses5xx.Speed1Last, _reportResponses5xx.Speed1prLast, _reportResponses5xx.Speed1prWindow) } if (_shouldLog || _reportRequestsBody.Changed) && _reportRequestsBody.Touched { log.Printf ("[ii] [d2d45f12] [stats...] resp-sz %7.2f GB | MBps %7.2f %6.1f%% (%+.1f%%)\n", _reportRequestsBody.ValueLast, _reportRequestsBody.Speed1Last, _reportRequestsBody.Speed1prLast, _reportRequestsBody.Speed1prWindow) } if (_shouldLog || _reportUsageCpuTotal.Changed) && _reportUsageCpuTotal.Touched { log.Printf ("[ii] [e27def0f] [stats...] cpu-all %7.2f h | load %7.2f %6.1f%% (%+.1f%%)\n", _reportUsageCpuTotal.ValueLast / 3600, _reportUsageCpuTotal.Speed1Last, _reportUsageCpuTotal.Speed1prLast, _reportUsageCpuTotal.Speed1prWindow) } if (_shouldLog || _reportUsageCpuSys.Changed) && _reportUsageCpuSys.Touched { log.Printf ("[ii] [4a13138d] [stats...] cpu-sys %7.2f h | load %7.2f %6.1f%% (%+.1f%%)\n", _reportUsageCpuSys.ValueLast / 3600, _reportUsageCpuSys.Speed1Last, _reportUsageCpuSys.Speed1prLast, _reportUsageCpuSys.Speed1prWindow) } if (_shouldLog || _reportUsageSwitchPreempted.Changed) && _reportUsageSwitchPreempted.Touched { log.Printf ("[ii] [9e3fe2a8] [stats...] csw-prem %7.2f M | kps %7.2f %6.1f%% (%+.1f%%)\n", _reportUsageSwitchPreempted.ValueLast, _reportUsageSwitchPreempted.Speed1Last, _reportUsageSwitchPreempted.Speed1prLast, _reportUsageSwitchPreempted.Speed1prWindow) } if (_shouldLog || _reportUsageSwitchVoluntary.Changed) && _reportUsageSwitchVoluntary.Touched { log.Printf ("[ii] [e8059c56] [stats...] csw-norm %7.2f M | kps %7.2f %6.1f%% (%+.1f%%)\n", _reportUsageSwitchVoluntary.ValueLast, _reportUsageSwitchVoluntary.Speed1Last, _reportUsageSwitchVoluntary.Speed1prLast, _reportUsageSwitchVoluntary.Speed1prWindow) } if (_shouldLog || _reportUsageFaultsMajor.Changed) && _reportUsageFaultsMajor.Touched { log.Printf ("[ii] [08adfe80] [stats...] mem-maj %7.2f GB | MBps %7.2f %6.1f%% (%+.1f%%)\n", _reportUsageFaultsMajor.ValueLast, _reportUsageFaultsMajor.Speed1Last / 1024, _reportUsageFaultsMajor.Speed1prLast, _reportUsageFaultsMajor.Speed1prWindow) } if (_shouldLog || _reportUsageFaultsMinor.Changed) && _reportUsageFaultsMinor.Touched { log.Printf ("[ii] [6681ca7b] [stats...] mem-min %7.2f GB | MBps %7.2f %6.1f%% (%+.1f%%)\n", _reportUsageFaultsMinor.ValueLast, _reportUsageFaultsMinor.Speed1Last / 1024, _reportUsageFaultsMinor.Speed1prLast, _reportUsageFaultsMinor.Speed1prWindow) } if (_shouldLog || _reportUsageIoReads.Changed) && _reportUsageIoReads.Touched { log.Printf ("[ii] [db937bbe] [stats...] io-read %7.2f GB | MBps %7.2f %6.1f%% (%+.1f%%)\n", _reportUsageIoReads.ValueLast, _reportUsageIoReads.Speed1Last / 1024, _reportUsageIoReads.Speed1prLast, _reportUsageIoReads.Speed1prWindow) } if (_shouldLog || _reportUsageIoWrites.Changed) && _reportUsageIoWrites.Touched { log.Printf ("[ii] [ea16e474] [stats...] io-write %7.2f GB | MBps %7.2f %6.1f%% (%+.1f%%)\n", _reportUsageIoWrites.ValueLast, _reportUsageIoWrites.Speed1Last / 1024, _reportUsageIoWrites.Speed1prLast, _reportUsageIoWrites.Speed1prWindow) } if _shouldLog && !_reportStatsQuiet { log.Printf ("[--] [ ]\n") } } func reportUpdateUsage () () { var _usage syscall.Rusage if _error := syscall.Getrusage (syscall.RUSAGE_SELF, &_usage); _error != nil { LogError (_error, "[c1d79147] [stats...] failed getting usage!") return } _pageSize := uint64 (syscall.Getpagesize ()) _ioSize := uint64 (512) _statsUsageCpuUser = (uint64 (_usage.Utime.Sec) * 1000 * 1000) + uint64 (_usage.Utime.Usec) _statsUsageCpuSys = (uint64 (_usage.Stime.Sec) * 1000 * 1000) + uint64 (_usage.Stime.Usec) _statsUsageCpuTotal = _statsUsageCpuUser + _statsUsageCpuSys _statsUsageSwitchVoluntary = uint64 (_usage.Nvcsw) _statsUsageSwitchPreempted = uint64 (_usage.Nivcsw) _statsUsageFaultsMajor = uint64 (_usage.Majflt) * _pageSize _statsUsageFaultsMinor = uint64 (_usage.Minflt) * _pageSize _statsUsageIoReads = uint64 (_usage.Inblock) * _ioSize _statsUsageIoWrites = uint64 (_usage.Oublock) * _ioSize } var _reportStatsEnabled = true var _reportStatsQuiet = false var _reportStatsStop = false var _reportHeartbeatCounter uint64 var _reportHeartbeat = & StatMetric { MetricSource : &_reportHeartbeatCounter, } var _reportRequestsTotal = & StatMetric { MetricSource : &_statsRequestsTotal, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 0.1, } var _reportRequestsBody = & StatMetric { MetricSource : &_statsRequestsBody, ValueScale : 1024 * 1024 * 1024, // GiB SpeedScale : 1024, // KiB SpeedThreshold : 1024, } var _reportRequestsFast = & StatMetric { MetricSource : &_statsRequestsFast, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 1.0, } var _reportRequestsSlow = & StatMetric { MetricSource : &_statsRequestsSlow, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 1.0, } var _reportResponses1xx = & StatMetric { MetricSource : &_statsResponses1xx, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 0.01, } var _reportResponses2xx = & StatMetric { MetricSource : &_statsResponses2xx, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 8.0, } var _reportResponses3xx = & StatMetric { MetricSource : &_statsResponses3xx, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 0.01, } var _reportResponses4xx = & StatMetric { MetricSource : &_statsResponses4xx, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 0.01, } var _reportResponses5xx = & StatMetric { MetricSource : &_statsResponses5xx, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 0.01, } var _reportUsageCpuTotal = & StatMetric { MetricSource : &_statsUsageCpuTotal, ValueScale : 1000 * 1000, // microseconds -> seconds SpeedScale : 1000 * 1000, // microseconds -> seconds SpeedThreshold : 0.50, // -> percent } var _reportUsageCpuUser = & StatMetric { MetricSource : &_statsUsageCpuUser, ValueScale : 1000 * 1000, // microseconds -> seconds SpeedScale : 1000 * 1000, // microseconds -> seconds SpeedThreshold : 0.10, // -> percent } var _reportUsageCpuSys = & StatMetric { MetricSource : &_statsUsageCpuSys, ValueScale : 1000 * 1000, // microseconds -> seconds SpeedScale : 1000 * 1000, // microseconds -> seconds SpeedThreshold : 0.25, // -> percent } var _reportUsageSwitchPreempted = & StatMetric { MetricSource : &_statsUsageSwitchPreempted, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 0.1, } var _reportUsageSwitchVoluntary = & StatMetric { MetricSource : &_statsUsageSwitchVoluntary, ValueScale : 1000 * 1000, SpeedScale : 1000, SpeedThreshold : 10, } var _reportUsageFaultsMajor = & StatMetric { MetricSource : &_statsUsageFaultsMajor, ValueScale : 1024 * 1024 * 1024, // GiB SpeedScale : 1024, // KiB SpeedThreshold : 1024, } var _reportUsageFaultsMinor = & StatMetric { MetricSource : &_statsUsageFaultsMinor, ValueScale : 1024 * 1024 * 1024, // GiB SpeedScale : 1024, // KiB SpeedThreshold : 1024, } var _reportUsageIoReads = & StatMetric { MetricSource : &_statsUsageIoReads, ValueScale : 1024 * 1024 * 1024, // GiB SpeedScale : 1024, // KiB SpeedThreshold : 1024, } var _reportUsageIoWrites = & StatMetric { MetricSource : &_statsUsageIoWrites, ValueScale : 1024 * 1024 * 1024, // GiB SpeedScale : 1024, // KiB SpeedThreshold : 16, } //go:embed usage.txt var usageText string //go:embed manual.txt var manualText string //go:embed manual.html var manualHtml string func init () { usageText = strings.ReplaceAll (usageText, "@{SCHEMA}", CurrentSchemaVersion) }