requests.go 24.6 KB
Newer Older
1
package irma
2
3

import (
4
	"encoding/json"
5
	"fmt"
6
	"io/ioutil"
7
8
9
	"strconv"
	"time"

10
	"github.com/bwesterb/go-atum"
11
	"github.com/dgrijalva/jwt-go"
Sietse Ringers's avatar
Sietse Ringers committed
12
	"github.com/go-errors/errors"
13
	"github.com/privacybydesign/gabi"
14
	"github.com/privacybydesign/gabi/big"
15
	"github.com/privacybydesign/irmago/internal/fs"
Sietse Ringers's avatar
Sietse Ringers committed
16
)
17

18
19
// BaseRequest contains the context and nonce for an IRMA session.
type BaseRequest struct {
20
21
	// Denotes session type, must be "disclosing", "signing" or "issuing"
	Type Action `json:"type"`
22
23
	// Message version. Current version is 2.
	Version int `json:"v"`
24

25
	// Chosen by the IRMA server during the session
26
27
28
	Context         *big.Int         `json:"context,omitempty"`
	Nonce           *big.Int         `json:"nonce,omitempty"`
	ProtocolVersion *ProtocolVersion `json:"protocolVersion,omitempty"`
29
30
31

	// cache for Identifiers() method
	ids *IrmaIdentifierSet
Sietse Ringers's avatar
Sietse Ringers committed
32
33
}

34
35
// An AttributeCon is only satisfied if all of its containing attribute requests are satisfied.
type AttributeCon []AttributeRequest
36

37
38
// An AttributeDisCon is satisfied if at least one of its containing AttributeCon is satisfied.
type AttributeDisCon []AttributeCon
39

40
41
// AttributeConDisCon is only satisfied if all of the containing AttributeDisCon are satisfied.
type AttributeConDisCon []AttributeDisCon
42

43
44
// A DisclosureRequest is a request to disclose certain attributes. Construct new instances using
// NewDisclosureRequest().
45
type DisclosureRequest struct {
46
	BaseRequest
47
48
49

	Disclose AttributeConDisCon       `json:"disclose"`
	Labels   map[int]TranslatedString `json:"labels"`
50
51
}

52
53
// A SignatureRequest is a a request to sign a message with certain attributes. Construct new
// instances using NewSignatureRequest().
54
type SignatureRequest struct {
55
	DisclosureRequest
56
	Message string `json:"message"`
57
58
}

Sietse Ringers's avatar
Sietse Ringers committed
59
// An IssuanceRequest is a request to issue certain credentials,
60
61
// optionally also asking for certain attributes to be simultaneously disclosed. Construct new
// instances using NewIssuanceRequest().
62
type IssuanceRequest struct {
63
	DisclosureRequest
64
	Credentials []*CredentialRequest `json:"credentials"`
Tomas's avatar
Tomas committed
65
66
67

	// Derived data
	CredentialInfoList        CredentialInfoList `json:",omitempty"`
68
	RemovalCredentialInfoList CredentialInfoList `json:",omitempty"`
69
70
}

Sietse Ringers's avatar
Sietse Ringers committed
71
72
// A CredentialRequest contains the attributes and metadata of a credential
// that will be issued in an IssuanceRequest.
73
type CredentialRequest struct {
74
75
	Validity         *Timestamp               `json:"validity,omitempty"`
	KeyCounter       int                      `json:"keyCounter,omitempty"`
76
77
	CredentialTypeID CredentialTypeIdentifier `json:"credential"`
	Attributes       map[string]string        `json:"attributes"`
Sietse Ringers's avatar
Sietse Ringers committed
78
79
}

80
81
82
83
// SessionRequest instances contain all information the irmaclient needs to perform an IRMA session.
type SessionRequest interface {
	Validator
	Base() *BaseRequest
84
	GetNonce(timestamp *atum.Timestamp) *big.Int
85
86
87
88
89
90
91
92
	Disclosure() *DisclosureRequest
	Identifiers() *IrmaIdentifierSet
	Action() Action
}

// Timestamp is a time.Time that marshals to Unix timestamps.
type Timestamp time.Time

Sietse Ringers's avatar
Sietse Ringers committed
93
94
95
96
97
98
99
// ServerJwt contains standard JWT fields.
type ServerJwt struct {
	Type       string    `json:"sub"`
	ServerName string    `json:"iss"`
	IssuedAt   Timestamp `json:"iat"`
}

