123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687 |
- package vhost
- import (
- "bytes"
- "io"
- "net/http"
- "os"
- "github.com/fatedier/frp/pkg/util/log"
- "github.com/fatedier/frp/pkg/util/version"
- )
- var NotFoundPagePath = ""
- const (
- NotFound = `<!DOCTYPE html>
- <html>
- <head>
- <title>Not Found</title>
- <style>
- body {
- width: 35em;
- margin: 0 auto;
- font-family: Tahoma, Verdana, Arial, sans-serif;
- }
- </style>
- </head>
- <body>
- <h1>The page you requested was not found.</h1>
- <p>Sorry, the page you are looking for is currently unavailable.<br/>
- Please try again later.</p>
- <p>The server is powered by <a href="https://github.com/fatedier/frp">frp</a>.</p>
- <p><em>Faithfully yours, frp.</em></p>
- </body>
- </html>
- `
- )
- func getNotFoundPageContent() []byte {
- var (
- buf []byte
- err error
- )
- if NotFoundPagePath != "" {
- buf, err = os.ReadFile(NotFoundPagePath)
- if err != nil {
- log.Warnf("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
- }
|