api.go 7.79 KB
Newer Older
1
package core
2
3
4

import (
	"encoding/json"
5
	"io/ioutil"
6
	"net/http"
7
	"path/filepath"
8
	"regexp"
9
	"strings"
10
11
12

	"github.com/Sirupsen/logrus"
	"github.com/go-errors/errors"
13
14
	"github.com/privacybydesign/gabi"
	"github.com/privacybydesign/gabi/big"
15
	"github.com/privacybydesign/irmago"
Sietse Ringers's avatar
Sietse Ringers committed
16
	"github.com/privacybydesign/irmago/server"
17
18
)

Sietse Ringers's avatar
Sietse Ringers committed
19
func Initialize(configuration *server.Configuration) error {
20
21
22
23
24
25
26
	conf = configuration

	if conf.Logger == nil {
		conf.Logger = logrus.New()
		conf.Logger.Level = logrus.DebugLevel
		conf.Logger.Formatter = &logrus.TextFormatter{}
	}
Sietse Ringers's avatar
Sietse Ringers committed
27
	server.Logger = conf.Logger
Sietse Ringers's avatar
Sietse Ringers committed
28
	irma.Logger = conf.Logger
29
30
31

	if conf.IrmaConfiguration == nil {
		var err error
32
		var path, assets string
33
		if conf.CachePath == "" {
34
35
36
37
38
			path = conf.IrmaConfigurationPath
			assets = ""
		} else {
			path = filepath.Join(conf.CachePath, "irma_configuration")
			assets = conf.IrmaConfigurationPath
39
		}
40
		conf.IrmaConfiguration, err = irma.NewConfiguration(path, assets)
41
		if err != nil {
42
43
			return err
		}
44
45

		if conf.IrmaConfigurationPath == "" {
46
47
48
			if err := conf.IrmaConfiguration.DownloadDefaultSchemes(); err != nil {
				return err
			}
49
50
51
52
		} else {
			if err = conf.IrmaConfiguration.ParseFolder(); err != nil {
				return err
			}
53
		}
54
55
	}

56
57
58
59
	if len(conf.IrmaConfiguration.SchemeManagers) == 0 {
		return errors.New("no schemes found in irma_configuration folder " + conf.IrmaConfiguration.Path)
	}

60
61
	if conf.IssuerPrivateKeys == nil {
		conf.IssuerPrivateKeys = make(map[irma.IssuerIdentifier]*gabi.PrivateKey)
62
	}
63
64
	if conf.IssuerPrivateKeysPath != "" {
		files, err := ioutil.ReadDir(conf.IssuerPrivateKeysPath)
65
66
67
68
69
70
71
72
73
		if err != nil {
			return err
		}
		for _, file := range files {
			filename := file.Name()
			issid := irma.NewIssuerIdentifier(strings.TrimSuffix(filename, filepath.Ext(filename))) // strip .xml
			if _, ok := conf.IrmaConfiguration.Issuers[issid]; !ok {
				return errors.Errorf("Private key %s belongs to an unknown issuer", filename)
			}
74
			sk, err := gabi.NewPrivateKeyFromFile(filepath.Join(conf.IssuerPrivateKeysPath, filename))
75
76
77
			if err != nil {
				return err
			}
78
			conf.IssuerPrivateKeys[issid] = sk
79
80
		}
	}
81
	for issid, sk := range conf.IssuerPrivateKeys {
82
83
84
85
86
87
88
89
90
91
92
93
		pk, err := conf.IrmaConfiguration.PublicKey(issid, int(sk.Counter))
		if err != nil {
			return err
		}
		if pk == nil {
			return errors.Errorf("Missing public key belonging to private key %s-%d", issid.String(), sk.Counter)
		}
		if new(big.Int).Mul(sk.P, sk.Q).Cmp(pk.N) != 0 {
			return errors.Errorf("Private key %s-%d does not belong to corresponding public key", issid.String(), sk.Counter)
		}
	}

94
95
96
	if conf.URL != "" {
		if !strings.HasSuffix(conf.URL, "/") {
			conf.URL = conf.URL + "/"
97
98
99
100
101
		}
	} else {
		conf.Logger.Warn("No url parameter specified in configuration; unless an url is elsewhere prepended in the QR, the IRMA client will not be able to connect")
	}

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
	return nil
}