100
101
// RequestorBaseRequest contains fields present in all RequestorRequest types
// with which the requestor configures an IRMA session.
102
type RequestorBaseRequest struct {
Sietse Ringers's avatar
Sietse Ringers committed
103
104
105
	ResultJwtValidity int    `json:"validity,omitempty"`    // Validity of session result JWT in seconds
	ClientTimeout     int    `json:"timeout,omitempty"`     // Wait this many seconds for the IRMA app to connect before the session times out
	CallbackUrl       string `json:"callbackUrl,omitempty"` // URL to post session result to
106
107
}

108
109
// RequestorRequest is the message with which requestors start an IRMA session. It contains a
// SessionRequest instance for the irmaclient along with extra fields in a RequestorBaseRequest.
110
111
112
type RequestorRequest interface {
	Validator
	SessionRequest() SessionRequest
113
	Base() RequestorBaseRequest
114
115
}

Sietse Ringers's avatar
Sietse Ringers committed
116
117
// A ServiceProviderRequest contains a disclosure request.
type ServiceProviderRequest struct {
118
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
119
120
121
122
123
	Request *DisclosureRequest `json:"request"`
}

// A SignatureRequestorRequest contains a signing request.
type SignatureRequestorRequest struct {
124
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
125
126
127
128
129
	Request *SignatureRequest `json:"request"`
}

// An IdentityProviderRequest contains an issuance request.
type IdentityProviderRequest struct {
130
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
131
132
133
134
135
136
	Request *IssuanceRequest `json:"request"`
}

// ServiceProviderJwt is a requestor JWT for a disclosure session.
type ServiceProviderJwt struct {
	ServerJwt
137
	Request *ServiceProviderRequest `json:"sprequest"`
Sietse Ringers's avatar
Sietse Ringers committed
138
139
140
141
142
}

// SignatureRequestorJwt is a requestor JWT for a signing session.
type SignatureRequestorJwt struct {
	ServerJwt
143
	Request *SignatureRequestorRequest `json:"absrequest"`
Sietse Ringers's avatar
Sietse Ringers committed
144
145
146
147
148
}

// IdentityProviderJwt is a requestor JWT for issuance session.
type IdentityProviderJwt struct {
	ServerJwt
149
150
151
	Request *IdentityProviderRequest `json:"iprequest"`
}

152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
// A RequestorJwt contains an IRMA session object.
type RequestorJwt interface {
	Action() Action
	RequestorRequest() RequestorRequest
	SessionRequest() SessionRequest
	Requestor() string
	Valid() error
	Sign(jwt.SigningMethod, interface{}) (string, error)
}

// A DisclosureChoice contains the attributes chosen to be disclosed.
type DisclosureChoice struct {
	Attributes [][]*AttributeIdentifier
}

// An AttributeRequest asks for an instance of an attribute type, possibly requiring it to have
// a specified value, in a session request.
type AttributeRequest struct {
	Type  AttributeTypeIdentifier
	Value *string
}

var (
	bigZero = big.NewInt(0)
	bigOne  = big.NewInt(1)
)

func (b *BaseRequest) GetContext() *big.Int {
	if b.Context == nil {
		return bigOne
182
	}
183
	return b.Context
184
185
}

186
func (b *BaseRequest) GetNonce(*atum.Timestamp) *big.Int {
187
188
	if b.Nonce == nil {
		return bigZero
189
	}
190
	return b.Nonce
191
192
}

193
194
195
196
197
198
// CredentialTypes returns an array of all credential types occuring in this conjunction.
func (c AttributeCon) CredentialTypes() []CredentialTypeIdentifier {
	var result []CredentialTypeIdentifier
	tmp := map[CredentialTypeIdentifier]struct{}{}
	for _, attr := range c {
		tmp[attr.Type.CredentialTypeIdentifier()] = struct{}{}
199
	}
200
201
202
203
	for cred := range tmp {
		result = append(result, cred)
	}
	return result
204
205
}

