requests.go 20.4 KB
Newer Older
1
package irma
2 3 4

import (
	"fmt"
5
	"io/ioutil"
6 7 8
	"strconv"
	"time"

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

16 17
// BaseRequest contains the context and nonce for an IRMA session.
type BaseRequest struct {
18 19
	Context *big.Int `json:"context,omitempty"`
	Nonce   *big.Int `json:"nonce,omitempty"`
20
	Type    Action   `json:"type"`
21

22 23 24
	Candidates [][]*AttributeIdentifier `json:"-"`
	Choice     *DisclosureChoice        `json:"-"`
	Ids        *IrmaIdentifierSet       `json:"-"`
25

26
	Version *ProtocolVersion `json:"protocolVersion,omitempty"`
Sietse Ringers's avatar
Sietse Ringers committed
27 28
}

29
func (sr *BaseRequest) SetCandidates(candidates [][]*AttributeIdentifier) {
30 31 32
	sr.Candidates = candidates
}

33
// DisclosureChoice returns the attributes to be disclosed in this session.
34
func (sr *BaseRequest) DisclosureChoice() *DisclosureChoice {
35
	return sr.Choice
Sietse Ringers's avatar
Sietse Ringers committed
36 37
}

38
// SetDisclosureChoice sets the attributes to be disclosed in this session.
39
func (sr *BaseRequest) SetDisclosureChoice(choice *DisclosureChoice) {
40
	sr.Choice = choice
41 42
}

43
// ...
44
func (sr *BaseRequest) SetVersion(v *ProtocolVersion) {
45
	sr.Version = v
46 47 48
}

// ...
49
func (sr *BaseRequest) GetVersion() *ProtocolVersion {
50
	return sr.Version
51 52
}

Sietse Ringers's avatar
Sietse Ringers committed
53
// A DisclosureRequest is a request to disclose certain attributes.
54
type DisclosureRequest struct {
55
	BaseRequest
56 57 58
	Content AttributeDisjunctionList `json:"content"`
}

Sietse Ringers's avatar
Sietse Ringers committed
59
// A SignatureRequest is a a request to sign a message with certain attributes.
60 61
type SignatureRequest struct {
	DisclosureRequest
62 63 64
	Message string `json:"message"`

	// Session state
65
	Timestamp *atum.Timestamp `json:"-"`
66 67
}

Sietse Ringers's avatar
Sietse Ringers committed
68 69
// An IssuanceRequest is a request to issue certain credentials,
// optionally also asking for certain attributes to be simultaneously disclosed.
70
type IssuanceRequest struct {
71
	BaseRequest
Tomas's avatar
Tomas committed
72 73 74 75 76
	Credentials []*CredentialRequest     `json:"credentials"`
	Disclose    AttributeDisjunctionList `json:"disclose"`

	// Derived data
	CredentialInfoList        CredentialInfoList `json:",omitempty"`
77
	RemovalCredentialInfoList CredentialInfoList `json:",omitempty"`
78 79
}

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

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

96 97
// RequestorBaseRequest contains fields present in all RequestorRequest types
// with which the requestor configures an IRMA session.
98
type RequestorBaseRequest struct {
Sietse Ringers's avatar
Sietse Ringers committed
99 100 101
	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
102 103
}

104 105
// 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.
106 107 108
type RequestorRequest interface {
	Validator
	SessionRequest() SessionRequest
109
	Base() RequestorBaseRequest
110 111
}

Sietse Ringers's avatar
Sietse Ringers committed
112 113
// A ServiceProviderRequest contains a disclosure request.
type ServiceProviderRequest struct {
114
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
115 116 117 118 119
	Request *DisclosureRequest `json:"request"`
}

// A SignatureRequestorRequest contains a signing request.
type SignatureRequestorRequest struct {
120
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
121 122 123 124 125
	Request *SignatureRequest `json:"request"`
}

// An IdentityProviderRequest contains an issuance request.
type IdentityProviderRequest struct {
126
	RequestorBaseRequest
Sietse Ringers's avatar
Sietse Ringers committed
127 128 129 130 131 132
	Request *IssuanceRequest `json:"request"`
}

// ServiceProviderJwt is a requestor JWT for a disclosure session.
type ServiceProviderJwt struct {
	ServerJwt
133
	Request *ServiceProviderRequest `json:"sprequest"`
Sietse Ringers's avatar
Sietse Ringers committed
134 135 136 137 138
}

// SignatureRequestorJwt is a requestor JWT for a signing session.
type SignatureRequestorJwt struct {
	ServerJwt
139
	Request *SignatureRequestorRequest `json:"absrequest"`
Sietse Ringers's avatar
Sietse Ringers committed
140 141 142 143 144
}

// IdentityProviderJwt is a requestor JWT for issuance session.
type IdentityProviderJwt struct {
	ServerJwt
145 146 147 148 149 150 151 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
	Request *IdentityProviderRequest `json:"iprequest"`
}

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

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
Sietse Ringers's avatar
Sietse Ringers committed
179 180
}

