-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
auth_info.go
69 lines (61 loc) · 2.24 KB
/
auth_info.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
// Copyright (C) MongoDB, Inc. 2014-present.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License. You may obtain
// a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
// Package auth provides utilities for performing tasks related to authentication.
package auth
import (
"fmt"
"strings"
"github.com/kontenbase/kb-mongotools/common/db"
"github.com/kontenbase/kb-mongotools/common/util"
"go.mongodb.org/mongo-driver/bson"
)
// GetAuthVersion gets the authentication schema version of the connected server
// and returns that value as an integer along with any error that occurred.
func GetAuthVersion(sessionProvider *db.SessionProvider) (int, error) {
results := bson.M{}
err := sessionProvider.Run(
bson.D{
{"getParameter", 1},
{"authSchemaVersion", 1},
},
&results,
"admin",
)
if err != nil {
errMessage := err.Error()
// as a necessary hack, if the error message takes a certain form,
// we can infer version 1. This is because early versions of mongodb
// had no concept of an "auth schema version", so asking for the
// authSchemaVersion value will return a "no option found" or "no such cmd"
if errMessage == "no option found to get" ||
strings.Contains(errMessage, "no such cmd") {
return 1, nil
}
// otherwise it's a connection error, so bubble it up
return 0, err
}
version, err := util.ToInt(results["authSchemaVersion"])
if err != nil {
return 0, fmt.Errorf("getParameter command returned non-numeric result: %v, error: %v", results["authSchemaVersion"], err)
}
return version, nil
}
// VerifySystemAuthVersion returns an error if authentication is not set up for
// the given server.
func VerifySystemAuthVersion(sessionProvider *db.SessionProvider) error {
session, err := sessionProvider.GetSession()
if err != nil {
return fmt.Errorf("error getting session from server: %v", err)
}
authSchemaQuery := bson.M{"_id": "authSchema"}
count, err := session.Database("admin").Collection("system.version").CountDocuments(nil, authSchemaQuery)
if err != nil {
return fmt.Errorf("error checking pressence of auth version: %v", err)
} else if count == 0 {
return fmt.Errorf("found no auth version")
}
return nil
}