206
207
208
func (c *AttributeCon) MarshalJSON() ([]byte, error) {
	var vals bool
	m := map[AttributeTypeIdentifier]*string{}
209
	l := make([]AttributeTypeIdentifier, 0, len(*c))
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224

	for _, attr := range *c {
		m[attr.Type] = attr.Value
		if attr.Value == nil {
			l = append(l, attr.Type)
		} else {
			vals = true
		}
	}

	if vals {
		return json.Marshal(m)
	} else {
		return json.Marshal(l)
	}
225
226
}

227
228
func (c *AttributeCon) UnmarshalJSON(bts []byte) error {
	var err error
229
230
231
232
233
234
235
236
237

	var l []AttributeTypeIdentifier
	if err = json.Unmarshal(bts, &l); err == nil {
		for _, id := range l {
			*c = append(*c, AttributeRequest{Type: id})
		}
		return nil
	}

238
239
240
241
242
243
244
245
	m := map[AttributeTypeIdentifier]*string{}
	if err = json.Unmarshal(bts, &m); err == nil {
		for id, val := range m {
			*c = append(*c, AttributeRequest{Type: id, Value: val})
		}
		return nil
	}

246
247
248
249
	var s string
	if err = json.Unmarshal(bts, &s); err == nil {
		*c = append(*c, NewAttributeRequest(s))
		return nil
250
251
	}

252
	return errors.New("Failed to unmarshal attribute conjunction")
253
254
}

255
256
func (ar *AttributeRequest) Satisfy(attr AttributeTypeIdentifier, val *string) bool {
	return ar.Type == attr && (ar.Value == nil || (val != nil && *ar.Value == *val))
Sietse Ringers's avatar
Sietse Ringers committed
257
258
}

259
func (c AttributeCon) Satisfy(proofs gabi.ProofList, indices []*DisclosedAttributeIndex, conf *Configuration) (bool, []*DisclosedAttribute, error) {
260
	if len(indices) < len(c) {
261
		return false, nil, nil
262
263
	}
	attrs := make([]*DisclosedAttribute, 0, len(c))
264
265
266
	if len(c) == 0 {
		return true, attrs, nil
	}
267
268
269
270
271

	for j := range c {
		index := indices[j]
		attr, val, err := extractAttribute(proofs, index, conf)
		if err != nil {
272
			return false, nil, err
273
274
		}
		if !c[j].Satisfy(attr.Identifier, val) {
275
			return false, nil, nil
276
277
278
		}
		attrs = append(attrs, attr)
	}
279
	return true, attrs, nil
280
281
}

282
func (dc AttributeDisCon) Satisfy(proofs gabi.ProofList, indices []*DisclosedAttributeIndex, conf *Configuration) (bool, []*DisclosedAttribute, error) {
283
	for _, con := range dc {
284
285
286
		satisfied, attrs, err := con.Satisfy(proofs, indices, conf)
		if satisfied || err != nil {
			return true, attrs, err
287
288
		}
	}
289
	return false, nil, nil
290
291
}

292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
func (cdc AttributeConDisCon) Validate(conf *Configuration) error {
	for _, discon := range cdc {
		for _, con := range discon {
			var nonsingleton *CredentialTypeIdentifier
			for _, attr := range con {
				typ := attr.Type.CredentialTypeIdentifier()
				if !conf.CredentialTypes[typ].IsSingleton {
					if nonsingleton != nil && *nonsingleton != typ {
						return errors.New("Multiple non-singletons within one inner conjunction are not allowed")
					} else {
						nonsingleton = &typ
					}
				}
			}
		}
	}
	return nil
}

311
312
313
314
315
316
317
318
func (cdc AttributeConDisCon) Satisfy(disclosure *Disclosure, conf *Configuration) (bool, [][]*DisclosedAttribute, error) {
	if len(disclosure.Indices) < len(cdc) {
		return false, nil, nil
	}
	list := make([][]*DisclosedAttribute, len(cdc))
	complete := true

	for i, discon := range cdc {
319
		satisfied, attrs, err := discon.Satisfy(disclosure.Proofs, disclosure.Indices[i], conf)
320
321
322
		if err != nil {
			return false, nil, err
		}
323
		if satisfied {
324
325
326
327
328
329
330
331
			list[i] = attrs
		} else {
			complete = false
			list[i] = nil
		}
	}

	return complete, list, nil
332
333
}

