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

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

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

20 21 22 23 24 25
const (
	LDContextDisclosureRequest = "https://irma.app/ld/request/disclosure/v2"
	LDContextSignatureRequest  = "https://irma.app/ld/request/signature/v2"
	LDContextIssuanceRequest   = "https://irma.app/ld/request/issuance/v2"
)

26 27
// BaseRequest contains the context and nonce for an IRMA session.
type BaseRequest struct {
28
	LDContext string `json:"@context,omitempty"`
29

30 31 32 33 34 35 36 37 38
	// Revocation instructs the client to include nonrevocation zero-knowledge proofs for the
	// specified credential types.
	Revocation RevocationSet `json:"revocation,omitempty"`

	// Set by the IRMA server during the session
	Context           *big.Int                                         `json:"context,omitempty"`
	Nonce             *big.Int                                         `json:"nonce,omitempty"`
	ProtocolVersion   *ProtocolVersion                                 `json:"protocolVersion,omitempty"`
	RevocationUpdates map[CredentialTypeIdentifier][]revocation.Record `json:"revocationUpdates,omitempty"`
39

40 41 42 43
	ids *IrmaIdentifierSet // cache for Identifiers() method

	legacy bool   // Whether or not this was deserialized from a legacy (pre-condiscon) request
	Type   Action `json:"type,omitempty"` // Session type, only used in legacy code
44 45

	ClientReturnURL string `json:"clientReturnUrl,omitempty"` // URL to proceed to when IRMA session is completed
Sietse Ringers's avatar
Sietse Ringers committed
46 47
}

48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
type RevocationSet map[CredentialTypeIdentifier]struct{}

func (r *RevocationSet) MarshalJSON() ([]byte, error) {
	if r == nil {
		return json.Marshal(nil)
	}
	l := make([]CredentialTypeIdentifier, 0, len(*r))
	for c := range *r {
		l = append(l, c)
	}
	return json.Marshal(l)
}

func (r *RevocationSet) UnmarshalJSON(bts []byte) error {
	var l []CredentialTypeIdentifier
	if err := json.Unmarshal(bts, &l); err != nil {
		return err
	}
	if *r == nil {
		*r = RevocationSet{}
	}
	for _, c := range l {
		(*r)[c] = struct{}{}
	}
	return nil
}

75 76
// An AttributeCon is only satisfied if all of its containing attribute requests are satisfied.
type AttributeCon []AttributeRequest
77

78 79
// An AttributeDisCon is satisfied if at least one of its containing AttributeCon is satisfied.
type AttributeDisCon []AttributeCon
80

81 82
// AttributeConDisCon is only satisfied if all of the containing AttributeDisCon are satisfied.
type AttributeConDisCon []AttributeDisCon
83

84 85
// A DisclosureRequest is a request to disclose certain attributes. Construct new instances using
// NewDisclosureRequest().
86
type DisclosureRequest struct {
87
	BaseRequest
88

89 90
	Disclose AttributeConDisCon       `json:"disclose,omitempty"`
	Labels   map[int]TranslatedString `json:"labels,omitempty"`
91 92
}

93 94
// A SignatureRequest is a a request to sign a message with certain attributes. Construct new
// instances using NewSignatureRequest().
95
type SignatureRequest struct {
96
	DisclosureRequest
97
	Message string `json:"message"`
98 99
}

Sietse Ringers's avatar
Sietse Ringers committed
100
// An IssuanceRequest is a request to issue certain credentials,
101 102
// optionally also asking for certain attributes to be simultaneously disclosed. Construct new
// instances using NewIssuanceRequest().
103
type IssuanceRequest struct {
104
	DisclosureRequest
105
	Credentials []*CredentialRequest `json:"credentials"`
Tomas's avatar
Tomas committed
106 107 108

	// Derived data
	CredentialInfoList        CredentialInfoList `json:",omitempty"`
109
	RemovalCredentialInfoList CredentialInfoList `json:",omitempty"`
110 111
}

