func readObjRec(l *list.List, encKey *tao.Keys, id *protected_objects.ObjectIdMessage) (*string, []byte, error) { elem := protected_objects.FindElementById(l, *id.ObjName, *id.ObjEpoch) if elem == nil { return nil, nil, errors.New("object not found") } pObj := elem.Value.(protected_objects.ProtectedObjectMessage) if pObj.ProtectorObjId == nil { // Decrypt root using encKeys. rootKey, err := encKey.CryptingKey.Decrypt(pObj.GetBlob()) if err != nil { return nil, nil, err } str := "key" return &str, rootKey, nil } parentType, parentKey, err := readObjRec(l, encKey, pObj.ProtectorObjId) if err != nil { return nil, nil, err } if *parentType != "key" { return nil, nil, errors.New("internal node with type not key") } obj, err := protected_objects.RecoverProtectedObject(&pObj, parentKey) if err != nil { return nil, nil, err } return obj.ObjType, obj.ObjVal, nil }
func TestAddAndRotate(t *testing.T) { obj_type := "file" status := "active" nb := time.Now() validFor := 365 * 24 * time.Hour na := nb.Add(validFor) protectorKeys := []byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0xa, 0xb, 0xc, 0xd, 0xe, 0xf, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0xa, 0xb, 0xc, 0xd, 0xe, 0xf} obj_1, err := protected_objects.CreateObject("/jlm/file/file1", 1, &obj_type, &status, &nb, &na, nil) if err != nil { t.Fatal("Can't create object") } fmt.Printf("Obj: %s\n", *obj_1.NotBefore) obj_type = "key" obj_2, _ := protected_objects.CreateObject("/jlm/key/key1", 1, &obj_type, &status, &nb, &na, protectorKeys) obj_3, _ := protected_objects.CreateObject("/jlm/key/key2", 1, &obj_type, &status, &nb, &na, protectorKeys) // add them to object list obj_list := list.New() err = protected_objects.AddObject(obj_list, *obj_1) if err != nil { t.Fatal("Can't add object") } _ = protected_objects.AddObject(obj_list, *obj_2) _ = protected_objects.AddObject(obj_list, *obj_3) newkey := []byte{0xff, 0xfe, 0xff, 0xfe, 0xff, 0xfe, 0xff, 0xfe, 0x01, 0x02, 0x01, 0x02, 0x01, 0x02, 0x01, 0x02, 0x07, 0x08, 0x07, 0x08, 0x07, 0x08, 0x07, 0x08, 0xa6, 0xa5, 0xa6, 0xa5, 0xa6, 0xa5, 0xa6, 0xa5} p_obj_1, err := protected_objects.MakeProtectedObject(*obj_1, "/jlm/key/key1", 1, protectorKeys) if err != nil { t.Fatal("Can't make protected object") } if p_obj_1 == nil { t.Fatal("Bad protected object") } p_obj_2, err := protected_objects.MakeProtectedObject(*obj_2, "/jlm/key/key2", 1, protectorKeys) if err != nil { t.Fatal("Can't make protected object") } if p_obj_2 == nil { t.Fatal("Bad protected object") } protected_obj_list := list.New() err = protected_objects.AddProtectedObject(protected_obj_list, *p_obj_1) if err != nil { t.Fatal("Can't add protected object") } err = protected_objects.AddProtectedObject(protected_obj_list, *p_obj_2) if err != nil { t.Fatal("Can't add protected object") } fmt.Printf("\n\n") fmt.Printf("Initial Objects\n") for e := obj_list.Front(); e != nil; e = e.Next() { o := e.Value.(protected_objects.ObjectMessage) protected_objects.PrintObject(&o) } fmt.Printf("\n\nInitial protected objects\n") for e := protected_obj_list.Front(); e != nil; e = e.Next() { o := e.Value.(protected_objects.ProtectedObjectMessage) protected_objects.PrintProtectedObject(&o) } fmt.Printf("\n\n") new_obj, err := AddAndRotateNewKeyEpoch("/jlm/key/key2", "key", "active", "active", nb.String(), na.String(), newkey, obj_list, protected_obj_list) if err != nil { fmt.Printf("Err: %s\n", err) t.Fatal("Can't AddAndRotateNewKeyEpoch") } fmt.Printf("\n\n") fmt.Printf("New key: %s, %d\n", new_obj.ObjId.ObjName, new_obj.ObjId.ObjEpoch) fmt.Printf("\n\n") fmt.Printf("Protected objects\n") for e := protected_obj_list.Front(); e != nil; e = e.Next() { o := e.Value.(protected_objects.ProtectedObjectMessage) protected_objects.PrintProtectedObject(&o) } fmt.Printf("\n\n") fmt.Printf("Objects\n") for e := obj_list.Front(); e != nil; e = e.Next() { o := e.Value.(protected_objects.ObjectMessage) protected_objects.PrintObject(&o) } fmt.Printf("\n\n") // Check we can open protected object with new protector protected_kids := protected_objects.FindProtectedObjects(protected_obj_list, *new_obj.ObjId.ObjName, *new_obj.ObjId.ObjEpoch) if err != nil { t.Fatal("Can't FindProtected kids") } e := protected_kids.Front() o := e.Value.(protected_objects.ProtectedObjectMessage) obj, err := protected_objects.RecoverProtectedObject(&o, new_obj.ObjVal) if err != nil || obj == nil { t.Fatal("Can't recover first kid") } fmt.Printf("\n\nRecovered:\n") protected_objects.PrintObject(obj) }