334
335
336
337
338
339
340
341
342
343
344
345
func (cdc AttributeConDisCon) Iterate(f func(attr *AttributeRequest) error) error {
	var err error
	for _, discon := range cdc {
		for _, con := range discon {
			for _, attr := range con {
				if err = f(&attr); err != nil {
					return err
				}
			}
		}
	}
	return nil
346
347
}

348
349
350
351
352
353
354
func (dr *DisclosureRequest) AddSingle(attr AttributeTypeIdentifier, value *string, label TranslatedString) {
	dr.Disclose = append(dr.Disclose, AttributeDisCon{AttributeCon{{Type: attr, Value: value}}})
	dr.Labels[len(dr.Disclose)-1] = label
}

func NewDisclosureRequest(attrs ...AttributeTypeIdentifier) *DisclosureRequest {
	request := &DisclosureRequest{
355
		BaseRequest: BaseRequest{Type: ActionDisclosing, Version: 2},
356
357
358
359
360
361
362
363
364
365
366
367
		Labels:      map[int]TranslatedString{},
	}
	for _, attr := range attrs {
		request.AddSingle(attr, nil, nil)
	}
	return request
}

func NewSignatureRequest(message string, attrs ...AttributeTypeIdentifier) *SignatureRequest {
	dr := NewDisclosureRequest(attrs...)
	dr.Type = ActionSigning
	return &SignatureRequest{
368
		DisclosureRequest: *dr,
369
370
371
372
373
374
375
376
		Message:           message,
	}
}

func NewIssuanceRequest(creds []*CredentialRequest, attrs ...AttributeTypeIdentifier) *IssuanceRequest {
	dr := NewDisclosureRequest(attrs...)
	dr.Type = ActionIssuing
	return &IssuanceRequest{
377
		DisclosureRequest: *dr,
378
379
380
381
382
383
384
385
386
		Credentials:       creds,
	}
}

func (dr *DisclosureRequest) Disclosure() *DisclosureRequest {
	return dr
}

func (dr *DisclosureRequest) Identifiers() *IrmaIdentifierSet {
387
388
	if dr.ids == nil {
		dr.ids = &IrmaIdentifierSet{
389
390
391
392
393
394
395
396
			SchemeManagers:  map[SchemeManagerIdentifier]struct{}{},
			Issuers:         map[IssuerIdentifier]struct{}{},
			CredentialTypes: map[CredentialTypeIdentifier]struct{}{},
			PublicKeys:      map[IssuerIdentifier][]int{},
		}

		_ = dr.Disclose.Iterate(func(a *AttributeRequest) error {
			attr := a.Type
397
398
399
			dr.ids.SchemeManagers[attr.CredentialTypeIdentifier().IssuerIdentifier().SchemeManagerIdentifier()] = struct{}{}
			dr.ids.Issuers[attr.CredentialTypeIdentifier().IssuerIdentifier()] = struct{}{}
			dr.ids.CredentialTypes[attr.CredentialTypeIdentifier()] = struct{}{}
400
401
402
			return nil
		})
	}
403
	return dr.ids
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
}

func (dr *DisclosureRequest) Base() *BaseRequest {
	return &dr.BaseRequest
}

func (dr *DisclosureRequest) Action() Action { return ActionDisclosing }

func (dr *DisclosureRequest) Validate() error {
	if dr.Type != ActionDisclosing {
		return errors.New("Not a disclosure request")
	}
	if len(dr.Disclose) == 0 {
		return errors.New("Disclosure request had no attributes")
	}
419
420
421
422
423
	for _, discon := range dr.Disclose {
		if len(discon) == 0 {
			return errors.New("Empty disjunction")
		}
	}
424
425
	return nil
}
Sietse Ringers's avatar
Sietse Ringers committed
426

427
428
func (cr *CredentialRequest) Info(conf *Configuration, metadataVersion byte) (*CredentialInfo, error) {
	list, err := cr.AttributeList(conf, metadataVersion)
429
430
431
	if err != nil {
		return nil, err
	}
432
	return NewCredentialInfo(list.Ints, conf), nil
433
434
}

