requests.go 23.9 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 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224
func (c *AttributeCon) MarshalJSON() ([]byte, error) {
	var vals bool
	m := map[AttributeTypeIdentifier]*string{}
	var l []AttributeTypeIdentifier

	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 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276
func (c AttributeCon) Satisfy(proofs gabi.ProofList, indices []*DisclosedAttributeIndex, conf *Configuration) ([]*DisclosedAttribute, error) {
	if len(indices) < len(c) {
		return nil, nil
	}
	attrs := make([]*DisclosedAttribute, 0, len(c))

	for j := range c {
		index := indices[j]
		attr, val, err := extractAttribute(proofs, index, conf)
		if err != nil {
			return nil, err
		}
		if !c[j].Satisfy(attr.Identifier, val) {
			return nil, nil
		}
		attrs = append(attrs, attr)
	}
	return attrs, nil
277 278
}

279 280 281 282 283 284 285 286
func (dc AttributeDisCon) Satisfy(proofs gabi.ProofList, indices []*DisclosedAttributeIndex, conf *Configuration) ([]*DisclosedAttribute, error) {
	for _, con := range dc {
		attrs, err := con.Satisfy(proofs, indices, conf)
		if len(attrs) > 0 || err != nil {
			return attrs, err
		}
	}
	return nil, nil
287 288
}

289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309
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 {
		attrs, err := discon.Satisfy(disclosure.Proofs, disclosure.Indices[i], conf)
		if err != nil {
			return false, nil, err
		}
		if len(attrs) > 0 {
			list[i] = attrs
		} else {
			complete = false
			list[i] = nil
		}
	}

	return complete, list, nil
310 311
}

312 313 314 315 316 317 318 319 320 321 322 323
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
324 325
}

326 327 328 329 330 331 332
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{
333
		BaseRequest: BaseRequest{Type: ActionDisclosing, Version: 2},
334 335 336 337 338 339 340 341 342 343 344 345
		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{
346
		DisclosureRequest: *dr,
347 348 349 350 351 352 353 354
		Message:           message,
	}
}

func NewIssuanceRequest(creds []*CredentialRequest, attrs ...AttributeTypeIdentifier) *IssuanceRequest {
	dr := NewDisclosureRequest(attrs...)
	dr.Type = ActionIssuing
	return &IssuanceRequest{
355
		DisclosureRequest: *dr,
356 357 358 359 360 361 362 363 364
		Credentials:       creds,
	}
}

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

func (dr *DisclosureRequest) Identifiers() *IrmaIdentifierSet {
365 366
	if dr.ids == nil {
		dr.ids = &IrmaIdentifierSet{
367 368 369 370 371 372 373 374
			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
375 376 377
			dr.ids.SchemeManagers[attr.CredentialTypeIdentifier().IssuerIdentifier().SchemeManagerIdentifier()] = struct{}{}
			dr.ids.Issuers[attr.CredentialTypeIdentifier().IssuerIdentifier()] = struct{}{}
			dr.ids.CredentialTypes[attr.CredentialTypeIdentifier()] = struct{}{}
378 379 380
			return nil
		})
	}
381
	return dr.ids
382 383 384 385 386 387 388 389 390 391 392 393 394 395 396
}

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")
	}
397 398 399 400 401
	for _, discon := range dr.Disclose {
		if len(discon) == 0 {
			return errors.New("Empty disjunction")
		}
	}
402 403
	return nil
}
Sietse Ringers's avatar
Sietse Ringers committed
404

405 406
func (cr *CredentialRequest) Info(conf *Configuration, metadataVersion byte) (*CredentialInfo, error) {
	list, err := cr.AttributeList(conf, metadataVersion)
407 408 409
	if err != nil {
		return nil, err
	}
410
	return NewCredentialInfo(list.Ints, conf), nil
411 412
}

413 414 415 416
// 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 {
417
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
Sietse Ringers's avatar
Sietse Ringers committed
418
	if credtype == nil {
419
		return errors.New("Credential request of unknown credential type")
Sietse Ringers's avatar
Sietse Ringers committed
420
	}
