fix: handle tokens where claims of type map is not tagged (#29)

Author: birme

examples/parse.ts

@@ -12,7 +12,7 @@ async function main() {
   const result = await parser.validate(process.argv[2], 'mac', {
     issuer: 'eyevinn'
   });
-  console.log(result);
+  console.dir(result, { depth: null });
 }
 
 main();

src/cat.test.ts

@@ -183,6 +183,38 @@ describe('CAT', () => {
     });
   });
 
+  test('can create a CAT object from a dict with tags', async () => {
+    const cat = new CommonAccessToken({
+      1: 'eyevinn',
+      2: 'jonas',
+      3: 'coap://light.example.com',
+      4: 1444064944,
+      5: 1443944944,
+      6: 1443944944,
+      323: {
+        0: 1,
+        1: 60,
+        2: 10,
+        5: ['Secure', 'HttpOnly', 'Domain=.streaming.a2d.tv']
+      } as any
+    });
+    expect(cat.claims).toEqual({
+      iss: 'eyevinn',
+      sub: 'jonas',
+      aud: 'coap://light.example.com',
+      exp: 1444064944,
+      nbf: 1443944944,
+      iat: 1443944944,
+      catr: {
+        type: 'cookie',
+        expadd: 60,
+        deadline: 10,
+        'cookie-params': ['Secure', 'HttpOnly', 'Domain=.streaming.a2d.tv']
+      },
+      catv: 1
+    });
+  });
+
   test('can determine whether the token should be renewed', async () => {
     const now = Math.floor(Date.now() / 1000);
     const cat = new CommonAccessToken({

src/cat.ts

@@ -208,28 +208,34 @@ function updateMapFromClaims(
   for (const param in dict) {
     const key = claimsToLabels[param] ? claimsToLabels[param] : parseInt(param);
     if (key === claimsToLabels['catu'] && !(dict[param] instanceof Map)) {
-      map.set(key, CommonAccessTokenUri.fromDict(dict[param] as any).payload);
+      map.set(
+        key,
+        CommonAccessTokenUri.fromDictTags(dict[param] as any).payload
+      );
     } else if (
       key === claimsToLabels['catr'] &&
       !(dict[param] instanceof Map)
     ) {
       map.set(
         key,
-        CommonAccessTokenRenewal.fromDict(dict[param] as any).payload
+        CommonAccessTokenRenewal.fromDictTags(dict[param] as any).payload
       );
     } else if (
       key === claimsToLabels['cath'] &&
       !(dict[param] instanceof Map)
     ) {
       map.set(
         key,
-        CommonAccessTokenHeader.fromDict(dict[param] as any).payload
+        CommonAccessTokenHeader.fromDictTags(dict[param] as any).payload
       );
     } else if (
       key === claimsToLabels['catif'] &&
       !(dict[param] instanceof Map)
     ) {
-      map.set(key, CommonAccessTokenIf.fromDict(dict[param] as any).payload);
+      map.set(
+        key,
+        CommonAccessTokenIf.fromDictTags(dict[param] as any).payload
+      );
     } else {
       const value = claimTransform[param]
         ? claimTransform[param](dict[param] as string)

src/cath.ts

@@ -10,6 +10,19 @@ export type CommonAccessTokenHeaderMap = Map<string, MatchMap>;
 export class CommonAccessTokenHeader {
   private cathMap: CommonAccessTokenHeaderMap = new Map();
 
+  public static fromDictTags(dict: { [key: number]: any }) {
+    const newDict: { [key: string]: any } = {};
+    for (const headerTag in dict) {
+      const matchDict: { [key: string]: any } = {};
+      for (const matchTag in dict[headerTag]) {
+        const tag = parseInt(matchTag);
+        matchDict[labelsToMatch[tag]] = dict[headerTag][matchTag];
+      }
+      newDict[labelsToMatch[parseInt(headerTag)]] = matchDict;
+    }
+    return CommonAccessTokenHeader.fromDict(newDict);
+  }
+
   public static fromDict(dict: { [key: string]: any }) {
     const cath = new CommonAccessTokenHeader();
     for (const header in dict) {

src/catif.ts

@@ -14,6 +14,15 @@ export type CatIfDictValue = {
 export class CommonAccessTokenIf {
   private catIfMap: CommonAccessTokenIfMap = new Map();
 
+  public static fromDictTags(dict: { [key: number]: any }) {
+    const newDict: { [key: string]: any } = {};
+    for (const key in dict) {
+      const tag = parseInt(key);
+      newDict[labelsToClaim[tag]] = dict[key];
+    }
+    return CommonAccessTokenIf.fromDict(newDict);
+  }
+
   public static fromDict(dict: { [key: string]: any }) {
     const catif = new CommonAccessTokenIf();
     for (const catIfClaim in dict) {

src/catr.ts

@@ -51,6 +51,19 @@ export type CommonAccessTokenRenewalMap = Map<number, CatrPartValue>;
 export class CommonAccessTokenRenewal {
   private catrMap: CommonAccessTokenRenewalMap = new Map();
 
+  public static fromDictTags(dict: { [key: number]: any }) {
+    const newDict: { [key: string]: any } = {};
+    for (const key in dict) {
+      const tag = parseInt(key);
+      if (labelsToCatrPart[tag] === 'type') {
+        newDict[labelsToCatrPart[tag]] = labelsToCatrRenewalType[dict[key]];
+      } else {
+        newDict[labelsToCatrPart[tag]] = dict[key];
+      }
+    }
+    return CommonAccessTokenRenewal.fromDict(newDict);
+  }
+
   public static fromDict(dict: { [key: string]: any }) {
     const catr = new CommonAccessTokenRenewal();
     for (const catrPart in dict) {

src/catu.ts

@@ -48,6 +48,19 @@ export type CommonAccessTokenUriMap = Map<number, UriPartMap>;
 export class CommonAccessTokenUri {
   private catuMap: CommonAccessTokenUriMap = new Map();
 
+  public static fromDictTags(dict: { [key: number]: any }) {
+    const newDict: { [key: string]: any } = {};
+    for (const uriPartTag in dict) {
+      const matchDict: { [key: string]: any } = {};
+      for (const matchTag in dict[uriPartTag]) {
+        const tag = parseInt(matchTag);
+        matchDict[labelsToMatch[tag]] = dict[uriPartTag][matchTag];
+      }
+      newDict[labelsToUriPart[parseInt(uriPartTag)]] = matchDict;
+    }
+    return CommonAccessTokenUri.fromDict(newDict);
+  }
+
   public static fromDict(dict: { [key: string]: any }) {
     const catu = new CommonAccessTokenUri();
     for (const uriPart in dict) {