GitVita
New repositoryImport repositoryNew issue
Your repositories Your dashboard Sign in

cloudflare/cloudflared

Public

mirrored from https://github.com/cloudflare/cloudflaredAvailable

Watch0Fork0Star0
CodeCommitsIssuesPull requestsActionsInsightsSecurity
2020.2.1

Branches

Tags

  • No tags available.
0Branches0Tags
Go to file
Add file
Code

Clone

HTTPS

Download ZIP

cloudflared/hello

hello/hello.go

196lines · modecode

RawDownload
Latest commit unavailable.
1package hello
2
3import (
4 "bytes"
5 "crypto/tls"
6 "encoding/json"
7 "fmt"
8 "html/template"
9 "io/ioutil"
10 "net"
11 "net/http"
12 "os"
13 "time"
14
15 "github.com/gorilla/websocket"
16 "github.com/sirupsen/logrus"
17
18 "github.com/cloudflare/cloudflared/tlsconfig"
19)
20
21type templateData struct {
22 ServerName string
23 Request *http.Request
24 Body string
25}
26
27type OriginUpTime struct {
28 StartTime time.Time `json:"startTime"`
29 UpTime string `json:"uptime"`
30}
31
32const defaultServerName = "the Argo Tunnel test server"
33const indexTemplate = `
34<!DOCTYPE html>
35<html lang="en">
36 <head>
37 <meta charset="utf-8">
38 <meta http-equiv="X-UA-Compatible" content="IE=Edge">
39 <title>
40 Argo Tunnel Connection
41 </title>
42 <meta name="author" content="">
43 <meta name="description" content="Argo Tunnel Connection">
44 <meta name="viewport" content="width=device-width, initial-scale=1">
45 <style>
46 html{line-height:1.15;-ms-text-size-adjust:100%;-webkit-text-size-adjust:100%}body{margin:0}section{display:block}h1{font-size:2em;margin:.67em 0}a{background-color:transparent;-webkit-text-decoration-skip:objects}/* 1 */::-webkit-file-upload-button{-webkit-appearance:button;font:inherit}/* 1 */a,body,dd,div,dl,dt,h1,h4,html,p,section{box-sizing:border-box}.bt{border-top-style:solid;border-top-width:1px}.bl{border-left-style:solid;border-left-width:1px}.b--orange{border-color:#f38020}.br1{border-radius:.125rem}.bw2{border-width:.25rem}.dib{display:inline-block}.sans-serif{font-family:open sans,-apple-system,BlinkMacSystemFont,avenir next,avenir,helvetica neue,helvetica,ubuntu,roboto,noto,segoe ui,arial,sans-serif}.overflow-x-auto{overflow-x:auto}.code{font-family:Consolas,monaco,monospace}.b{font-weight:700}.fw3{font-weight:300}.fw4{font-weight:400}.fw5{font-weight:500}.fw6{font-weight:600}.lh-copy{line-height:1.5}.link{text-decoration:none}.link,.link:active,.link:focus,.link:hover,.link:link,.link:visited{transition:color .15s ease-in}.link:focus{outline:1px dotted currentColor}.mw-100{max-width:100%}.mw4{max-width:8rem}.mw7{max-width:48rem}.bg-light-gray{background-color:#f7f7f7}.link-hover:hover{background-color:#1f679e}.white{color:#fff}.bg-white{background-color:#fff}.bg-blue{background-color:#408bc9}.pb2{padding-bottom:.5rem}.pb6{padding-bottom:8rem}.pt3{padding-top:1rem}.pt5{padding-top:4rem}.pv2{padding-top:.5rem;padding-bottom:.5rem}.ph3{padding-left:1rem;padding-right:1rem}.ph4{padding-left:2rem;padding-right:2rem}.ml0{margin-left:0}.mb1{margin-bottom:.25rem}.mb2{margin-bottom:.5rem}.mb3{margin-bottom:1rem}.mt5{margin-top:4rem}.ttu{text-transform:uppercase}.f4{font-size:1.25rem}.f5{font-size:1rem}.f6{font-size:.875rem}.f7{font-size:.75rem}.measure{max-width:30em}.center{margin-left:auto}.center{margin-right:auto}@media screen and (min-width:30em){.f2-ns{font-size:2.25rem}}@media screen and (min-width:30em) and (max-width:60em){.f5-m{font-size:1rem}}@media screen and (min-width:60em){.f4-l{font-size:1.25rem}}
47 .st0{fill:#FFF}.st1{fill:#f48120}.st2{fill:#faad3f}.st3{fill:#404041}
48 </style>
49 </head>
50 <body class="sans-serif black">
51 <div class="bt bw2 b--orange bg-white pb6">
52 <div class="mw7 center ph4 pt3">
53 <svg id="Layer_2" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 109 40.5" class="mw4">
54 <path class="st0" d="M98.6 14.2L93 12.9l-1-.4-25.7.2v12.4l32.3.1z"/>
55 <path class="st1" d="M88.1 24c.3-1 .2-2-.3-2.6-.5-.6-1.2-1-2.1-1.1l-17.4-.2c-.1 0-.2-.1-.3-.1-.1-.1-.1-.2 0-.3.1-.2.2-.3.4-.3l17.5-.2c2.1-.1 4.3-1.8 5.1-3.8l1-2.6c0-.1.1-.2 0-.3-1.1-5.1-5.7-8.9-11.1-8.9-5 0-9.3 3.2-10.8 7.7-1-.7-2.2-1.1-3.6-1-2.4.2-4.3 2.2-4.6 4.6-.1.6 0 1.2.1 1.8-3.9.1-7.1 3.3-7.1 7.3 0 .4 0 .7.1 1.1 0 .2.2.3.3.3h32.1c.2 0 .4-.1.4-.3l.3-1.1z"/>
56 <path class="st2" d="M93.6 12.8h-.5c-.1 0-.2.1-.3.2l-.7 2.4c-.3 1-.2 2 .3 2.6.5.6 1.2 1 2.1 1.1l3.7.2c.1 0 .2.1.3.1.1.1.1.2 0 .3-.1.2-.2.3-.4.3l-3.8.2c-2.1.1-4.3 1.8-5.1 3.8l-.2.9c-.1.1 0 .3.2.3h13.2c.2 0 .3-.1.3-.3.2-.8.4-1.7.4-2.6 0-5.2-4.3-9.5-9.5-9.5"/>
57 <path class="st3" d="M104.4 30.8c-.5 0-.9-.4-.9-.9s.4-.9.9-.9.9.4.9.9-.4.9-.9.9m0-1.6c-.4 0-.7.3-.7.7 0 .4.3.7.7.7.4 0 .7-.3.7-.7 0-.4-.3-.7-.7-.7m.4 1.2h-.2l-.2-.3h-.2v.3h-.2v-.9h.5c.2 0 .3.1.3.3 0 .1-.1.2-.2.3l.2.3zm-.3-.5c.1 0 .1 0 .1-.1s-.1-.1-.1-.1h-.3v.3h.3zM14.8 29H17v6h3.8v1.9h-6zM23.1 32.9c0-2.3 1.8-4.1 4.3-4.1s4.2 1.8 4.2 4.1-1.8 4.1-4.3 4.1c-2.4 0-4.2-1.8-4.2-4.1m6.3 0c0-1.2-.8-2.2-2-2.2s-2 1-2 2.1.8 2.1 2 2.1c1.2.2 2-.8 2-2M34.3 33.4V29h2.2v4.4c0 1.1.6 1.7 1.5 1.7s1.5-.5 1.5-1.6V29h2.2v4.4c0 2.6-1.5 3.7-3.7 3.7-2.3-.1-3.7-1.2-3.7-3.7M45 29h3.1c2.8 0 4.5 1.6 4.5 3.9s-1.7 4-4.5 4h-3V29zm3.1 5.9c1.3 0 2.2-.7 2.2-2s-.9-2-2.2-2h-.9v4h.9zM55.7 29H62v1.9h-4.1v1.3h3.7V34h-3.7v2.9h-2.2zM65.1 29h2.2v6h3.8v1.9h-6zM76.8 28.9H79l3.4 8H80l-.6-1.4h-3.1l-.6 1.4h-2.3l3.4-8zm2 4.9l-.9-2.2-.9 2.2h1.8zM85.2 29h3.7c1.2 0 2 .3 2.6.9.5.5.7 1.1.7 1.8 0 1.2-.6 2-1.6 2.4l1.9 2.8H90l-1.6-2.4h-1v2.4h-2.2V29zm3.6 3.8c.7 0 1.2-.4 1.2-.9 0-.6-.5-.9-1.2-.9h-1.4v1.9h1.4zM95.3 29h6.4v1.8h-4.2V32h3.8v1.8h-3.8V35h4.3v1.9h-6.5zM10 33.9c-.3.7-1 1.2-1.8 1.2-1.2 0-2-1-2-2.1s.8-2.1 2-2.1c.9 0 1.6.6 1.9 1.3h2.3c-.4-1.9-2-3.3-4.2-3.3-2.4 0-4.3 1.8-4.3 4.1s1.8 4.1 4.2 4.1c2.1 0 3.7-1.4 4.2-3.2H10z"/>
58 </svg>
59 <h1 class="f4 f2-ns mt5 fw5">Congrats! You created your first tunnel!</h1>
60 <p class="f6 f5-m f4-l measure lh-copy fw3">
61 Argo Tunnel exposes locally running applications to the internet by
62 running an encrypted, virtual tunnel from your laptop or server to
63 Cloudflare's edge network.
64 </p>
65 <p class="b f5 mt5 fw6">Ready for the next step?</p>
66 <a
67 class="fw6 link white bg-blue ph4 pv2 br1 dib f5 link-hover"
68 style="border-bottom: 1px solid #1f679e"
69 href="https://developers.cloudflare.com/argo-tunnel/">
70 Get started here
71 </a>
72 <section>
73 <h4 class="f6 fw4 pt5 mb2">Request</h4>
74 <dl class="bl bw2 b--orange ph3 pt3 pb2 bg-light-gray f7 code overflow-x-auto mw-100">
75 <dd class="ml0 mb3 f5">Method: {{.Request.Method}}</dd>
76 <dd class="ml0 mb3 f5">Protocol: {{.Request.Proto}}</dd>
77 <dd class="ml0 mb3 f5">Request URL: {{.Request.URL}}</dd>
78 <dd class="ml0 mb3 f5">Transfer encoding: {{.Request.TransferEncoding}}</dd>
79 <dd class="ml0 mb3 f5">Host: {{.Request.Host}}</dd>
80 <dd class="ml0 mb3 f5">Remote address: {{.Request.RemoteAddr}}</dd>
81 <dd class="ml0 mb3 f5">Request URI: {{.Request.RequestURI}}</dd>
82{{range $key, $value := .Request.Header}}
83 <dd class="ml0 mb3 f5">Header: {{$key}}, Value: {{$value}}</dd>
84{{end}}
85 <dd class="ml0 mb3 f5">Body: {{.Body}}</dd>
86 </dl>
87 </section>
88 </div>
89 </div>
90 </body>
91</html>
92`
93
94func StartHelloWorldServer(logger *logrus.Logger, listener net.Listener, shutdownC <-chan struct{}) error {
95 logger.Infof("Starting Hello World server at %s", listener.Addr())
96 serverName := defaultServerName
97 if hostname, err := os.Hostname(); err == nil {
98 serverName = hostname
99 }
100
101 upgrader := websocket.Upgrader{
102 ReadBufferSize: 1024,
103 WriteBufferSize: 1024,
104 }
105
106 muxer := http.NewServeMux()
107 muxer.HandleFunc("/uptime", uptimeHandler(time.Now()))
108 muxer.HandleFunc("/ws", websocketHandler(logger, upgrader))
109 muxer.HandleFunc("/", rootHandler(serverName))
110 httpServer := &http.Server{Addr: listener.Addr().String(), Handler: muxer}
111 go func() {
112 <-shutdownC
113 httpServer.Close()
114 }()
115
116 err := httpServer.Serve(listener)
117 return err
118}
119
120func CreateTLSListener(address string) (net.Listener, error) {
121 certificate, err := tlsconfig.GetHelloCertificate()
122 if err != nil {
123 return nil, err
124 }
125
126 // If the port in address is empty, a port number is automatically chosen
127 listener, err := tls.Listen(
128 "tcp",
129 address,
130 &tls.Config{Certificates: []tls.Certificate{certificate}})
131
132 return listener, err
133}
134
135func uptimeHandler(startTime time.Time) http.HandlerFunc {
136 return func(w http.ResponseWriter, r *http.Request) {
137 // Note that if autoupdate is enabled, the uptime is reset when a new client
138 // release is available
139 resp := &OriginUpTime{StartTime: startTime, UpTime: time.Now().Sub(startTime).String()}
140 respJson, err := json.Marshal(resp)
141 if err != nil {
142 w.WriteHeader(http.StatusInternalServerError)
143 } else {
144 w.Header().Set("Content-Type", "application/json")
145 w.Write(respJson)
146 }
147 }
148}
149
150// This handler will echo message
151func websocketHandler(logger *logrus.Logger, upgrader websocket.Upgrader) http.HandlerFunc {
152 return func(w http.ResponseWriter, r *http.Request) {
153 conn, err := upgrader.Upgrade(w, r, nil)
154 if err != nil {
155 return
156 }
157 defer conn.Close()
158 for {
159 mt, message, err := conn.ReadMessage()
160 if err != nil {
161 logger.WithError(err).Error("websocket read message error")
162 break
163 }
164
165 if err := conn.WriteMessage(mt, message); err != nil {
166 logger.WithError(err).Error("websocket write message error")
167 break
168 }
169 }
170 }
171}
172
173func rootHandler(serverName string) http.HandlerFunc {
174 responseTemplate := template.Must(template.New("index").Parse(indexTemplate))
175 return func(w http.ResponseWriter, r *http.Request) {
176 var buffer bytes.Buffer
177 var body string
178 rawBody, err := ioutil.ReadAll(r.Body)
179 if err == nil {
180 body = string(rawBody)
181 } else {
182 body = ""
183 }
184 err = responseTemplate.Execute(&buffer, &templateData{
185 ServerName: serverName,
186 Request: r,
187 Body: body,
188 })
189 if err != nil {
190 w.WriteHeader(http.StatusInternalServerError)
191 fmt.Fprintf(w, "error: %v", err)
192 } else {
193 buffer.WriteTo(w)
194 }
195 }
196}
197