421 422 423 424 425

	// Check that there are no attributes in the credential request that aren't
	// in the credential descriptor.
	for crName := range cr.Attributes {
		found := false
426
		for _, ad := range credtype.AttributeTypes {
427 428 429 430 431 432
			if ad.ID == crName {
				found = true
				break
			}
		}
		if !found {
433
			return errors.New("Credential request contaiins unknown attribute")
434
		}
Sietse Ringers's avatar
Sietse Ringers committed
435 436
	}

437
	for _, attrtype := range credtype.AttributeTypes {
438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461
		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
462
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
463
	attrs := make([]*big.Int, len(credtype.AttributeTypes)+1)
Sietse Ringers's avatar
Sietse Ringers committed
464
	attrs[0] = meta.Int
465
	for i, attrtype := range credtype.AttributeTypes {
466
		attrs[i+1] = new(big.Int)
Sietse Ringers's avatar
Sietse Ringers committed
467
		if str, present := cr.Attributes[attrtype.ID]; present {
468
			// Set attribute to str << 1 + 1
469
			attrs[i+1].SetBytes([]byte(str))
470 471 472 473
			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
474 475 476
		}
	}

477
	return NewAttributeListFromInts(attrs, conf), nil
Sietse Ringers's avatar
Sietse Ringers committed
478 479
}

Sietse Ringers's avatar
Sietse Ringers committed
480
func (ir *IssuanceRequest) Identifiers() *IrmaIdentifierSet {
481 482
	if ir.ids == nil {
		ir.ids = &IrmaIdentifierSet{
483 484 485 486
			SchemeManagers:  map[SchemeManagerIdentifier]struct{}{},
			Issuers:         map[IssuerIdentifier]struct{}{},
			CredentialTypes: map[CredentialTypeIdentifier]struct{}{},
			PublicKeys:      map[IssuerIdentifier][]int{},
Sietse Ringers's avatar
Sietse Ringers committed
487 488
		}

Sietse Ringers's avatar
Sietse Ringers committed
489
		for _, credreq := range ir.Credentials {
Sietse Ringers's avatar
Sietse Ringers committed
490
			issuer := credreq.CredentialTypeID.IssuerIdentifier()
491 492 493 494 495
			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{}
496
			}
497
			ir.ids.PublicKeys[issuer] = append(ir.ids.PublicKeys[issuer], credreq.KeyCounter)
498 499
		}

500
		ir.ids.join(ir.DisclosureRequest.Identifiers())
Sietse Ringers's avatar
Sietse Ringers committed
501
	}
502
	return ir.ids
Sietse Ringers's avatar
Sietse Ringers committed
503 504
}

Tomas's avatar
Tomas committed
505 506 507 508 509 510 511 512 513 514 515 516 517
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
}

518 519
func (ir *IssuanceRequest) Action() Action { return ActionIssuing }

520 521 522 523 524 525 526
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")
	}
527 528 529 530 531
	for _, cred := range ir.Credentials {
		if cred.Validity.Floor().Before(Timestamp(time.Now())) {
			return errors.New("Expired credential request")
		}
	}
532 533 534
	return nil
}

Sietse Ringers's avatar
Sietse Ringers committed
535 536
// GetNonce returns the nonce of this signature session
// (with the message already hashed into it).
537 538
func (sr *SignatureRequest) GetNonce(timestamp *atum.Timestamp) *big.Int {
	return ASN1ConvertSignatureNonce(sr.Message, sr.BaseRequest.GetNonce(nil), timestamp)
539 540
}

541
func (sr *SignatureRequest) SignatureFromMessage(message interface{}, timestamp *atum.Timestamp) (*SignedMessage, error) {
542
	signature, ok := message.(*Disclosure)
543 544 545 546 547

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

548 549 550 551
	nonce := sr.Nonce
	if nonce == nil {
		nonce = bigZero
	}
552
	return &SignedMessage{
553
		Signature: signature.Proofs,
554
		Indices:   signature.Indices,
555 556
		Nonce:     nonce,
		Context:   sr.GetContext(),
557
		Message:   sr.Message,
558
		Timestamp: timestamp,
559 560 561
	}, nil
}

