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/irmago/internal/fs"
Sietse Ringers's avatar
Sietse Ringers committed
17
)
18

19 20 21 22
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"
23
	LDContextRevocationRequest = "https://irma.app/ld/request/revocation/v1"
24 25
)

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

30 31
	// Revocation instructs the client to include nonrevocation zero-knowledge proofs for the
	// specified credential types.
32
	Revocation []CredentialTypeIdentifier `json:"revocation,omitempty"`
33 34

	// Set by the IRMA server during the session
35 36 37 38
	Context           *big.Int                                         `json:"context,omitempty"`
	Nonce             *big.Int                                         `json:"nonce,omitempty"`
	ProtocolVersion   *ProtocolVersion                                 `json:"protocolVersion,omitempty"`
	RevocationUpdates map[CredentialTypeIdentifier][]*RevocationRecord `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
// An AttributeCon is only satisfied if all of its containing attribute requests are satisfied.
type AttributeCon []AttributeRequest
50

51 52
// An AttributeDisCon is satisfied if at least one of its containing AttributeCon is satisfied.
type AttributeDisCon []AttributeCon
53

54 55
// AttributeConDisCon is only satisfied if all of the containing AttributeDisCon are satisfied.
type AttributeConDisCon []AttributeDisCon
56

57 58
// A DisclosureRequest is a request to disclose certain attributes. Construct new instances using
// NewDisclosureRequest().
59
type DisclosureRequest struct {
60
	BaseRequest
61

62 63
	Disclose AttributeConDisCon       `json:"disclose,omitempty"`
	Labels   map[int]TranslatedString `json:"labels,omitempty"`
64 65
}

66 67
// A SignatureRequest is a a request to sign a message with certain attributes. Construct new
// instances using NewSignatureRequest().
68
type SignatureRequest struct {
69
	DisclosureRequest
70
	Message string `json:"message"`
71 72
}

Sietse Ringers's avatar
Sietse Ringers committed
73
// An IssuanceRequest is a request to issue certain credentials,
74 75
// optionally also asking for certain attributes to be simultaneously disclosed. Construct new
// instances using NewIssuanceRequest().
76
type IssuanceRequest struct {
77
	DisclosureRequest
78
	Credentials []*CredentialRequest `json:"credentials"`
Tomas's avatar
Tomas committed
79 80 81

	// Derived data
	CredentialInfoList        CredentialInfoList `json:",omitempty"`
82
	RemovalCredentialInfoList CredentialInfoList `json:",omitempty"`
83 84
}

Sietse Ringers's avatar
Sietse Ringers committed
85 86
// A CredentialRequest contains the attributes and metadata of a credential
// that will be issued in an IssuanceRequest.
87
type CredentialRequest struct {
88 89
	Validity         *Timestamp               `json:"validity,omitempty"`
	KeyCounter       int                      `json:"keyCounter,omitempty"`
90 91
	CredentialTypeID CredentialTypeIdentifier `json:"credential"`
	Attributes       map[string]string        `json:"attributes"`
92
	RevocationKey    string                   `json:"revocationKey,omitempty"`
Sietse Ringers's avatar
Sietse Ringers committed
93 94
}

95 96 97 98
// SessionRequest instances contain all information the irmaclient needs to perform an IRMA session.
type SessionRequest interface {
	Validator
	Base() *BaseRequest
99
	GetNonce(timestamp *atum.Timestamp) *big.Int
100 101 102
	Disclosure() *DisclosureRequest
	Identifiers() *IrmaIdentifierSet
	Action() Action
103
	Legacy() (SessionRequest, error)
104 105 106 107 108
}

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

Sietse Ringers's avatar
Sietse Ringers committed
109 110 111 112 113 114 115
// ServerJwt contains standard JWT fields.
type ServerJwt struct {
	Type       string    `json:"sub"`
	ServerName string    `json:"iss"`
	IssuedAt   Timestamp `json:"iat"`
}

116 117
// RequestorBaseRequest contains fields present in all RequestorRequest types
// with which the requestor configures an IRMA session.
118
type RequestorBaseRequest struct {
Sietse Ringers's avatar
Sietse Ringers committed
119 120
	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
121
	CallbackURL       string `json:"callbackUrl,omitempty"` // URL to post session result to
122 123
}

124 125
// 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.
126 127 128
type RequestorRequest interface {
	Validator
	SessionRequest() SessionRequest
129
	Base() RequestorBaseRequest
130 131
}

Sietse Ringers's avatar
Sietse Ringers committed
132 133
// A ServiceProviderRequest contains a disclosure request.
type ServiceProviderRequest struct {
134
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
135 136 137 138 139
	Request *DisclosureRequest `json:"request"`
}

// A SignatureRequestorRequest contains a signing request.
type SignatureRequestorRequest struct {
140
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
141 142 143 144 145
	Request *SignatureRequest `json:"request"`
}

// An IdentityProviderRequest contains an issuance request.
type IdentityProviderRequest struct {
146
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
147 148 149 150 151 152
	Request *IssuanceRequest `json:"request"`
}

// ServiceProviderJwt is a requestor JWT for a disclosure session.
type ServiceProviderJwt struct {
	ServerJwt
153
	Request *ServiceProviderRequest `json:"sprequest"`
Sietse Ringers's avatar
Sietse Ringers committed
154 155 156 157 158
}

// SignatureRequestorJwt is a requestor JWT for a signing session.
type SignatureRequestorJwt struct {
	ServerJwt
159
	Request *SignatureRequestorRequest `json:"absrequest"`
Sietse Ringers's avatar
Sietse Ringers committed
160 161 162 163 164
}

// IdentityProviderJwt is a requestor JWT for issuance session.
type IdentityProviderJwt struct {
	ServerJwt
165 166 167
	Request *IdentityProviderRequest `json:"iprequest"`
}

168 169 170 171 172
type RevocationJwt struct {
	ServerJwt
	Request *RevocationRequest `json:"revrequest"`
}

173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190
// 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 {
191 192 193
	Type    AttributeTypeIdentifier `json:"type"`
	Value   *string                 `json:"value,omitempty"`
	NotNull bool                    `json:"notNull,omitempty"`
194 195
}

196 197 198 199 200 201 202 203 204 205 206 207 208
type RevocationRequest struct {
	LDContext      string                   `json:"@context,omitempty"`
	CredentialType CredentialTypeIdentifier `json:"type"`
	Key            string                   `json:"key"`
}

func (r *RevocationRequest) Validate() error {
	if r.LDContext == LDContextRevocationRequest {
		return errors.New("not a revocation request")
	}
	return nil
}

209 210 211 212 213
var (
	bigZero = big.NewInt(0)
	bigOne  = big.NewInt(1)
)

214 215 216 217
func (b *BaseRequest) Legacy() bool {
	return b.legacy
}

218 219 220
func (b *BaseRequest) GetContext() *big.Int {
	if b.Context == nil {
		return bigOne
221
	}
222
	return b.Context
223 224
}

225
func (b *BaseRequest) GetNonce(*atum.Timestamp) *big.Int {
226 227
	if b.Nonce == nil {
		return bigZero
228
	}
229
	return b.Nonce
230 231
}

232 233 234 235 236 237
func (b *BaseRequest) RequestsRevocation(id CredentialTypeIdentifier) bool {
	// If the requestor wants us to include a nonrevocation proof,
	// it will have sent us the latest revocation update messages
	return len(b.RevocationUpdates) > 0 && len(b.RevocationUpdates[id]) > 0
}

238 239
const revocationUpdateCount = 5

240 241 242
// CredentialTypes returns an array of all credential types occuring in this conjunction.
func (c AttributeCon) CredentialTypes() []CredentialTypeIdentifier {
	var result []CredentialTypeIdentifier
243

244
	for _, attr := range c {
245 246 247 248
		typ := attr.Type.CredentialTypeIdentifier()
		if len(result) == 0 || result[len(result)-1] != typ {
			result = append(result, typ)
		}
249
	}
250

251
	return result
252 253
}

254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271
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
}

272 273
// AttributeRequest synonym with default JSON (un)marshaler
type jsonAttributeRequest AttributeRequest
274

275 276
func (ar *AttributeRequest) UnmarshalJSON(bts []byte) error {
	var s AttributeTypeIdentifier
277

278 279 280
	// first try to parse as JSON string into s
	if err := json.Unmarshal(bts, &s); err == nil {
		*ar = AttributeRequest{Type: s}
281 282 283
		return nil
	}

284 285
	return json.Unmarshal(bts, (*jsonAttributeRequest)(ar))
}
286

287
func (ar *AttributeRequest) MarshalJSON() ([]byte, error) {
288
	if !ar.NotNull && ar.Value == nil {
289
		return json.Marshal(ar.Type)
290
	}
291
	return json.Marshal((*jsonAttributeRequest)(ar))
292 293
}

294
// Satisfy indicates whether the given attribute type and value satisfies this AttributeRequest.
295
func (ar *AttributeRequest) Satisfy(attr AttributeTypeIdentifier, val *string) bool {
296
	return ar.Type == attr &&
297
		(!ar.NotNull || val != nil) &&
298
		(ar.Value == nil || (val != nil && *ar.Value == *val))
Sietse Ringers's avatar
Sietse Ringers committed
299 300
}

301 302
// 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.
303
func (c AttributeCon) Satisfy(proofs gabi.ProofList, indices []*DisclosedAttributeIndex, conf *Configuration) (bool, []*DisclosedAttribute, error) {
304
	if len(indices) < len(c) {
305
		return false, nil, nil
306 307
	}
	attrs := make([]*DisclosedAttribute, 0, len(c))
308 309 310
	if len(c) == 0 {
		return true, attrs, nil
	}
311 312 313 314 315

	for j := range c {
		index := indices[j]
		attr, val, err := extractAttribute(proofs, index, conf)
		if err != nil {
316
			return false, nil, err
317 318
		}
		if !c[j].Satisfy(attr.Identifier, val) {
319
			return false, nil, nil
320 321 322
		}
		attrs = append(attrs, attr)
	}
323
	return true, attrs, nil
324 325
}

326 327 328 329 330 331 332 333 334 335 336 337 338
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
}

339 340
// 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.
341
func (dc AttributeDisCon) Satisfy(proofs gabi.ProofList, indices []*DisclosedAttributeIndex, conf *Configuration) (bool, []*DisclosedAttribute, error) {
342
	for _, con := range dc {
343 344 345
		satisfied, attrs, err := con.Satisfy(proofs, indices, conf)
		if satisfied || err != nil {
			return true, attrs, err
346 347
		}
	}
348
	return false, nil, nil
349 350
}

351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369
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
}

370 371
// Satisfy returns true if each of the contained AttributeDisCon is satisfied by the specified disclosure.
// If so it also returns the disclosed attributes.
372 373 374 375 376 377 378 379
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 {
380
		satisfied, attrs, err := discon.Satisfy(disclosure.Proofs, disclosure.Indices[i], conf)
381 382 383
		if err != nil {
			return false, nil, err
		}
384
		if satisfied {
385 386 387 388 389 390 391 392
			list[i] = attrs
		} else {
			complete = false
			list[i] = nil
		}
	}

	return complete, list, nil
393 394
}

395 396 397 398 399 400 401 402 403 404 405 406
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
407 408
}

409 410 411 412 413 414 415
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{
416
		BaseRequest: BaseRequest{LDContext: LDContextDisclosureRequest},
417 418 419 420 421 422 423 424 425 426
		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...)
427
	dr.LDContext = LDContextSignatureRequest
428
	return &SignatureRequest{
429
		DisclosureRequest: *dr,
430 431 432 433 434 435
		Message:           message,
	}
}

func NewIssuanceRequest(creds []*CredentialRequest, attrs ...AttributeTypeIdentifier) *IssuanceRequest {
	dr := NewDisclosureRequest(attrs...)
436
	dr.LDContext = LDContextIssuanceRequest
437
	return &IssuanceRequest{
438
		DisclosureRequest: *dr,
439 440 441 442 443 444 445 446
		Credentials:       creds,
	}
}

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

447 448 449 450 451 452
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
453
		AttributeTypes:  map[AttributeTypeIdentifier]struct{}{},
454 455 456 457 458 459 460
	}

	_ = 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
461
		ids.AttributeTypes[attr] = struct{}{}
462 463 464 465 466 467
		return nil
	})

	return ids
}

468
func (dr *DisclosureRequest) Identifiers() *IrmaIdentifierSet {
469
	if dr.ids == nil {
470
		dr.ids = dr.identifiers()
471
	}
472
	return dr.ids
473 474 475 476 477 478 479 480 481
}

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

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

func (dr *DisclosureRequest) Validate() error {
482
	if dr.LDContext != LDContextDisclosureRequest {
483 484 485 486 487
		return errors.New("Not a disclosure request")
	}
	if len(dr.Disclose) == 0 {
		return errors.New("Disclosure request had no attributes")
	}
488
	var err error
489
	for _, discon := range dr.Disclose {
490 491
		if err = discon.Validate(); err != nil {
			return err
492 493
		}
	}
494 495
	return nil
}
Sietse Ringers's avatar
Sietse Ringers committed
496

497 498
func (cr *CredentialRequest) Info(conf *Configuration, metadataVersion byte) (*CredentialInfo, error) {
	list, err := cr.AttributeList(conf, metadataVersion)
499 500 501
	if err != nil {
		return nil, err
	}
502
	return NewCredentialInfo(list.Ints, conf), nil
503 504
}

505 506 507 508
// 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 {
509
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
Sietse Ringers's avatar
Sietse Ringers committed
510
	if credtype == nil {
511
		return errors.New("Credential request of unknown credential type")
Sietse Ringers's avatar
Sietse Ringers committed
512
	}
513 514 515 516 517

	// Check that there are no attributes in the credential request that aren't
	// in the credential descriptor.
	for crName := range cr.Attributes {
		found := false
518
		for _, ad := range credtype.AttributeTypes {
519 520 521 522 523 524
			if ad.ID == crName {
				found = true
				break
			}
		}
		if !found {
525
			return errors.New("Credential request contains unknown attribute")
526
		}
Sietse Ringers's avatar
Sietse Ringers committed
527 528
	}

529
	for _, attrtype := range credtype.AttributeTypes {
530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553
		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
554
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
555
	attrs := make([]*big.Int, len(credtype.AttributeTypes)+1)
Sietse Ringers's avatar
Sietse Ringers committed
556
	attrs[0] = meta.Int
557
	for i, attrtype := range credtype.AttributeTypes {
558
		attrs[i+1] = new(big.Int)
Sietse Ringers's avatar
Sietse Ringers committed
559
		if str, present := cr.Attributes[attrtype.ID]; present {
560
			// Set attribute to str << 1 + 1
561
			attrs[i+1].SetBytes([]byte(str))
562 563 564 565
			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
566 567 568
		}
	}

569
	return NewAttributeListFromInts(attrs, conf), nil
Sietse Ringers's avatar
Sietse Ringers committed
570 571
}

Sietse Ringers's avatar
Sietse Ringers committed
572
func (ir *IssuanceRequest) Identifiers() *IrmaIdentifierSet {
573 574
	if ir.ids == nil {
		ir.ids = &IrmaIdentifierSet{
575 576 577
			SchemeManagers:  map[SchemeManagerIdentifier]struct{}{},
			Issuers:         map[IssuerIdentifier]struct{}{},
			CredentialTypes: map[CredentialTypeIdentifier]struct{}{},
Leon's avatar
Leon committed
578
			AttributeTypes:  map[AttributeTypeIdentifier]struct{}{},
579
			PublicKeys:      map[IssuerIdentifier][]int{},
Sietse Ringers's avatar
Sietse Ringers committed
580 581
		}

Sietse Ringers's avatar
Sietse Ringers committed
582
		for _, credreq := range ir.Credentials {
Sietse Ringers's avatar
Sietse Ringers committed
583
			issuer := credreq.CredentialTypeID.IssuerIdentifier()
584 585
			ir.ids.SchemeManagers[issuer.SchemeManagerIdentifier()] = struct{}{}
			ir.ids.Issuers[issuer] = struct{}{}
Leon's avatar
Leon committed
586 587 588 589 590
			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{}{}
			}
591 592
			if ir.ids.PublicKeys[issuer] == nil {
				ir.ids.PublicKeys[issuer] = []int{}
593
			}
594
			ir.ids.PublicKeys[issuer] = append(ir.ids.PublicKeys[issuer], credreq.KeyCounter)
595 596
		}

597
		ir.ids.join(ir.DisclosureRequest.identifiers())
Sietse Ringers's avatar
Sietse Ringers committed
598
	}
599
	return ir.ids
Sietse Ringers's avatar
Sietse Ringers committed
600 601
}

Tomas's avatar
Tomas committed
602 603 604 605 606 607 608 609 610 611 612 613 614
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
}

615 616
func (ir *IssuanceRequest) Action() Action { return ActionIssuing }

617
func (ir *IssuanceRequest) Validate() error {
618
	if ir.LDContext != LDContextIssuanceRequest {
619 620 621 622 623
		return errors.New("Not an issuance request")
	}
	if len(ir.Credentials) == 0 {
		return errors.New("Empty issuance request")
	}
624
	for _, cred := range ir.Credentials {
625
		if cred.Validity != nil && cred.Validity.Floor().Before(Timestamp(time.Now())) {
626 627 628
			return errors.New("Expired credential request")
		}
	}
629 630 631 632 633 634
	var err error
	for _, discon := range ir.Disclose {
		if err = discon.Validate(); err != nil {
			return err
		}
	}
635 636 637
	return nil
}

Sietse Ringers's avatar
Sietse Ringers committed
638 639
// GetNonce returns the nonce of this signature session
// (with the message already hashed into it).
640 641
func (sr *SignatureRequest) GetNonce(timestamp *atum.Timestamp) *big.Int {
	return ASN1ConvertSignatureNonce(sr.Message, sr.BaseRequest.GetNonce(nil), timestamp)
642 643
}

644
func (sr *SignatureRequest) SignatureFromMessage(message interface{}, timestamp *atum.Timestamp) (*SignedMessage, error) {
645
	signature, ok := message.(*Disclosure)
646 647 648 649 650

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

651 652 653 654
	nonce := sr.Nonce
	if nonce == nil {
		nonce = bigZero
	}
655
	return &SignedMessage{
656
		LDContext: LDContextSignedMessage,
657
		Signature: signature.Proofs,
658
		Indices:   signature.Indices,
659 660
		Nonce:     nonce,
		Context:   sr.GetContext(),
661
		Message:   sr.Message,
662
		Timestamp: timestamp,
663 664 665
	}, nil
}

666 667
func (sr *SignatureRequest) Action() Action { return ActionSigning }

668
func (sr *SignatureRequest) Validate() error {
669
	if sr.LDContext != LDContextSignatureRequest {
670 671 672 673 674
		return errors.New("Not a signature request")
	}
	if sr.Message == "" {
		return errors.New("Signature request had empty message")
	}
675 676
	if len(sr.Disclose) == 0 {
		return errors.New("Signature request had no attributes")
677
	}
678
	var err error
679
	for _, discon := range sr.Disclose {
680 681
		if err = discon.Validate(); err != nil {
			return err
682 683
		}
	}
684 685 686
	return nil
}

687 688 689 690 691
// 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))
}

692 693 694 695
func (t Timestamp) After(u Timestamp) bool {
	return time.Time(t).After(time.Time(u))
}

696 697 698 699 700
// To check whether Timestamp is uninitialized
func (t Timestamp) IsZero() bool {
	return time.Time(t).IsZero()
}

701 702 703 704 705 706 707 708 709 710 711 712 713
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
}

714 715
// MarshalJSON marshals a timestamp.
func (t *Timestamp) MarshalJSON() ([]byte, error) {
716
	return []byte(t.String()), nil
717 718 719 720 721 722 723 724 725 726 727
}

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

729 730 731 732 733
// Timestamp implements Stringer.
func (t *Timestamp) String() string {
	return fmt.Sprint(time.Time(*t).Unix())
}

734 735 736 737
func (t *Timestamp) Floor() Timestamp {
	return Timestamp(time.Unix((time.Time(*t).Unix()/ExpiryFactor)*ExpiryFactor, 0))
}

738 739 740 741 742 743 744 745
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
	}
746 747
	bts, err := ioutil.ReadFile(path)
	if err != nil {
748
		return nil, true, errors.New("Could not read scheme manager timestamp")
749
	}
750 751
	ts, err := parseTimestamp(bts)
	return ts, true, err
752 753
}

754
func parseTimestamp(bts []byte) (*Timestamp, error) {
755 756 757 758 759 760
	// 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)
761
	if err != nil {
762
		return nil, err
763
	}
764 765
	ts := Timestamp(time.Unix(str, 0))
	return &ts, nil
766 767
}

Sietse Ringers's avatar
Sietse Ringers committed
768 769 770 771 772 773 774 775
// 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",
		},
776 777 778 779
		Request: &ServiceProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              dr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
780 781 782 783 784 785 786 787 788 789 790
	}
}

// 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",
		},
791 792 793 794
		Request: &SignatureRequestorRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              sr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
795 796 797 798 799 800 801 802 803 804 805
	}
}

// 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",
		},
806 807 808 809
		Request: &IdentityProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              ir,
		},
Sietse Ringers's avatar
Sietse Ringers committed
810 811 812
	}
}

813 814 815 816 817 818 819
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
820 821
}

822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858
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
}
859

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

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

866
// SessionRequest returns an IRMA session object.
867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888
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" {
889 890 891

		return errors.New("Verification jwt has invalid subject")
	}
892
	if time.Time(claims.IssuedAt).After(time.Now()) {
893 894 895 896 897
		return errors.New("Verification jwt not yet valid")
	}
	return nil
}

898 899
func (claims *SignatureRequestorJwt) Valid() error {
	if claims.Type != "signature_request" {
900 901
		return errors.New("Signature jwt has invalid subject")
	}
902
	if time.Time(claims.IssuedAt).After(time.Now()) {
903 904 905 906 907
		return errors.New("Signature jwt not yet valid")
	}
	return nil
}

908 909
func (claims *IdentityProviderJwt) Valid() error {
	if claims.Type != "issue_request" {
910 911
		return errors.New("Issuance jwt has invalid subject")
	}
912
	if time.Time(claims.IssuedAt).After(time.Now()) {
913 914 915 916 917
		return errors.New("Issuance jwt not yet valid")
	}
	return nil
}

918 919 920 921 922 923 924
func (claims *RevocationJwt) Valid() error {
	if time.Time(claims.IssuedAt).After(time.Now()) {
		return errors.New("Signature jwt not yet valid")
	}
	return nil
}

925 926 927
func (claims *ServiceProviderJwt) Action() Action { return ActionDisclosing }

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

929
func (claims *IdentityProviderJwt) Action() Action { return ActionIssuing }
930

931
func SignSessionRequest(request SessionRequest, alg jwt.SigningMethod, key interface{}, name string) (string, error) {
932 933 934 935 936 937 938 939 940 941 942
	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)
}
943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958

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)
}
959 960 961 962 963

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