435
436
437
438
// Validate checks that this credential request is consistent with the specified Configuration:
// the credential type is known, all required attributes are present and no unknown attributes
// are given.
func (cr *CredentialRequest) Validate(conf *Configuration) error {
439
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
Sietse Ringers's avatar
Sietse Ringers committed
440
	if credtype == nil {
441
		return errors.New("Credential request of unknown credential type")
Sietse Ringers's avatar
Sietse Ringers committed
442
	}
443
444
445
446
447

	// Check that there are no attributes in the credential request that aren't
	// in the credential descriptor.
	for crName := range cr.Attributes {
		found := false
448
		for _, ad := range credtype.AttributeTypes {
449
450
451
452
453
454
			if ad.ID == crName {
				found = true
				break
			}
		}
		if !found {
455
			return errors.New("Credential request contaiins unknown attribute")
456
		}
Sietse Ringers's avatar
Sietse Ringers committed
457
458
	}

459
	for _, attrtype := range credtype.AttributeTypes {
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
		if _, present := cr.Attributes[attrtype.ID]; !present && attrtype.Optional != "true" {
			return errors.New("Required attribute not present in credential request")
		}
	}

	return nil
}

// AttributeList returns the list of attributes from this credential request.
func (cr *CredentialRequest) AttributeList(conf *Configuration, metadataVersion byte) (*AttributeList, error) {
	if err := cr.Validate(conf); err != nil {
		return nil, err
	}

	// Compute metadata attribute
	meta := NewMetadataAttribute(metadataVersion)
	meta.setKeyCounter(cr.KeyCounter)
	meta.setCredentialTypeIdentifier(cr.CredentialTypeID.String())
	meta.setSigningDate()
	if err := meta.setExpiryDate(cr.Validity); err != nil {
		return nil, err
	}

	// Compute other attributes
484
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
485
	attrs := make([]*big.Int, len(credtype.AttributeTypes)+1)
Sietse Ringers's avatar
Sietse Ringers committed
486
	attrs[0] = meta.Int
487
	for i, attrtype := range credtype.AttributeTypes {
488
		attrs[i+1] = new(big.Int)
Sietse Ringers's avatar
Sietse Ringers committed
489
		if str, present := cr.Attributes[attrtype.ID]; present {
490
			// Set attribute to str << 1 + 1
491
			attrs[i+1].SetBytes([]byte(str))
492
493
494
495
			if meta.Version() >= 0x03 {
				attrs[i+1].Lsh(attrs[i+1], 1)             // attr <<= 1
				attrs[i+1].Add(attrs[i+1], big.NewInt(1)) // attr += 1
			}
Sietse Ringers's avatar
Sietse Ringers committed
496
497
498
		}
	}

499
	return NewAttributeListFromInts(attrs, conf), nil
Sietse Ringers's avatar
Sietse Ringers committed
500
501
}

Sietse Ringers's avatar
Sietse Ringers committed
502
func (ir *IssuanceRequest) Identifiers() *IrmaIdentifierSet {
503
504
	if ir.ids == nil {
		ir.ids = &IrmaIdentifierSet{
505
506
507
508
			SchemeManagers:  map[SchemeManagerIdentifier]struct{}{},
			Issuers:         map[IssuerIdentifier]struct{}{},
			CredentialTypes: map[CredentialTypeIdentifier]struct{}{},
			PublicKeys:      map[IssuerIdentifier][]int{},
Sietse Ringers's avatar
Sietse Ringers committed
509
510
		}

Sietse Ringers's avatar
Sietse Ringers committed
511
		for _, credreq := range ir.Credentials {
Sietse Ringers's avatar
Sietse Ringers committed
512
			issuer := credreq.CredentialTypeID.IssuerIdentifier()
513
514
515
516
517
			ir.ids.SchemeManagers[issuer.SchemeManagerIdentifier()] = struct{}{}
			ir.ids.Issuers[issuer] = struct{}{}
			ir.ids.CredentialTypes[credreq.CredentialTypeID] = struct{}{}
			if ir.ids.PublicKeys[issuer] == nil {
				ir.ids.PublicKeys[issuer] = []int{}
518
			}
519
			ir.ids.PublicKeys[issuer] = append(ir.ids.PublicKeys[issuer], credreq.KeyCounter)
520
521
		}

522
		ir.ids.join(ir.DisclosureRequest.Identifiers())
Sietse Ringers's avatar
Sietse Ringers committed
523
	}
524
	return ir.ids
Sietse Ringers's avatar
Sietse Ringers committed
525
526
}

