80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
package pki
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"math"
|
|
"os"
|
|
"path"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
const SymlinkPrefix = "_AUTH_ID_"
|
|
|
|
func GetCertificateAssociation(serial uint64) string {
|
|
return fmt.Sprintf(SymlinkPrefix+"%0[2]*[1]X", serial, int(math.Ceil(math.Log2(float64(serial))/8)*2))
|
|
}
|
|
|
|
func GetAssociation(dirname string) (assocs string, err error) {
|
|
return os.Readlink(dirname)
|
|
}
|
|
|
|
func GetAssociations(dirname string) (assocs []string, err error) {
|
|
if ds, errr := ioutil.ReadDir(dirname); errr != nil {
|
|
return nil, errr
|
|
} else {
|
|
for _, d := range ds {
|
|
if d.Mode()&os.ModeSymlink == os.ModeSymlink {
|
|
assocs = append(assocs, d.Name())
|
|
}
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
func GetTeamSerials(dirname string, id_team int64) (serials []uint64, err error) {
|
|
// As futher comparaisons will be made with strings, convert it only one time
|
|
str_tid := fmt.Sprintf("%d", id_team)
|
|
|
|
var assocs []string
|
|
if assocs, err = GetAssociations(dirname); err != nil {
|
|
return
|
|
} else {
|
|
for _, assoc := range assocs {
|
|
var tid string
|
|
if tid, err = os.Readlink(path.Join(dirname, assoc)); err == nil && tid == str_tid && strings.HasPrefix(assoc, SymlinkPrefix) {
|
|
if serial, err := strconv.ParseUint(assoc[9:], 16, 64); err == nil {
|
|
serials = append(serials, serial)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetTeamAssociations(dirname string, id_team int64) (teamAssocs []string, err error) {
|
|
// As futher comparaisons will be made with strings, convert it only one time
|
|
str_tid := fmt.Sprintf("%d", id_team)
|
|
|
|
var assocs []string
|
|
if assocs, err = GetAssociations(dirname); err != nil {
|
|
return
|
|
} else {
|
|
for _, assoc := range assocs {
|
|
var tid string
|
|
if tid, err = os.Readlink(path.Join(dirname, assoc)); err == nil && tid == str_tid && !strings.HasPrefix(assoc, SymlinkPrefix) {
|
|
teamAssocs = append(teamAssocs, assoc)
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func DeleteTeamAssociation(dirname string, assoc string) error {
|
|
if err := os.Remove(path.Join(dirname, assoc)); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|