Sietse Ringers's avatar
Sietse Ringers committed
112 113
// A CredentialRequest contains the attributes and metadata of a credential
// that will be issued in an IssuanceRequest.
114
type CredentialRequest struct {
115 116
	Validity         *Timestamp               `json:"validity,omitempty"`
	KeyCounter       int                      `json:"keyCounter,omitempty"`
117 118
	CredentialTypeID CredentialTypeIdentifier `json:"credential"`
	Attributes       map[string]string        `json:"attributes"`
119
	RevocationKey    string                   `json:"revocationKey,omitempty"`
Sietse Ringers's avatar
Sietse Ringers committed
120 121
}

122 123 124 125
// SessionRequest instances contain all information the irmaclient needs to perform an IRMA session.
type SessionRequest interface {
	Validator
	Base() *BaseRequest
126
	GetNonce(timestamp *atum.Timestamp) *big.Int
127 128 129
	Disclosure() *DisclosureRequest
	Identifiers() *IrmaIdentifierSet
	Action() Action
130
	Legacy() (SessionRequest, error)
131 132 133 134 135
}

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

Sietse Ringers's avatar
Sietse Ringers committed
136 137 138 139 140 141 142
// ServerJwt contains standard JWT fields.
type ServerJwt struct {
	Type       string    `json:"sub"`
	ServerName string    `json:"iss"`
	IssuedAt   Timestamp `json:"iat"`
}

143 144
// RequestorBaseRequest contains fields present in all RequestorRequest types
// with which the requestor configures an IRMA session.
145
type RequestorBaseRequest struct {
Sietse Ringers's avatar
Sietse Ringers committed
146 147
	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
148
	CallbackURL       string `json:"callbackUrl,omitempty"` // URL to post session result to
149 150
}

151 152
// 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.
153 154 155
type RequestorRequest interface {
	Validator
	SessionRequest() SessionRequest
156
	Base() RequestorBaseRequest
157 158
}

Sietse Ringers's avatar
Sietse Ringers committed
159 160
// A ServiceProviderRequest contains a disclosure request.
type ServiceProviderRequest struct {
161
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
162 163 164 165 166
	Request *DisclosureRequest `json:"request"`
}

// A SignatureRequestorRequest contains a signing request.
type SignatureRequestorRequest struct {
167
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
168 169 170 171 172
	Request *SignatureRequest `json:"request"`
}

// An IdentityProviderRequest contains an issuance request.
type IdentityProviderRequest struct {
173
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
174 175 176 177 178 179
	Request *IssuanceRequest `json:"request"`
}

// ServiceProviderJwt is a requestor JWT for a disclosure session.
type ServiceProviderJwt struct {
	ServerJwt
180
	Request *ServiceProviderRequest `json:"sprequest"`
Sietse Ringers's avatar
Sietse Ringers committed
181 182 183 184 185
}

// SignatureRequestorJwt is a requestor JWT for a signing session.
type SignatureRequestorJwt struct {
	ServerJwt
186
	Request *SignatureRequestorRequest `json:"absrequest"`
Sietse Ringers's avatar
Sietse Ringers committed
187 188 189 190 191
}

// IdentityProviderJwt is a requestor JWT for issuance session.
type IdentityProviderJwt struct {
	ServerJwt
192 193 194
	Request *IdentityProviderRequest `json:"iprequest"`
}

195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212
// 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 {
213 214 215
	Type    AttributeTypeIdentifier `json:"type"`
	Value   *string                 `json:"value,omitempty"`
	NotNull bool                    `json:"notNull,omitempty"`
216 217 218 219 220 221 222
}

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

223 224 225 226
func (b *BaseRequest) Legacy() bool {
	return b.legacy
}

227 228 229
func (b *BaseRequest) GetContext() *big.Int {
	if b.Context == nil {
		return bigOne
230
	}
231
	return b.Context
232 233
}