Tomas's avatar
Tomas committed
527
528
529
530
531
532
533
534
535
536
537
538
539
func (ir *IssuanceRequest) GetCredentialInfoList(conf *Configuration, version *ProtocolVersion) (CredentialInfoList, error) {
	if ir.CredentialInfoList == nil {
		for _, credreq := range ir.Credentials {
			info, err := credreq.Info(conf, GetMetadataVersion(version))
			if err != nil {
				return nil, err
			}
			ir.CredentialInfoList = append(ir.CredentialInfoList, info)
		}
	}
	return ir.CredentialInfoList, nil
}

540
541
func (ir *IssuanceRequest) Action() Action { return ActionIssuing }

542
543
544
545
546
547
548
func (ir *IssuanceRequest) Validate() error {
	if ir.Type != ActionIssuing {
		return errors.New("Not an issuance request")
	}
	if len(ir.Credentials) == 0 {
		return errors.New("Empty issuance request")
	}
549
	for _, cred := range ir.Credentials {
550
		if cred.Validity != nil && cred.Validity.Floor().Before(Timestamp(time.Now())) {
551
552
553
			return errors.New("Expired credential request")
		}
	}
554
555
556
	return nil
}

Sietse Ringers's avatar
Sietse Ringers committed
557
558
// GetNonce returns the nonce of this signature session
// (with the message already hashed into it).
559
560
func (sr *SignatureRequest) GetNonce(timestamp *atum.Timestamp) *big.Int {
	return ASN1ConvertSignatureNonce(sr.Message, sr.BaseRequest.GetNonce(nil), timestamp)
561
562
}

563
func (sr *SignatureRequest) SignatureFromMessage(message interface{}, timestamp *atum.Timestamp) (*SignedMessage, error) {
564
	signature, ok := message.(*Disclosure)
565
566
567
568
569

	if !ok {
		return nil, errors.Errorf("Type assertion failed")
	}

570
571
572
573
	nonce := sr.Nonce
	if nonce == nil {
		nonce = bigZero
	}
574
	return &SignedMessage{
575
		Signature: signature.Proofs,
576
		Indices:   signature.Indices,
577
578
		Nonce:     nonce,
		Context:   sr.GetContext(),
579
		Message:   sr.Message,
580
		Timestamp: timestamp,
581
		Version:   2,
582
583
584
	}, nil
}

585
586
func (sr *SignatureRequest) Action() Action { return ActionSigning }

587
588
589
590
591
592
593
func (sr *SignatureRequest) Validate() error {
	if sr.Type != ActionSigning {
		return errors.New("Not a signature request")
	}
	if sr.Message == "" {
		return errors.New("Signature request had empty message")
	}
594
595
	if len(sr.Disclose) == 0 {
		return errors.New("Signature request had no attributes")
596
	}
597
598
599
600
601
	for _, discon := range sr.Disclose {
		if len(discon) == 0 {
			return errors.New("Empty disjunction")
		}
	}
602
603
604
	return nil
}

605
606
607
608
609
// Check if Timestamp is before other Timestamp. Used for checking expiry of attributes
func (t Timestamp) Before(u Timestamp) bool {
	return time.Time(t).Before(time.Time(u))
}

610
611
612
613
func (t Timestamp) After(u Timestamp) bool {
	return time.Time(t).After(time.Time(u))
}

614
615
// MarshalJSON marshals a timestamp.
func (t *Timestamp) MarshalJSON() ([]byte, error) {
616
	return []byte(t.String()), nil
617
618
619
620
621
622
623
624
625
626
627
}

// UnmarshalJSON unmarshals a timestamp.
func (t *Timestamp) UnmarshalJSON(b []byte) error {
	ts, err := strconv.Atoi(string(b))
	if err != nil {
		return err
	}
	*t = Timestamp(time.Unix(int64(ts), 0))
	return nil
}
Sietse Ringers's avatar
Sietse Ringers committed
628

629
630
631
632
633
// Timestamp implements Stringer.
func (t *Timestamp) String() string {
	return fmt.Sprint(time.Time(*t).Unix())
}

634
635
636
637
func (t *Timestamp) Floor() Timestamp {
	return Timestamp(time.Unix((time.Time(*t).Unix()/ExpiryFactor)*ExpiryFactor, 0))
}

