aboutsummaryrefslogtreecommitdiff
path: root/challange_http.go
blob: 7b2de54bd9dbe924cc2df7d591f45b9e1f8084d1 (plain)
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
package acme

import (
	"io"
	"io/ioutil"
	"net"
	"net/http"
	"os"
	"path"
	"sync"
)

const wellKnown = `/.well-known/acme-challenge/`

func init() {
	registerChallenge(ChallengeHTTP)
}

type httpChallenge struct {
	Challenge
	Addr string
	wg   sync.WaitGroup
}

func (c *httpChallenge) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	if r.URL.Path == path.Join(wellKnown, c.Token) {
		io.WriteString(w, c.KeyAuthorization)
		c.wg.Done()
	}
}

func (c *httpChallenge) Solve() error {
	l, err := net.Listen("tcp", c.Addr)
	if err != nil {
		return err
	}
	defer l.Close()
	c.wg.Add(1)
	go http.Serve(l, c)
	c.wg.Wait()
	return nil
}

func (c *httpChallenge) Abort() error {
	c.wg.Done()
	return nil
}

type webRoot struct {
	Challenge
	Webroot string
}

func (c *webRoot) Solve() error {
	file := path.Join(c.Webroot, wellKnown, c.Token)
	if err := os.MkdirAll(path.Dir(file), 0755); err != nil {
		return err
	}
	return ioutil.WriteFile(file, []byte(c.KeyAuthorization), 0644)
}

func (c *webRoot) Abort() error {
	file := path.Join(c.Webroot, wellKnown, c.Token)
	return os.Remove(file)
}