181 182 183 184 185 186 187 188 189 190 191 192
func (r *ServiceProviderRequest) Base() RequestorBaseRequest {
	return r.RequestorBaseRequest
}

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

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

193
// SessionRequest instances contain all information the irmaclient needs to perform an IRMA session.
194
type SessionRequest interface {
195
	Validator
196 197 198 199
	GetNonce() *big.Int
	SetNonce(*big.Int)
	GetContext() *big.Int
	SetContext(*big.Int)
200
	GetVersion() *ProtocolVersion
201
	SetVersion(*ProtocolVersion)
202
	ToDisclose() AttributeDisjunctionList
203 204
	DisclosureChoice() *DisclosureChoice
	SetDisclosureChoice(choice *DisclosureChoice)
205
	SetCandidates(candidates [][]*AttributeIdentifier)
206
	Identifiers() *IrmaIdentifierSet
207
	Action() Action
208 209
}

Sietse Ringers's avatar
Sietse Ringers committed
210 211 212
// Timestamp is a time.Time that marshals to Unix timestamps.
type Timestamp time.Time

213 214
func (cr *CredentialRequest) Info(conf *Configuration, metadataVersion byte) (*CredentialInfo, error) {
	list, err := cr.AttributeList(conf, metadataVersion)
215 216 217
	if err != nil {
		return nil, err
	}
218
	return NewCredentialInfo(list.Ints, conf), nil
219 220
}

221 222 223 224
// 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 {
225
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
Sietse Ringers's avatar
Sietse Ringers committed
226
	if credtype == nil {
227
		return errors.New("Credential request of unknown credential type")
Sietse Ringers's avatar
Sietse Ringers committed
228
	}
229 230 231 232 233

	// Check that there are no attributes in the credential request that aren't
	// in the credential descriptor.
	for crName := range cr.Attributes {
		found := false
234
		for _, ad := range credtype.AttributeTypes {
235 236 237 238 239 240
			if ad.ID == crName {
				found = true
				break
			}
		}
		if !found {
241
			return errors.New("Credential request contaiins unknown attribute")
242
		}
Sietse Ringers's avatar
Sietse Ringers committed
243 244
	}

245
	for _, attrtype := range credtype.AttributeTypes {
246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269
		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
270
	credtype := conf.CredentialTypes[cr.CredentialTypeID]
271
	attrs := make([]*big.Int, len(credtype.AttributeTypes)+1)
Sietse Ringers's avatar
Sietse Ringers committed
272
	attrs[0] = meta.Int
273
	for i, attrtype := range credtype.AttributeTypes {
274
		attrs[i+1] = new(big.Int)
Sietse Ringers's avatar
Sietse Ringers committed
275
		if str, present := cr.Attributes[attrtype.ID]; present {
276
			// Set attribute to str << 1 + 1
277
			attrs[i+1].SetBytes([]byte(str))
278 279 280 281
			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
282 283 284
		}
	}

285
	return NewAttributeListFromInts(attrs, conf), nil
Sietse Ringers's avatar
Sietse Ringers committed
286 287
}

