// Copyright 2017 fatedier, fatedier@gmail.com // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package vhost import ( "bytes" "io" "net/http" "os" frpLog "github.com/fatedier/frp/pkg/util/log" "github.com/fatedier/frp/pkg/util/version" ) var NotFoundPagePath = "" const ( NotFound = `
Sorry, the page you are looking for is currently unavailable.
Please try again later.
The server is powered by frp.
Faithfully yours, frp.
` ) func getNotFoundPageContent() []byte { var ( buf []byte err error ) if NotFoundPagePath != "" { buf, err = os.ReadFile(NotFoundPagePath) if err != nil { frpLog.Warn("read custom 404 page error: %v", err) buf = []byte(NotFound) } } else { buf = []byte(NotFound) } return buf } func notFoundResponse() *http.Response { header := make(http.Header) header.Set("server", "frp/"+version.Full()) header.Set("Content-Type", "text/html") content := getNotFoundPageContent() res := &http.Response{ Status: "Not Found", StatusCode: 404, Proto: "HTTP/1.1", ProtoMajor: 1, ProtoMinor: 1, Header: header, Body: io.NopCloser(bytes.NewReader(content)), ContentLength: int64(len(content)), } return res } func noAuthResponse() *http.Response { header := make(map[string][]string) header["WWW-Authenticate"] = []string{`Basic realm="Restricted"`} res := &http.Response{ Status: "401 Not authorized", StatusCode: 401, Proto: "HTTP/1.1", ProtoMajor: 1, ProtoMinor: 1, Header: header, } return res }