forked from engineering/cloudservices
dir changes
parent
6a9d31f1a9
commit
dae9750d08
|
|
@ -1,96 +0,0 @@
|
|||
// ignore this error
|
||||
package main
|
||||
|
||||
import (
|
||||
"crypto/sha1"
|
||||
"crypto/x509"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"encoding/pem"
|
||||
"encoding/xml"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
)
|
||||
|
||||
// HandleError handles an error by panicing.
|
||||
func HandleError(e error) {
|
||||
if e != nil {
|
||||
panic(e)
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
type CertificateAbstract struct {
|
||||
SignatureAlgorithm string
|
||||
PublicKeyAlgorithm string
|
||||
ExtendedKeyUsage []string
|
||||
PolicyIdentifiers []string
|
||||
FingerPrint string
|
||||
}
|
||||
type CompleteCertificate struct {
|
||||
XMLName xml.Name `xml:"CertificateParse"`
|
||||
RawParse *x509.Certificate
|
||||
AbstractParse CertificateAbstract
|
||||
}
|
||||
certificateFile, err := ioutil.ReadFile(os.Args[1])
|
||||
HandleError(err)
|
||||
certificatePemDecode, _ := pem.Decode(certificateFile)
|
||||
if certificatePemDecode == nil {
|
||||
fmt.Println("Can't do that.")
|
||||
os.Exit(1)
|
||||
}
|
||||
certificateParse, err := x509.ParseCertificate(certificatePemDecode.Bytes)
|
||||
HandleError(err)
|
||||
policyIdentifiers := []string{}
|
||||
extendedKeyUsages := []string{}
|
||||
for _, value := range certificateParse.PolicyIdentifiers {
|
||||
policyIdentifiers = append(policyIdentifiers, value.String())
|
||||
}
|
||||
for _, value := range certificateParse.ExtKeyUsage {
|
||||
switch value {
|
||||
case 0:
|
||||
extendedKeyUsages = append(extendedKeyUsages, "All/Any Usages")
|
||||
break
|
||||
case 1:
|
||||
extendedKeyUsages = append(extendedKeyUsages, "TLS Web Server Authentication")
|
||||
break
|
||||
case 2:
|
||||
extendedKeyUsages = append(extendedKeyUsages, "TLS Web Client Authentication")
|
||||
break
|
||||
case 3:
|
||||
extendedKeyUsages = append(extendedKeyUsages, "Code Signing")
|
||||
break
|
||||
case 4:
|
||||
extendedKeyUsages = append(extendedKeyUsages, "E-mail Protection (S/MIME)")
|
||||
default:
|
||||
break
|
||||
}
|
||||
}
|
||||
sum := sha1.Sum(certificateParse.Raw)
|
||||
certificateStruct := CompleteCertificate{
|
||||
RawParse: certificateParse,
|
||||
AbstractParse: CertificateAbstract{
|
||||
SignatureAlgorithm: certificateParse.SignatureAlgorithm.String(),
|
||||
PublicKeyAlgorithm: certificateParse.PublicKeyAlgorithm.String(),
|
||||
PolicyIdentifiers: policyIdentifiers,
|
||||
ExtendedKeyUsage: extendedKeyUsages,
|
||||
FingerPrint: hex.EncodeToString(sum[:]),
|
||||
},
|
||||
}
|
||||
if len(os.Args) >= 3 {
|
||||
if os.Args[2] == "json" {
|
||||
data, err := json.MarshalIndent(certificateStruct, "", " ")
|
||||
HandleError(err)
|
||||
fmt.Printf("%v\n", string(data))
|
||||
} else if os.Args[2] == "xml" {
|
||||
data, err := xml.MarshalIndent(certificateStruct, "", " ")
|
||||
HandleError(err)
|
||||
fmt.Printf("%v\n", string(data))
|
||||
}
|
||||
} else {
|
||||
data, err := json.MarshalIndent(certificateStruct, "", " ")
|
||||
HandleError(err)
|
||||
fmt.Printf("%v\n", string(data))
|
||||
}
|
||||
}
|
||||
|
|
@ -1,26 +0,0 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
// DirSize This function returns the total size of a directory in bytes.
|
||||
func DirSize(path* string) float64 {
|
||||
var dirSize int64 = 0
|
||||
|
||||
readSize := func(path string, file os.FileInfo, err error) error {
|
||||
if !file.IsDir() {
|
||||
dirSize += file.Size()
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
err := filepath.Walk(*path, readSize)
|
||||
HandleError(err, 0)
|
||||
|
||||
size := float64(dirSize)
|
||||
|
||||
return size
|
||||
}
|
||||
|
|
@ -1,99 +0,0 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/go-redis/redis/v7"
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
"io/ioutil"
|
||||
"time"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Collection the MongoDB Account collection
|
||||
var Collection *mongo.Collection
|
||||
|
||||
// RedisClient the Redis client
|
||||
var RedisClient *redis.Client
|
||||
|
||||
// Account represents an user's account.
|
||||
type Account struct {
|
||||
Username string `json:"username"`
|
||||
UserID string `json:"userID"`
|
||||
EmailAddress string `json:"emailAddress"`
|
||||
CreatedBy string `json:"createdBy"`
|
||||
CreatedAt time.Time `json:"createdAt"`
|
||||
Locked bool `json:"locked"`
|
||||
}
|
||||
|
||||
// HandleError This function panics if an error exists.
|
||||
func HandleError(e error, serv int) {
|
||||
if e != nil && serv == 1 {
|
||||
panic(e)
|
||||
} else if e != nil && serv == 0 {
|
||||
fmt.Println(e)
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
type Config struct {
|
||||
MongoDB string `json:"mongoURL"`
|
||||
}
|
||||
config := &Config{}
|
||||
file, err := ioutil.ReadFile("../config.json")
|
||||
HandleError(err, 1)
|
||||
err = json.Unmarshal(file, &config)
|
||||
|
||||
client, err := mongo.NewClient(options.Client().ApplyURI(config.MongoDB))
|
||||
HandleError(err, 1)
|
||||
ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
err = client.Connect(ctx)
|
||||
HandleError(err, 1)
|
||||
err = client.Ping(context.TODO(), nil)
|
||||
fmt.Printf("Connected to MongoDB [GO]\n")
|
||||
HandleError(err, 1)
|
||||
|
||||
Collection = client.Database("cloudservices").Collection("accounts")
|
||||
|
||||
RedisClient = redis.NewClient(&redis.Options{
|
||||
Addr: "localhost:6379",
|
||||
Password: "",
|
||||
DB: 0,
|
||||
})
|
||||
|
||||
_, err = RedisClient.Ping().Result()
|
||||
fmt.Printf("Connected to Redis [GO]\n")
|
||||
HandleError(err, 1)
|
||||
|
||||
for {
|
||||
fmt.Printf("Calling handler func [GO]\n")
|
||||
handler()
|
||||
time.Sleep(1000000 * time.Millisecond)
|
||||
}
|
||||
}
|
||||
|
||||
func handler() {
|
||||
cur, err := Collection.Find(context.TODO(), bson.D{})
|
||||
HandleError(err, 0)
|
||||
|
||||
for cur.Next(context.TODO()) {
|
||||
go checkAccountSizeAndUpdate(cur.Current.Lookup("username").String(), cur.Current.Lookup("id").String())
|
||||
fmt.Printf("Checking account information for %s\n", cur.Current.Lookup("username").String())
|
||||
time.Sleep(600000 * time.Millisecond)
|
||||
}
|
||||
}
|
||||
|
||||
func checkAccountSizeAndUpdate(username string, id string) {
|
||||
var size float64 = 0
|
||||
var userHomeDirectory string = strings.Replace(strings.Join([]string{"/home/", string(username)}, ""), "\"", "", -1)
|
||||
fmt.Println(userHomeDirectory)
|
||||
sizeHome := DirSize(&userHomeDirectory)
|
||||
size += sizeHome
|
||||
sizeMail := DirSize(&userHomeDirectory)
|
||||
size += sizeMail
|
||||
RedisClient.Set("storage"+"-"+string(id), size, 0)
|
||||
fmt.Printf("Set Call | Username: %v, ID: %v | Bytes: %f\n", string(username), string(id), size)
|
||||
}
|
||||
Loading…
Reference in New Issue