Sietse Ringers's avatar
Sietse Ringers committed
288
func (ir *IssuanceRequest) Identifiers() *IrmaIdentifierSet {
289 290
	if ir.Ids == nil {
		ir.Ids = &IrmaIdentifierSet{
291 292 293 294
			SchemeManagers:  map[SchemeManagerIdentifier]struct{}{},
			Issuers:         map[IssuerIdentifier]struct{}{},
			CredentialTypes: map[CredentialTypeIdentifier]struct{}{},
			PublicKeys:      map[IssuerIdentifier][]int{},
Sietse Ringers's avatar
Sietse Ringers committed
295 296
		}

Sietse Ringers's avatar
Sietse Ringers committed
297
		for _, credreq := range ir.Credentials {
Sietse Ringers's avatar
Sietse Ringers committed
298
			issuer := credreq.CredentialTypeID.IssuerIdentifier()
299 300
			ir.Ids.SchemeManagers[issuer.SchemeManagerIdentifier()] = struct{}{}
			ir.Ids.Issuers[issuer] = struct{}{}
301
			ir.Ids.CredentialTypes[credreq.CredentialTypeID] = struct{}{}
302 303
			if ir.Ids.PublicKeys[issuer] == nil {
				ir.Ids.PublicKeys[issuer] = []int{}
304
			}
305
			ir.Ids.PublicKeys[issuer] = append(ir.Ids.PublicKeys[issuer], credreq.KeyCounter)
306 307
		}

Sietse Ringers's avatar
Sietse Ringers committed
308
		for _, disjunction := range ir.Disclose {
309
			for _, attr := range disjunction.Attributes {
Sietse Ringers's avatar
Sietse Ringers committed
310 311 312 313 314 315
				var cti CredentialTypeIdentifier
				if !attr.IsCredential() {
					cti = attr.CredentialTypeIdentifier()
				} else {
					cti = NewCredentialTypeIdentifier(attr.String())
				}
316 317 318
				ir.Ids.SchemeManagers[cti.IssuerIdentifier().SchemeManagerIdentifier()] = struct{}{}
				ir.Ids.Issuers[cti.IssuerIdentifier()] = struct{}{}
				ir.Ids.CredentialTypes[cti] = struct{}{}
319
			}
Sietse Ringers's avatar
Sietse Ringers committed
320 321
		}
	}
322
	return ir.Ids
Sietse Ringers's avatar
Sietse Ringers committed
323 324
}

Sietse Ringers's avatar
Sietse Ringers committed
325
// ToDisclose returns the attributes that must be disclosed in this issuance session.
326 327 328 329 330 331 332
func (ir *IssuanceRequest) ToDisclose() AttributeDisjunctionList {
	if ir.Disclose == nil {
		return AttributeDisjunctionList{}
	}

	return ir.Disclose
}
Sietse Ringers's avatar
Sietse Ringers committed
333

Tomas's avatar
Tomas committed
334 335 336 337 338 339 340 341 342 343 344 345 346
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
}

Sietse Ringers's avatar
Sietse Ringers committed
347 348 349 350 351 352 353 354 355 356 357 358
// GetContext returns the context of this session.
func (ir *IssuanceRequest) GetContext() *big.Int { return ir.Context }

// SetContext sets the context of this session.
func (ir *IssuanceRequest) SetContext(context *big.Int) { ir.Context = context }

// GetNonce returns the nonce of this session.
func (ir *IssuanceRequest) GetNonce() *big.Int { return ir.Nonce }

// SetNonce sets the nonce of this session.
func (ir *IssuanceRequest) SetNonce(nonce *big.Int) { ir.Nonce = nonce }

359 360
func (ir *IssuanceRequest) Action() Action { return ActionIssuing }