func StartSession(request irma.SessionRequest) (*irma.Qr, string, error) {
	if err := request.Validate(); err != nil {
		return nil, "", err
	}
	action := irma.ActionUnknown
	switch request.(type) {
	case *irma.DisclosureRequest:
		action = irma.ActionDisclosing
	case *irma.SignatureRequest:
		action = irma.ActionSigning
	case *irma.IssuanceRequest:
		action = irma.ActionIssuing
		if err := validateIssuanceRequest(request.(*irma.IssuanceRequest)); err != nil {
			return nil, "", err
		}
	default:
		conf.Logger.Warnf("Attempt to start session of invalid type")
		return nil, "", errors.New("Invalid session type")
	}

	session := newSession(action, request)
	conf.Logger.Infof("%s session started, token %s", action, session.token)
	return &irma.Qr{
		Type: action,
129
		URL:  conf.URL + session.token,
130
131
132
	}, session.token, nil
}

Sietse Ringers's avatar
Sietse Ringers committed
133
func GetSessionResult(token string) *server.SessionResult {
Sietse Ringers's avatar
Sietse Ringers committed
134
	session := sessions.get(token)
135
	if session == nil {
Sietse Ringers's avatar
Sietse Ringers committed
136
137
138
139
140
		return nil
	}
	return session.result
}

141
142
143
144
145
146
147
148
149
func CancelSession(token string) error {
	session := sessions.get(token)
	if session == nil {
		return errors.New("Unknown session, can't cancel")
	}
	session.handleDelete()
	return nil
}

