1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
package gemini
import (
"bufio"
"crypto/tls"
"errors"
"io/ioutil"
"net/url"
"strconv"
"strings"
)
var (
ErrProtocol = errors.New("Protocol error")
)
// Client is a Gemini client.
type Client struct{}
// Request makes a request for the provided URL. The host is inferred from the URL.
func (c *Client) Request(url string) (*Response, error) {
req, err := NewRequest(url)
if err != nil {
return nil, err
}
return c.Do(req)
}
// ProxyRequest requests the provided URL from the provided host.
func (c *Client) ProxyRequest(host, url string) (*Response, error) {
req, err := NewProxyRequest(host, url)
if err != nil {
return nil, err
}
return c.Do(req)
}
// Request is a Gemini request.
//
// A Request can optionally be configured with a client certificate. Example:
//
// req := NewRequest(url)
// cert, err := tls.LoadX509KeyPair("client.crt", "client.key")
// if err != nil {
// panic(err)
// }
// req.Certificates = append(req.Certificates, cert)
//
type Request struct {
Host string // host or host:port
URL *url.URL // the requested URL
Certificates []tls.Certificate // client certificates
}
// NewRequest returns a new request. The host is inferred from the provided url.
func NewRequest(rawurl string) (*Request, error) {
u, err := url.Parse(rawurl)
if err != nil {
return nil, err
}
// Ignore UserInfo if present
u.User = nil
return &Request{
Host: u.Host,
URL: u,
}, nil
}
// NewProxyRequest makes a new request using the provided host.
func NewProxyRequest(host, rawurl string) (*Request, error) {
u, err := url.Parse(rawurl)
if err != nil {
return nil, err
}
// Ignore UserInfo if present
u.User = nil
return &Request{
Host: host,
URL: u,
}, nil
}
// Do sends a Gemini request and returns a Gemini response.
func (c *Client) Do(req *Request) (*Response, error) {
host := req.Host
if strings.LastIndex(host, ":") == -1 {
// The default port is 1965
host += ":1965"
}
config := tls.Config{}
// Allow self signed certificates
config.InsecureSkipVerify = true
config.Certificates = req.Certificates
conn, err := tls.Dial("tcp", host, &config)
if err != nil {
return nil, err
}
defer conn.Close()
// Write the request
request := req.URL.String() + "\r\n"
if _, err := conn.Write([]byte(request)); err != nil {
return nil, err
}
buf := bufio.NewReader(conn)
// Read the response header
code := make([]byte, 2)
if _, err := buf.Read(code); err != nil {
return nil, err
}
status, err := strconv.Atoi(string(code))
if err != nil {
return nil, err
}
// Read one space
if space, err := buf.ReadByte(); err != nil {
return nil, err
} else if space != ' ' {
return nil, ErrProtocol
}
// Read the meta
meta, err := readLine(buf)
if err != nil {
return nil, err
}
// Read the response body
body, err := ioutil.ReadAll(buf)
if err != nil {
return nil, err
}
return &Response{
Status: status,
Meta: meta,
Body: body,
}, nil
}
|