638
639
640
641
642
643
644
645
func readTimestamp(path string) (*Timestamp, bool, error) {
	exists, err := fs.PathExists(path)
	if err != nil {
		return nil, false, err
	}
	if !exists {
		return nil, false, nil
	}
646
647
	bts, err := ioutil.ReadFile(path)
	if err != nil {
648
		return nil, true, errors.New("Could not read scheme manager timestamp")
649
	}
650
651
	ts, err := parseTimestamp(bts)
	return ts, true, err
652
653
}

654
func parseTimestamp(bts []byte) (*Timestamp, error) {
655
656
657
658
659
660
	// Remove final character \n if present
	if bts[len(bts)-1] == '\n' {
		bts = bts[:len(bts)-1]
	}
	// convert from byte slice to string; parse as int
	str, err := strconv.ParseInt(string(bts), 10, 64)
661
	if err != nil {
662
		return nil, err
663
	}
664
665
	ts := Timestamp(time.Unix(str, 0))
	return &ts, nil
666
667
}

Sietse Ringers's avatar
Sietse Ringers committed
668
669
670
671
672
673
674
675
// NewServiceProviderJwt returns a new ServiceProviderJwt.
func NewServiceProviderJwt(servername string, dr *DisclosureRequest) *ServiceProviderJwt {
	return &ServiceProviderJwt{
		ServerJwt: ServerJwt{
			ServerName: servername,
			IssuedAt:   Timestamp(time.Now()),
			Type:       "verification_request",
		},
676
677
678
679
		Request: &ServiceProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              dr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
680
681
682
683
684
685
686
687
688
689
690
	}
}

// NewSignatureRequestorJwt returns a new SignatureRequestorJwt.
func NewSignatureRequestorJwt(servername string, sr *SignatureRequest) *SignatureRequestorJwt {
	return &SignatureRequestorJwt{
		ServerJwt: ServerJwt{
			ServerName: servername,
			IssuedAt:   Timestamp(time.Now()),
			Type:       "signature_request",
		},
691
692
693
694
		Request: &SignatureRequestorRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              sr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
695
696
697
698
699
700
701
702
703
704
705
	}
}

// NewIdentityProviderJwt returns a new IdentityProviderJwt.
func NewIdentityProviderJwt(servername string, ir *IssuanceRequest) *IdentityProviderJwt {
	return &IdentityProviderJwt{
		ServerJwt: ServerJwt{
			ServerName: servername,
			IssuedAt:   Timestamp(time.Now()),
			Type:       "issue_request",
		},
706
707
708
709
		Request: &IdentityProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              ir,
		},
Sietse Ringers's avatar
Sietse Ringers committed
710
711
712
	}
}

713
714
715
716
717
718
719
func (jwt *ServerJwt) Requestor() string { return jwt.ServerName }

func (r *ServiceProviderRequest) Validate() error {
	if r.Request == nil {
		return errors.New("Not a ServiceProviderRequest")
	}
	return r.Request.Validate()
Sietse Ringers's avatar
Sietse Ringers committed
720
721
}

722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
func (r *SignatureRequestorRequest) Validate() error {
	if r.Request == nil {
		return errors.New("Not a SignatureRequestorRequest")
	}
	return r.Request.Validate()
}

func (r *IdentityProviderRequest) Validate() error {
	if r.Request == nil {
		return errors.New("Not a IdentityProviderRequest")
	}
	return r.Request.Validate()
}

func (r *ServiceProviderRequest) SessionRequest() SessionRequest {
	return r.Request
}

func (r *SignatureRequestorRequest) SessionRequest() SessionRequest {
	return r.Request
}

func (r *IdentityProviderRequest) SessionRequest() SessionRequest {
	return r.Request
}

func (r *ServiceProviderRequest) Base() RequestorBaseRequest {
	return r.RequestorBaseRequest
}

func (r *SignatureRequestorRequest) Base() RequestorBaseRequest {
	return r.RequestorBaseRequest
}

func (r *IdentityProviderRequest) Base() RequestorBaseRequest {
	return r.RequestorBaseRequest
}
759

760
// SessionRequest returns an IRMA session object.
761
func (claims *ServiceProviderJwt) SessionRequest() SessionRequest { return claims.Request.Request }
Sietse Ringers's avatar
Sietse Ringers committed
762