150
151
152
153
154
func HandleProtocolMessage(
	path string,
	method string,
	headers map[string][]string,
	message []byte,
Sietse Ringers's avatar
Sietse Ringers committed
155
) (status int, output []byte, result *server.SessionResult) {
156
157
158
159
160
161
162
163
164
	// Parse path into session and action
	if len(path) > 0 { // Remove any starting and trailing slash
		if path[0] == '/' {
			path = path[1:]
		}
		if path[len(path)-1] == '/' {
			path = path[:len(path)-1]
		}
	}
165

166
	conf.Logger.Debugf("Routing protocol message: %s %s", method, path)
167
	pattern := regexp.MustCompile("(\\w+)/?(|commitments|proofs|status)$")
168
169
170
	matches := pattern.FindStringSubmatch(path)
	if len(matches) != 3 {
		conf.Logger.Warnf("Invalid URL: %s", path)
Sietse Ringers's avatar
Sietse Ringers committed
171
		status, output = server.JsonResponse(nil, server.RemoteError(server.ErrorInvalidRequest, ""))
172
		return
173
174
	}

Sietse Ringers's avatar
Sietse Ringers committed
175
	// Fetch the session
176
	token := matches[1]
177
	noun := matches[2]
178
179
180
	session := sessions.get(token)
	if session == nil {
		conf.Logger.Warnf("Session not found: %s", token)
Sietse Ringers's avatar
Sietse Ringers committed
181
		status, output = server.JsonResponse(nil, server.RemoteError(server.ErrorSessionUnknown, ""))
182
		return
183
	}
184
185
	session.Lock()
	defer session.Unlock()
186

187
	// However we return, if the session has been finished or cancelled by any of the handlers
188
189
	// then we should inform the user by returning a SessionResult - but only if we have not
	// already done this in the past, e.g. by a previous HTTP call handled by this function
190
	defer func() {
191
192
		if session.finished() && !session.returned {
			session.returned = true
193
194
			result = session.result
		}
195
		sessions.update(token, session)
196
197
	}()

198
	// Route to handler
199
	switch len(noun) {
200
	case 0:
201
		if method == http.MethodDelete {
202
203
204
			session.handleDelete()
			status = http.StatusOK
			return
205
		}
206
		if method == http.MethodGet {
207
208
209
210
			h := http.Header(headers)
			min := &irma.ProtocolVersion{}
			max := &irma.ProtocolVersion{}
			if err := json.Unmarshal([]byte(h.Get(irma.MinVersionHeader)), min); err != nil {
Sietse Ringers's avatar
Sietse Ringers committed
211
				status, output = server.JsonResponse(nil, session.fail(server.ErrorMalformedInput, err.Error()))
212
				return
213
214
			}
			if err := json.Unmarshal([]byte(h.Get(irma.MaxVersionHeader)), max); err != nil {
Sietse Ringers's avatar
Sietse Ringers committed
215
				status, output = server.JsonResponse(nil, session.fail(server.ErrorMalformedInput, err.Error()))
216
				return
217
			}
Sietse Ringers's avatar
Sietse Ringers committed
218
			status, output = server.JsonResponse(session.handleGetRequest(min, max))
219
			return
220
		}
Sietse Ringers's avatar
Sietse Ringers committed
221
		status, output = server.JsonResponse(nil, session.fail(server.ErrorInvalidRequest, ""))
222
		return
223
	default:
224
225
		if method == http.MethodGet && noun == "status" {
			status, output = server.JsonResponse(session.handleGetStatus())
Sietse Ringers's avatar
Sietse Ringers committed
226
			return
227
228
229
		}

		// Below are only POST enpoints
230
		if method != http.MethodPost {
Sietse Ringers's avatar
Sietse Ringers committed
231
			status, output = server.JsonResponse(nil, session.fail(server.ErrorInvalidRequest, ""))
Sietse Ringers's avatar
Sietse Ringers committed
232
233
234
			return
		}

235
		if noun == "commitments" && session.action == irma.ActionIssuing {
Sietse Ringers's avatar
Sietse Ringers committed
236
			commitments := &irma.IssueCommitmentMessage{}
Sietse Ringers's avatar
Sietse Ringers committed
237
			if err := irma.UnmarshalValidate(message, commitments); err != nil {
Sietse Ringers's avatar
Sietse Ringers committed
238
				status, output = server.JsonResponse(nil, session.fail(server.ErrorMalformedInput, ""))
239
				return
240
			}
Sietse Ringers's avatar
Sietse Ringers committed
241
			status, output = server.JsonResponse(session.handlePostCommitments(commitments))
Sietse Ringers's avatar
Sietse Ringers committed
242
243
			return
		}
244
		if noun == "proofs" && session.action == irma.ActionDisclosing {
Sietse Ringers's avatar
Sietse Ringers committed
245
246
			disclosure := irma.Disclosure{}
			if err := irma.UnmarshalValidate(message, &disclosure); err != nil {
Sietse Ringers's avatar
Sietse Ringers committed
247
				status, output = server.JsonResponse(nil, session.fail(server.ErrorMalformedInput, ""))
248
				return
249
			}
Sietse Ringers's avatar
Sietse Ringers committed
250
			status, output = server.JsonResponse(session.handlePostDisclosure(disclosure))
Sietse Ringers's avatar
Sietse Ringers committed
251
252
			return
		}
253
		if noun == "proofs" && session.action == irma.ActionSigning {
Sietse Ringers's avatar
Sietse Ringers committed
254
255
			signature := &irma.SignedMessage{}
			if err := irma.UnmarshalValidate(message, signature); err != nil {
Sietse Ringers's avatar
Sietse Ringers committed
256
				status, output = server.JsonResponse(nil, session.fail(server.ErrorMalformedInput, ""))
257
				return
258
			}
Sietse Ringers's avatar
Sietse Ringers committed
259
			status, output = server.JsonResponse(session.handlePostSignature(signature))
260
			return
261
		}
Sietse Ringers's avatar
Sietse Ringers committed
262

Sietse Ringers's avatar
Sietse Ringers committed
263
		status, output = server.JsonResponse(nil, session.fail(server.ErrorInvalidRequest, ""))
264
		return
265
266
	}
}