summaryrefslogtreecommitdiff
path: root/integration/plugin/logging/cmd/close_on_start/main.go
blob: 66a9ae257d54d1b79e83723323744634e3d6415b (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
package main

import (
	"encoding/json"
	"fmt"
	"net"
	"net/http"
	"os"
	"time"
)

type start struct {
	File string
}

func main() {
	l, err := net.Listen("unix", "/run/docker/plugins/plugin.sock")
	if err != nil {
		panic(err)
	}

	mux := http.NewServeMux()
	mux.HandleFunc("/LogDriver.StartLogging", func(w http.ResponseWriter, req *http.Request) {
		startReq := &start{}
		if err := json.NewDecoder(req.Body).Decode(startReq); err != nil {
			http.Error(w, err.Error(), http.StatusBadRequest)
			return
		}

		f, err := os.OpenFile(startReq.File, os.O_RDONLY, 0600)
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}

		// Close the file immediately, this allows us to test what happens in the daemon when the plugin has closed the
		// file or, for example, the plugin has crashed.
		f.Close()

		w.WriteHeader(http.StatusOK)
		fmt.Fprintln(w, `{}`)
	})
	server := http.Server{
		Addr:              l.Addr().String(),
		Handler:           mux,
		ReadHeaderTimeout: 2 * time.Second, // This server is not for production code; picked an arbitrary timeout to statisfy gosec (G112: Potential Slowloris Attack)
	}

	server.Serve(l)
}