234
func (b *BaseRequest) GetNonce(*atum.Timestamp) *big.Int {
235 236
	if b.Nonce == nil {
		return bigZero
237
	}
238
	return b.Nonce
239 240
}

241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260
const revocationUpdateCount = 5

func (b *BaseRequest) SetRevocationRecords(conf *Configuration) error {
	if len(b.Revocation) == 0 {
		return nil
	}
	b.RevocationUpdates = make(map[CredentialTypeIdentifier][]revocation.Record, len(b.Revocation))
	for credid := range b.Revocation {
		db, err := conf.RevocationDB(credid)
		if err != nil {
			return err
		}
		b.RevocationUpdates[credid], err = db.LatestRecords(revocationUpdateCount)
		if err != nil {
			return err
		}
	}
	return nil
}

261 262 263
// CredentialTypes returns an array of all credential types occuring in this conjunction.
func (c AttributeCon) CredentialTypes() []CredentialTypeIdentifier {
	var result []CredentialTypeIdentifier
264

265
	for _, attr := range c {
266 267 268 269
		typ := attr.Type.CredentialTypeIdentifier()
		if len(result) == 0 || result[len(result)-1] != typ {
			result = append(result, typ)
		}
270
	}
271

272
	return result
273 274
}

275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292
func (c AttributeCon) Validate() error {
	// Unlike AttributeDisCon, we don't have to check here that the current instance is of length 0,
	// as that is actually a valid conjunction: one that specifies that the containing disjunction
	// is optional.

	credtypes := map[CredentialTypeIdentifier]struct{}{}
	var last CredentialTypeIdentifier
	for _, attr := range c {
		typ := attr.Type.CredentialTypeIdentifier()
		if _, contains := credtypes[typ]; contains && last != typ {
			return errors.New("Within inner conjunctions, attributes from the same credential type must be adjacent")
		}
		last = typ
		credtypes[typ] = struct{}{}
	}
	return nil
}

293 294
// AttributeRequest synonym with default JSON (un)marshaler
type jsonAttributeRequest AttributeRequest
295

296 297
func (ar *AttributeRequest) UnmarshalJSON(bts []byte) error {
	var s AttributeTypeIdentifier
298

299 300 301
	// first try to parse as JSON string into s
	if err := json.Unmarshal(bts, &s); err == nil {
		*ar = AttributeRequest{Type: s}
302 303 304
		return nil
	}

305 306
	return json.Unmarshal(bts, (*jsonAttributeRequest)(ar))
}
307

308
func (ar *AttributeRequest) MarshalJSON() ([]byte, error) {
309
	if !ar.NotNull && ar.Value == nil {
310
		return json.Marshal(ar.Type)
311
	}
312
	return json.Marshal((*jsonAttributeRequest)(ar))
313 314
}

315
// Satisfy indicates whether the given attribute type and value satisfies this AttributeRequest.
316
func (ar *AttributeRequest) Satisfy(attr AttributeTypeIdentifier, val *string) bool {
317
	return ar.Type == attr &&
318
		(!ar.NotNull || val != nil) &&
319
		(ar.Value == nil || (val != nil && *ar.Value == *val))
Sietse Ringers's avatar
Sietse Ringers committed
320 321
}

322 323
// Satisfy returns if each of the attributes specified by proofs and indices satisfies each of
// the contained AttributeRequests's. If so it also returns a list of the disclosed attribute values.
324
func (c AttributeCon) Satisfy(proofs gabi.ProofList, indices []*DisclosedAttributeIndex, conf *Configuration) (bool, []*DisclosedAttribute, error) {
325
	if len(indices) < len(c) {
326
		return false, nil, nil
327 328
	}
	attrs := make([]*DisclosedAttribute, 0, len(c))
329 330 331
	if len(c) == 0 {
		return true, attrs, nil
	}
332 333 334 335 336

	for j := range c {
		index := indices[j]
		attr, val, err := extractAttribute(proofs, index, conf)
		if err != nil {
337
			return false, nil, err
338 339
		}
		if !c[j].Satisfy(attr.Identifier, val) {
340
			return false, nil, nil
341 342 343
		}
		attrs = append(attrs, attr)
	}
344
	return true, attrs, nil
345 346
}