361 362 363 364 365 366 367 368 369 370
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")
	}
	return nil
}

371
func (dr *DisclosureRequest) Identifiers() *IrmaIdentifierSet {
372 373
	if dr.Ids == nil {
		dr.Ids = &IrmaIdentifierSet{
374 375 376 377
			SchemeManagers:  map[SchemeManagerIdentifier]struct{}{},
			Issuers:         map[IssuerIdentifier]struct{}{},
			CredentialTypes: map[CredentialTypeIdentifier]struct{}{},
			PublicKeys:      map[IssuerIdentifier][]int{},
Sietse Ringers's avatar
Sietse Ringers committed
378
		}
379 380
		for _, disjunction := range dr.Content {
			for _, attr := range disjunction.Attributes {
381 382 383
				dr.Ids.SchemeManagers[attr.CredentialTypeIdentifier().IssuerIdentifier().SchemeManagerIdentifier()] = struct{}{}
				dr.Ids.Issuers[attr.CredentialTypeIdentifier().IssuerIdentifier()] = struct{}{}
				dr.Ids.CredentialTypes[attr.CredentialTypeIdentifier()] = struct{}{}
384
			}
Sietse Ringers's avatar
Sietse Ringers committed
385 386
		}
	}
387
	return dr.Ids
Sietse Ringers's avatar
Sietse Ringers committed
388 389
}

Sietse Ringers's avatar
Sietse Ringers committed
390 391
// ToDisclose returns the attributes to be disclosed in this session.
func (dr *DisclosureRequest) ToDisclose() AttributeDisjunctionList { return dr.Content }
Sietse Ringers's avatar
Sietse Ringers committed
392 393 394 395 396 397 398 399 400 401 402 403 404

// GetContext returns the context of this session.
func (dr *DisclosureRequest) GetContext() *big.Int { return dr.Context }

// SetContext sets the context of this session.
func (dr *DisclosureRequest) SetContext(context *big.Int) { dr.Context = context }

// GetNonce returns the nonce of this session.
func (dr *DisclosureRequest) GetNonce() *big.Int { return dr.Nonce }

// SetNonce sets the nonce of this session.
func (dr *DisclosureRequest) SetNonce(nonce *big.Int) { dr.Nonce = nonce }

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

407 408 409 410 411 412 413 414 415 416 417 418 419 420 421
func (dr *DisclosureRequest) Validate() error {
	if dr.Type != ActionDisclosing {
		return errors.New("Not a disclosure request")
	}
	if len(dr.Content) == 0 {
		return errors.New("Disclosure request had no attributes")
	}
	for _, disjunction := range dr.Content {
		if len(disjunction.Attributes) == 0 {
			return errors.New("Disclosure request had an empty disjunction")
		}
	}
	return nil
}

Sietse Ringers's avatar
Sietse Ringers committed
422 423
// GetNonce returns the nonce of this signature session
// (with the message already hashed into it).
424
func (sr *SignatureRequest) GetNonce() *big.Int {
425
	return ASN1ConvertSignatureNonce(sr.Message, sr.Nonce, sr.Timestamp)
426 427
}

428
func (sr *SignatureRequest) SignatureFromMessage(message interface{}) (*SignedMessage, error) {
429
	signature, ok := message.(*Disclosure)
430 431 432 433 434

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

435
	return &SignedMessage{
436
		Signature: signature.Proofs,
437
		Indices:   signature.Indices,
438 439 440
		Nonce:     sr.Nonce,
		Context:   sr.Context,
		Message:   sr.Message,
441
		Timestamp: sr.Timestamp,
442 443 444
	}, nil
}

445 446
func (sr *SignatureRequest) Action() Action { return ActionSigning }

447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464
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")
	}
	if len(sr.Content) == 0 {
		return errors.New("Disclosure request had no attributes")
	}
	for _, disjunction := range sr.Content {
		if len(disjunction.Attributes) == 0 {
			return errors.New("Disclosure request had an empty disjunction")
		}
	}
	return nil
}