763
// SessionRequest returns an IRMA session object.
764
func (claims *SignatureRequestorJwt) SessionRequest() SessionRequest { return claims.Request.Request }
Sietse Ringers's avatar
Sietse Ringers committed
765

766
// SessionRequest returns an IRMA session object.
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
func (claims *IdentityProviderJwt) SessionRequest() SessionRequest { return claims.Request.Request }

func (claims *ServiceProviderJwt) Sign(method jwt.SigningMethod, key interface{}) (string, error) {
	return jwt.NewWithClaims(method, claims).SignedString(key)
}

func (claims *SignatureRequestorJwt) Sign(method jwt.SigningMethod, key interface{}) (string, error) {
	return jwt.NewWithClaims(method, claims).SignedString(key)
}

func (claims *IdentityProviderJwt) Sign(method jwt.SigningMethod, key interface{}) (string, error) {
	return jwt.NewWithClaims(method, claims).SignedString(key)
}

func (claims *ServiceProviderJwt) RequestorRequest() RequestorRequest { return claims.Request }

func (claims *SignatureRequestorJwt) RequestorRequest() RequestorRequest { return claims.Request }

func (claims *IdentityProviderJwt) RequestorRequest() RequestorRequest { return claims.Request }

func (claims *ServiceProviderJwt) Valid() error {
	if claims.Type != "verification_request" {
789
790
791

		return errors.New("Verification jwt has invalid subject")
	}
792
	if time.Time(claims.IssuedAt).After(time.Now()) {
793
794
795
796
797
		return errors.New("Verification jwt not yet valid")
	}
	return nil
}

798
799
func (claims *SignatureRequestorJwt) Valid() error {
	if claims.Type != "signature_request" {
800
801
		return errors.New("Signature jwt has invalid subject")
	}
802
	if time.Time(claims.IssuedAt).After(time.Now()) {
803
804
805
806
807
		return errors.New("Signature jwt not yet valid")
	}
	return nil
}

808
809
func (claims *IdentityProviderJwt) Valid() error {
	if claims.Type != "issue_request" {
810
811
		return errors.New("Issuance jwt has invalid subject")
	}
812
	if time.Time(claims.IssuedAt).After(time.Now()) {
813
814
815
816
817
		return errors.New("Issuance jwt not yet valid")
	}
	return nil
}

818
819
820
func (claims *ServiceProviderJwt) Action() Action { return ActionDisclosing }

func (claims *SignatureRequestorJwt) Action() Action { return ActionSigning }
821

822
func (claims *IdentityProviderJwt) Action() Action { return ActionIssuing }
823

824
func SignSessionRequest(request SessionRequest, alg jwt.SigningMethod, key interface{}, name string) (string, error) {
825
826
827
828
829
830
831
832
833
834
835
	var jwtcontents RequestorJwt
	switch r := request.(type) {
	case *IssuanceRequest:
		jwtcontents = NewIdentityProviderJwt(name, r)
	case *DisclosureRequest:
		jwtcontents = NewServiceProviderJwt(name, r)
	case *SignatureRequest:
		jwtcontents = NewSignatureRequestorJwt(name, r)
	}
	return jwtcontents.Sign(alg, key)
}
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851

func SignRequestorRequest(request RequestorRequest, alg jwt.SigningMethod, key interface{}, name string) (string, error) {
	var jwtcontents RequestorJwt
	switch r := request.(type) {
	case *IdentityProviderRequest:
		jwtcontents = NewIdentityProviderJwt(name, nil)
		jwtcontents.(*IdentityProviderJwt).Request = r
	case *ServiceProviderRequest:
		jwtcontents = NewServiceProviderJwt(name, nil)
		jwtcontents.(*ServiceProviderJwt).Request = r
	case *SignatureRequestorRequest:
		jwtcontents = NewSignatureRequestorJwt(name, nil)
		jwtcontents.(*SignatureRequestorJwt).Request = r
	}
	return jwtcontents.Sign(alg, key)
}
852
853
854
855
856

// NewAttributeRequest requests the specified attribute.
func NewAttributeRequest(attr string) AttributeRequest {
	return AttributeRequest{Type: NewAttributeTypeIdentifier(attr)}
}