562 563
func (sr *SignatureRequest) Action() Action { return ActionSigning }

564 565 566 567 568 569 570
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")
	}
571 572
	if len(sr.Disclose) == 0 {
		return errors.New("Signature request had no attributes")
573
	}
574 575 576 577 578
	for _, discon := range sr.Disclose {
		if len(discon) == 0 {
			return errors.New("Empty disjunction")
		}
	}
579 580 581
	return nil
}

582 583 584 585 586
// 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))
}

587 588 589 590
func (t Timestamp) After(u Timestamp) bool {
	return time.Time(t).After(time.Time(u))
}

591 592
// MarshalJSON marshals a timestamp.
func (t *Timestamp) MarshalJSON() ([]byte, error) {
593
	return []byte(t.String()), nil
594 595 596 597 598 599 600 601 602 603 604
}

// 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
605

606 607 608 609 610
// Timestamp implements Stringer.
func (t *Timestamp) String() string {
	return fmt.Sprint(time.Time(*t).Unix())
}

611 612 613 614
func (t *Timestamp) Floor() Timestamp {
	return Timestamp(time.Unix((time.Time(*t).Unix()/ExpiryFactor)*ExpiryFactor, 0))
}

615 616 617 618 619 620 621 622
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
	}
623 624
	bts, err := ioutil.ReadFile(path)
	if err != nil {
625
		return nil, true, errors.New("Could not read scheme manager timestamp")
626
	}
627 628
	ts, err := parseTimestamp(bts)
	return ts, true, err
629 630
}

631
func parseTimestamp(bts []byte) (*Timestamp, error) {
632 633 634 635 636 637
	// 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)
638
	if err != nil {
639
		return nil, err
640
	}
641 642
	ts := Timestamp(time.Unix(str, 0))
	return &ts, nil
643 644
}

Sietse Ringers's avatar
Sietse Ringers committed
645 646 647 648 649 650 651 652
// 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",
		},
653 654 655 656
		Request: &ServiceProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              dr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
657 658 659 660 661 662 663 664 665 666 667
	}
}

// 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",
		},
668 669 670 671
		Request: &SignatureRequestorRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              sr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
672 673 674 675 676 677 678 679 680 681 682
	}
}

// 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",
		},
683 684 685 686
		Request: &IdentityProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              ir,
		},
Sietse Ringers's avatar
Sietse Ringers committed
687 688 689
	}
}

690 691 692 693 694 695 696
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
697 698
}

699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735
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
}
736

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

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

743
// SessionRequest returns an IRMA session object.
744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765
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" {
766 767 768

		return errors.New("Verification jwt has invalid subject")
	}
769
	if time.Time(claims.IssuedAt).After(time.Now()) {
770 771 772 773 774
		return errors.New("Verification jwt not yet valid")
	}
	return nil
}

775 776
func (claims *SignatureRequestorJwt) Valid() error {
	if claims.Type != "signature_request" {
777 778
		return errors.New("Signature jwt has invalid subject")
	}
779
	if time.Time(claims.IssuedAt).After(time.Now()) {
780 781 782 783 784
		return errors.New("Signature jwt not yet valid")
	}
	return nil
}

785 786
func (claims *IdentityProviderJwt) Valid() error {
	if claims.Type != "issue_request" {
787 788
		return errors.New("Issuance jwt has invalid subject")
	}
789
	if time.Time(claims.IssuedAt).After(time.Now()) {
790 791 792 793 794
		return errors.New("Issuance jwt not yet valid")
	}
	return nil
}

795 796 797
func (claims *ServiceProviderJwt) Action() Action { return ActionDisclosing }

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

799
func (claims *IdentityProviderJwt) Action() Action { return ActionIssuing }
800

801
func SignSessionRequest(request SessionRequest, alg jwt.SigningMethod, key interface{}, name string) (string, error) {
802 803 804 805 806 807 808 809 810 811 812
	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)
}
813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828

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)
}
829 830 831 832 833

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