347 348 349 350 351 352 353 354 355 356 357 358 359
func (dc AttributeDisCon) Validate() error {
	if len(dc) == 0 {
		return errors.New("Empty disjunction")
	}
	var err error
	for _, con := range dc {
		if err = con.Validate(); err != nil {
			return err
		}
	}
	return nil
}

360 361
// Satisfy returns true if the attributes specified by proofs and indices satisfies any one of the
// contained AttributeCon's. If so it also returns a list of the disclosed attribute values.
362
func (dc AttributeDisCon) Satisfy(proofs gabi.ProofList, indices []*DisclosedAttributeIndex, conf *Configuration) (bool, []*DisclosedAttribute, error) {
363
	for _, con := range dc {
364 365 366
		satisfied, attrs, err := con.Satisfy(proofs, indices, conf)
		if satisfied || err != nil {
			return true, attrs, err
367 368
		}
	}
369
	return false, nil, nil
370 371
}

372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390
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
}

391 392
// Satisfy returns true if each of the contained AttributeDisCon is satisfied by the specified disclosure.
// If so it also returns the disclosed attributes.
393 394 395 396 397 398 399 400
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 {
401
		satisfied, attrs, err := discon.Satisfy(disclosure.Proofs, disclosure.Indices[i], conf)
402 403 404
		if err != nil {
			return false, nil, err
		}
405
		if satisfied {
406 407 408 409 410 411 412 413
			list[i] = attrs
		} else {
			complete = false
			list[i] = nil
		}
	}

	return complete, list, nil
414 415
}

416 417 418 419 420 421 422 423 424 425 426 427
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
428 429
}

430 431 432 433 434 435 436
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{
437
		BaseRequest: BaseRequest{LDContext: LDContextDisclosureRequest},
438 439 440 441 442 443 444 445 446 447
		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...)
448
	dr.LDContext = LDContextSignatureRequest
449
	return &SignatureRequest{
450
		DisclosureRequest: *dr,
451 452 453 454 455 456
		Message:           message,
	}
}

func NewIssuanceRequest(creds []*CredentialRequest, attrs ...AttributeTypeIdentifier) *IssuanceRequest {
	dr := NewDisclosureRequest(attrs...)
457
	dr.LDContext = LDContextIssuanceRequest
458
	return &IssuanceRequest{
459
		DisclosureRequest: *dr,
460 461 462 463 464 465 466 467
		Credentials:       creds,
	}
}

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

468 469 470 471 472 473
func (dr *DisclosureRequest) identifiers() *IrmaIdentifierSet {
	ids := &IrmaIdentifierSet{
		SchemeManagers:  map[SchemeManagerIdentifier]struct{}{},
		Issuers:         map[IssuerIdentifier]struct{}{},
		CredentialTypes: map[CredentialTypeIdentifier]struct{}{},
		PublicKeys:      map[IssuerIdentifier][]int{},
Leon's avatar
Leon committed
474
		AttributeTypes:  map[AttributeTypeIdentifier]struct{}{},
475 476 477 478 479 480 481
	}

	_ = dr.Disclose.Iterate(func(a *AttributeRequest) error {
		attr := a.Type
		ids.SchemeManagers[attr.CredentialTypeIdentifier().IssuerIdentifier().SchemeManagerIdentifier()] = struct{}{}
		ids.Issuers[attr.CredentialTypeIdentifier().IssuerIdentifier()] = struct{}{}
		ids.CredentialTypes[attr.CredentialTypeIdentifier()] = struct{}{}
Leon's avatar
Leon committed
482
		ids.AttributeTypes[attr] = struct{}{}
483 484 485 486 487 488
		return nil
	})

	return ids
}