465 466 467 468 469
// 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))
}

470 471 472 473
func (t Timestamp) After(u Timestamp) bool {
	return time.Time(t).After(time.Time(u))
}

474 475
// MarshalJSON marshals a timestamp.
func (t *Timestamp) MarshalJSON() ([]byte, error) {
476
	return []byte(t.String()), nil
477 478 479 480 481 482 483 484 485 486 487
}

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

489 490 491 492 493
// Timestamp implements Stringer.
func (t *Timestamp) String() string {
	return fmt.Sprint(time.Time(*t).Unix())
}

494 495 496 497 498 499 500 501
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
	}
502 503
	bts, err := ioutil.ReadFile(path)
	if err != nil {
504
		return nil, true, errors.New("Could not read scheme manager timestamp")
505
	}
506 507
	ts, err := parseTimestamp(bts)
	return ts, true, err
508 509
}

510
func parseTimestamp(bts []byte) (*Timestamp, error) {
511 512 513 514 515 516
	// 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)
517
	if err != nil {
518
		return nil, err
519
	}
520 521
	ts := Timestamp(time.Unix(str, 0))
	return &ts, nil
522 523
}

Sietse Ringers's avatar
Sietse Ringers committed
524 525 526 527 528 529 530 531
// 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",
		},
532 533 534 535
		Request: &ServiceProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              dr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
536 537 538 539 540 541 542 543 544 545 546
	}
}

// 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",
		},
547 548 549 550
		Request: &SignatureRequestorRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              sr,
		},
Sietse Ringers's avatar
Sietse Ringers committed
551 552 553 554 555 556 557 558 559 560 561
	}
}

// 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",
		},
562 563 564 565
		Request: &IdentityProviderRequest{
			RequestorBaseRequest: RequestorBaseRequest{ResultJwtValidity: 120},
			Request:              ir,
		},
Sietse Ringers's avatar
Sietse Ringers committed
566 567 568 569 570
	}
}

// A RequestorJwt contains an IRMA session object.
type RequestorJwt interface {
571
	Action() Action
572
	RequestorRequest() RequestorRequest
573
	SessionRequest() SessionRequest
574
	Requestor() string
575
	Valid() error
576
	Sign(jwt.SigningMethod, interface{}) (string, error)
Sietse Ringers's avatar
Sietse Ringers committed
577 578
}

579 580
func (jwt *ServerJwt) Requestor() string { return jwt.ServerName }

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

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

587
// SessionRequest returns an IRMA session object.
588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609
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" {
610 611 612

		return errors.New("Verification jwt has invalid subject")
	}
613
	if time.Time(claims.IssuedAt).After(time.Now()) {
614 615 616 617 618
		return errors.New("Verification jwt not yet valid")
	}
	return nil
}

619 620
func (claims *SignatureRequestorJwt) Valid() error {
	if claims.Type != "signature_request" {
621 622
		return errors.New("Signature jwt has invalid subject")
	}
623
	if time.Time(claims.IssuedAt).After(time.Now()) {
624 625 626 627 628
		return errors.New("Signature jwt not yet valid")
	}
	return nil
}

629 630
func (claims *IdentityProviderJwt) Valid() error {
	if claims.Type != "issue_request" {
631 632
		return errors.New("Issuance jwt has invalid subject")
	}
633
	if time.Time(claims.IssuedAt).After(time.Now()) {
634 635 636 637 638
		return errors.New("Issuance jwt not yet valid")
	}
	return nil
}

639 640 641
func (claims *ServiceProviderJwt) Action() Action { return ActionDisclosing }

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

643
func (claims *IdentityProviderJwt) Action() Action { return ActionIssuing }
644

645
func SignSessionRequest(request SessionRequest, alg jwt.SigningMethod, key interface{}, name string) (string, error) {
646 647 648 649 650 651 652 653 654 655 656
	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)
}
657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672

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)
}