1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
13 "git.arvados.org/arvados.git/lib/config"
14 "git.arvados.org/arvados.git/lib/controller/rpc"
15 "git.arvados.org/arvados.git/sdk/go/arvados"
16 "git.arvados.org/arvados.git/sdk/go/arvadostest"
17 "git.arvados.org/arvados.git/sdk/go/auth"
18 "git.arvados.org/arvados.git/sdk/go/ctxlog"
19 check "gopkg.in/check.v1"
22 var _ = check.Suite(&CollectionSuite{})
24 type CollectionSuite struct {
25 cluster *arvados.Cluster
27 railsSpy *arvadostest.Proxy
30 func (s *CollectionSuite) TearDownSuite(c *check.C) {
31 // Undo any changes/additions to the user database so they
32 // don't affect subsequent tests.
33 arvadostest.ResetEnv()
34 c.Check(arvados.NewClientFromEnv().RequestAndDecode(nil, "POST", "database/reset", nil, nil), check.IsNil)
37 func (s *CollectionSuite) SetUpTest(c *check.C) {
38 cfg, err := config.NewLoader(nil, ctxlog.TestLogger(c)).Load()
39 c.Assert(err, check.IsNil)
40 s.cluster, err = cfg.GetCluster("")
41 c.Assert(err, check.IsNil)
42 s.localdb = NewConn(s.cluster)
43 s.railsSpy = arvadostest.NewProxy(c, s.cluster.Services.RailsAPI)
44 *s.localdb.railsProxy = *rpc.NewConn(s.cluster.ClusterID, s.railsSpy.URL, true, rpc.PassthroughTokenProvider)
47 func (s *CollectionSuite) TearDownTest(c *check.C) {
51 func (s *CollectionSuite) TestSignatures(c *check.C) {
52 ctx := auth.NewContext(context.Background(), &auth.Credentials{Tokens: []string{arvadostest.ActiveTokenV2}})
54 resp, err := s.localdb.CollectionGet(ctx, arvados.GetOptions{UUID: arvadostest.FooCollection})
55 c.Check(err, check.IsNil)
56 c.Check(resp.ManifestText, check.Matches, `(?ms).* acbd[^ ]*\+3\+A[0-9a-f]+@[0-9a-f]+ 0:.*`)
57 s.checkSignatureExpiry(c, resp.ManifestText, time.Hour*24*7*2)
59 resp, err = s.localdb.CollectionGet(ctx, arvados.GetOptions{UUID: arvadostest.FooCollection, Select: []string{"manifest_text"}})
60 c.Check(err, check.IsNil)
61 c.Check(resp.ManifestText, check.Matches, `(?ms).* acbd[^ ]*\+3\+A[0-9a-f]+@[0-9a-f]+ 0:.*`)
63 lresp, err := s.localdb.CollectionList(ctx, arvados.ListOptions{Limit: -1, Filters: []arvados.Filter{{"uuid", "=", arvadostest.FooCollection}}})
64 c.Check(err, check.IsNil)
65 if c.Check(lresp.Items, check.HasLen, 1) {
66 c.Check(lresp.Items[0].UUID, check.Equals, arvadostest.FooCollection)
67 c.Check(lresp.Items[0].ManifestText, check.Equals, "")
68 c.Check(lresp.Items[0].UnsignedManifestText, check.Equals, "")
71 lresp, err = s.localdb.CollectionList(ctx, arvados.ListOptions{Limit: -1, Filters: []arvados.Filter{{"uuid", "=", arvadostest.FooCollection}}, Select: []string{"manifest_text"}})
72 c.Check(err, check.IsNil)
73 if c.Check(lresp.Items, check.HasLen, 1) {
74 c.Check(lresp.Items[0].ManifestText, check.Matches, `(?ms).* acbd[^ ]*\+3\+A[0-9a-f]+@[0-9a-f]+ 0:.*`)
75 c.Check(lresp.Items[0].UnsignedManifestText, check.Equals, "")
78 lresp, err = s.localdb.CollectionList(ctx, arvados.ListOptions{Limit: -1, Filters: []arvados.Filter{{"uuid", "=", arvadostest.FooCollection}}, Select: []string{"unsigned_manifest_text"}})
79 c.Check(err, check.IsNil)
80 if c.Check(lresp.Items, check.HasLen, 1) {
81 c.Check(lresp.Items[0].ManifestText, check.Equals, "")
82 c.Check(lresp.Items[0].UnsignedManifestText, check.Matches, `(?ms).* acbd[^ ]*\+3 0:.*`)
85 // early trash date causes lower signature TTL
86 trashed, err := s.localdb.CollectionCreate(ctx, arvados.CreateOptions{
87 Attrs: map[string]interface{}{
88 "manifest_text": ". d41d8cd98f00b204e9800998ecf8427e+0 0:0:foo\n",
89 "trash_at": time.Now().UTC().Add(time.Hour),
91 c.Assert(err, check.IsNil)
92 resp, err = s.localdb.CollectionGet(ctx, arvados.GetOptions{UUID: trashed.UUID})
93 c.Assert(err, check.IsNil)
94 s.checkSignatureExpiry(c, resp.ManifestText, time.Hour)
96 // distant future trash date does not cause higher signature TTL
97 trashed, err = s.localdb.CollectionCreate(ctx, arvados.CreateOptions{
98 Attrs: map[string]interface{}{
99 "manifest_text": ". d41d8cd98f00b204e9800998ecf8427e+0 0:0:foo\n",
100 "trash_at": time.Now().UTC().Add(time.Hour * 24 * 365),
102 c.Assert(err, check.IsNil)
103 resp, err = s.localdb.CollectionGet(ctx, arvados.GetOptions{UUID: trashed.UUID})
104 c.Assert(err, check.IsNil)
105 s.checkSignatureExpiry(c, resp.ManifestText, time.Hour*24*7*2)
107 // Make sure groups/contents doesn't return manifest_text with
108 // collections (if it did, we'd need to sign it).
109 gresp, err := s.localdb.GroupContents(ctx, arvados.GroupContentsOptions{
111 Filters: []arvados.Filter{{"uuid", "=", arvadostest.FooCollection}},
112 Select: []string{"uuid", "manifest_text"},
114 c.Check(err, check.IsNil)
115 if c.Check(gresp.Items, check.HasLen, 1) {
116 c.Check(gresp.Items[0].(map[string]interface{})["uuid"], check.Equals, arvadostest.FooCollection)
117 c.Check(gresp.Items[0].(map[string]interface{})["manifest_text"], check.Equals, nil)
121 func (s *CollectionSuite) checkSignatureExpiry(c *check.C, manifestText string, expectedTTL time.Duration) {
122 m := regexp.MustCompile(`@([[:xdigit:]]+)`).FindStringSubmatch(manifestText)
123 c.Assert(m, check.HasLen, 2)
124 sigexp, err := strconv.ParseInt(m[1], 16, 64)
125 c.Assert(err, check.IsNil)
126 expectedExp := time.Now().Add(expectedTTL).Unix()
127 c.Check(sigexp > expectedExp-60, check.Equals, true)
128 c.Check(sigexp <= expectedExp, check.Equals, true)
131 func (s *CollectionSuite) TestSignaturesDisabled(c *check.C) {
132 s.localdb.cluster.Collections.BlobSigning = false
133 ctx := auth.NewContext(context.Background(), &auth.Credentials{Tokens: []string{arvadostest.ActiveTokenV2}})
135 resp, err := s.localdb.CollectionGet(ctx, arvados.GetOptions{UUID: arvadostest.FooCollection})
136 c.Check(err, check.IsNil)
137 c.Check(resp.ManifestText, check.Matches, `(?ms).* acbd[^ +]*\+3 0:.*`)