489
func (dr *DisclosureRequest) Identifiers() *IrmaIdentifierSet {
490
	if dr.ids == nil {
491
		dr.ids = dr.identifiers()
492
	}
493
	return dr.ids
494 495 496 497 498 499 500 501 502
}

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

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

func (dr *DisclosureRequest) Validate() error {
503
	if dr.LDContext != LDContextDisclosureRequest {
504 505 506 507 508
		return errors.New("Not a disclosure request")
	}
	if len(dr.Disclose) == 0 {
		return errors.New("Disclosure request had no attributes")
	}
509
	var err error
510
	for _, discon := range dr.Disclose {
511 512
		if err = discon.Validate(); err != nil {
			return err
513 514
		}
	}
515 516
	return nil
}
Sietse Ringers's avatar
Sietse Ringers committed
517

518 519
func (cr *CredentialRequest) Info(conf *Configuration, metadataVersion byte) (*CredentialInfo, error) {
	list, err := cr.AttributeList(conf, metadataVersion)
520 521 522
	if err != nil {
		return nil, err
	}
523
	return NewCredentialInfo(list.Ints, conf), nil
524 525
}

526 527 528 529
// 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 {
530
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
Sietse Ringers's avatar
Sietse Ringers committed
531
	if credtype == nil {
532
		return errors.New("Credential request of unknown credential type")
Sietse Ringers's avatar
Sietse Ringers committed
533
	}
534 535 536 537 538

	// Check that there are no attributes in the credential request that aren't
	// in the credential descriptor.
	for crName := range cr.Attributes {
		found := false
539
		for _, ad := range credtype.AttributeTypes {
540 541 542 543 544 545
			if ad.ID == crName {
				found = true
				break
			}
		}
		if !found {
546
			return errors.New("Credential request contains unknown attribute")
547
		}
Sietse Ringers's avatar
Sietse Ringers committed
548 549
	}

550
	for _, attrtype := range credtype.AttributeTypes {
551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574
		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
575
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
576
	attrs := make([]*big.Int, len(credtype.AttributeTypes)+1)
Sietse Ringers's avatar
Sietse Ringers committed
577
	attrs[0] = meta.Int
578
	for i, attrtype := range credtype.AttributeTypes {
579
		attrs[i+1] = new(big.Int)
Sietse Ringers's avatar
Sietse Ringers committed
580
		if str, present := cr.Attributes[attrtype.ID]; present {
581
			// Set attribute to str << 1 + 1
582
			attrs[i+1].SetBytes([]byte(str))
583 584 585 586
			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
587 588 589
		}
	}

590
	return NewAttributeListFromInts(attrs, conf), nil
Sietse Ringers's avatar
Sietse Ringers committed
591 592
}

Sietse Ringers's avatar
Sietse Ringers committed
593
func (ir *IssuanceRequest) Identifiers() *IrmaIdentifierSet {
594 595
	if ir.ids == nil {
		ir.ids = &IrmaIdentifierSet{
596 597 598
			SchemeManagers:  map[SchemeManagerIdentifier]struct{}{},
			Issuers:         map[IssuerIdentifier]struct{}{},
			CredentialTypes: map[CredentialTypeIdentifier]struct{}{},
Leon's avatar
Leon committed
599
			AttributeTypes:  map[AttributeTypeIdentifier]struct{}{},
600
			PublicKeys:      map[IssuerIdentifier][]int{},
Sietse Ringers's avatar
Sietse Ringers committed
601 602
		}

Sietse Ringers's avatar
Sietse Ringers committed
603
		for _, credreq := range ir.Credentials {
Sietse Ringers's avatar
Sietse Ringers committed
604
			issuer := credreq.CredentialTypeID.IssuerIdentifier()
605 606
			ir.ids.SchemeManagers[issuer.SchemeManagerIdentifier()] = struct{}{}
			ir.ids.Issuers[issuer] = struct{}{}
Leon's avatar
Leon committed
607 608 609 610 611
			credID := credreq.CredentialTypeID
			ir.ids.CredentialTypes[credID] = struct{}{}
			for attr, _ := range credreq.Attributes { // this is kind of ugly
				ir.ids.AttributeTypes[NewAttributeTypeIdentifier(credID.String()+"."+attr)] = struct{}{}
			}
612 613
			if ir.ids.PublicKeys[issuer] == nil {
				ir.ids.PublicKeys[issuer] = []int{}
614
			}
615
			ir.ids.PublicKeys[issuer] = append(ir.ids.PublicKeys[issuer], credreq.KeyCounter)
616 617
		}

618
		ir.ids.join(ir.DisclosureRequest.identifiers())
Sietse Ringers's avatar
Sietse Ringers committed
619
	}
620
	return ir.ids
Sietse Ringers's avatar
Sietse Ringers committed
621 622
}

Tomas's avatar
Tomas committed
623 624 625 626 627 628 629 630 631 632 633 634 635
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
}

636 637
func (ir *IssuanceRequest) Action() Action { return ActionIssuing }

638
func (ir *IssuanceRequest) Validate() error {
639
	if ir.LDContext != LDContextIssuanceRequest {
640 641 642 643 644
		return errors.New("Not an issuance request")
	}
	if len(ir.Credentials) == 0 {
		return errors.New("Empty issuance request")
	}
645
	for _, cred := range ir.Credentials {
646
		if cred.Validity != nil && cred.Validity.Floor().Before(Timestamp(time.Now())) {
647 648 649
			return errors.New("Expired credential request")
		}
	}
650 651 652 653 654 655
	var err error
	for _, discon := range ir.Disclose {
		if err = discon.Validate(); err != nil {
			return err
		}
	}
656 657 658
	return nil
}

Sietse Ringers's avatar
Sietse Ringers committed
659 660
// GetNonce returns the nonce of this signature session
// (with the message already hashed into it).
661 662
func (sr *SignatureRequest) GetNonce(timestamp *atum.Timestamp) *big.Int {
	return ASN1ConvertSignatureNonce(sr.Message, sr.BaseRequest.GetNonce(nil), timestamp)
663 664
}

665
func (sr *SignatureRequest) SignatureFromMessage(message interface{}, timestamp *atum.Timestamp) (*SignedMessage, error) {
666
	signature, ok := message.(*Disclosure)
667 668 669 670 671

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

672 673 674 675
	nonce := sr.Nonce
	if nonce == nil {
		nonce = bigZero
	}
676
	return &SignedMessage{
677
		LDContext: LDContextSignedMessage,
678
		Signature: signature.Proofs,
679
		Indices:   signature.Indices,
680 681
		Nonce:     nonce,
		Context:   sr.GetContext(),
682
		Message:   sr.Message,
683
		Timestamp: timestamp,
684 685 686
	}, nil
}

687 688
func (sr *SignatureRequest) Action() Action { return ActionSigning }

689
func (sr *SignatureRequest) Validate() error {
690
	if sr.LDContext != LDContextSignatureRequest {
691 692 693 694 695
		return errors.New("Not a signature request")
	}
	if sr.Message == "" {
		return errors.New("Signature request had empty message")
	}
696 697
	if len(sr.Disclose) == 0 {
		return errors.New("Signature request had no attributes")
698
	}
699
	var err error
700
	for _, discon := range sr.Disclose {
701 702
		if err = discon.Validate(); err != nil {
			return err
703 704
		}
	}
705 706 707
	return nil
}

708 709 710 711 712
// 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))
}

713 714 715 716
func (t Timestamp) After(u Timestamp) bool {
	return time.Time(t).After(time.Time(u))
}

717 718 719 720 721
// To check whether Timestamp is uninitialized
func (t Timestamp) IsZero() bool {
	return time.Time(t).IsZero()
}

722 723 724 725 726 727 728 729 730 731 732 733 734
func (t *Timestamp) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
	return e.EncodeElement(t.String(), start)
}

func (t *Timestamp) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
	var ts int64
	if err := d.DecodeElement(&ts, &start); err != nil {
		return err
	}
	*t = Timestamp(time.Unix(ts, 0))
	return nil
}

735 736
// MarshalJSON marshals a timestamp.
func (t *Timestamp) MarshalJSON() ([]byte, error) {
737
	return []byte(t.String()), nil
738 739 740 741 742 743 744 745 746 747 748
}

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

750 751 752 753 754
// Timestamp implements Stringer.
func (t *Timestamp) String() string {
	return fmt.Sprint(time.Time(*t).Unix())
}

755 756 757 758
func (t *Timestamp) Floor() Timestamp {
	return Timestamp(time.Unix((time.Time(*t).Unix()/ExpiryFactor)*ExpiryFactor, 0))
}

759 760 761 762 763 764 765 766
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
	}
767 768
	bts, err := ioutil.ReadFile(path)
	if err != nil {
769
		return nil, true, errors.New("Could not read scheme manager timestamp")
770
	}
771 772
	ts, err := parseTimestamp(bts)
	return ts, true, err
773 774
}

775
func parseTimestamp(bts []byte) (*Timestamp, error) {
776 777 778 779 780 781
	// 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)
782
	if err != nil {
783
		return nil, err
784
	}
785 786
	ts := Timestamp(time.Unix(str, 0))
	return &ts, nil
787 788
}

Sietse Ringers's avatar
Sietse Ringers committed
789 790 791 792 793 794 795 796
// 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",
		},
797 798 799 800
		Request: &ServiceProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              dr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
801 802 803 804 805 806 807 808 809 810 811
	}
}

// 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",
		},
812 813 814 815
		Request: &SignatureRequestorRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              sr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
816 817 818 819 820 821 822 823 824 825 826
	}
}

// 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",
		},
827 828 829 830
		Request: &IdentityProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              ir,
		},
Sietse Ringers's avatar
Sietse Ringers committed
831 832 833
	}
}

834 835 836 837 838 839 840
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
841 842
}

843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879
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
}
880

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

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

887
// SessionRequest returns an IRMA session object.
888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909
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" {
910 911 912

		return errors.New("Verification jwt has invalid subject")
	}
913
	if time.Time(claims.IssuedAt).After(time.Now()) {
914 915 916 917 918
		return errors.New("Verification jwt not yet valid")
	}
	return nil
}

919 920
func (claims *SignatureRequestorJwt) Valid() error {
	if claims.Type != "signature_request" {
921 922
		return errors.New("Signature jwt has invalid subject")
	}
923
	if time.Time(claims.IssuedAt).After(time.Now()) {
924 925 926 927 928
		return errors.New("Signature jwt not yet valid")
	}
	return nil
}

929 930
func (claims *IdentityProviderJwt) Valid() error {
	if claims.Type != "issue_request" {
931 932
		return errors.New("Issuance jwt has invalid subject")
	}
933
	if time.Time(claims.IssuedAt).After(time.Now()) {
934 935 936 937 938
		return errors.New("Issuance jwt not yet valid")
	}
	return nil
}

939 940 941
func (claims *ServiceProviderJwt) Action() Action { return ActionDisclosing }

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

943
func (claims *IdentityProviderJwt) Action() Action { return ActionIssuing }
944

945
func SignSessionRequest(request SessionRequest, alg jwt.SigningMethod, key interface{}, name string) (string, error) {
946 947 948 949 950 951 952 953 954 955 956
	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)
}
957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972

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)
}
973 974 975 976 977

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