parent
a5d8551cbc
commit
4b37ae11b5
20
main.go
20
main.go
|
@ -179,7 +179,7 @@ func main() {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal("Queue processed")
|
log.Fatal("Queue processed")
|
||||||
}
|
}
|
||||||
processFile(jsonPath)
|
processFile(c.certPath, jsonPath)
|
||||||
|
|
||||||
// Exit Signal Handle
|
// Exit Signal Handle
|
||||||
select {
|
select {
|
||||||
|
@ -202,7 +202,7 @@ func main() {
|
||||||
for _, f := range files {
|
for _, f := range files {
|
||||||
path[1] = f.Name()
|
path[1] = f.Name()
|
||||||
jsonPath = strings.Join(path, "/")
|
jsonPath = strings.Join(path, "/")
|
||||||
processFile(jsonPath)
|
processFile(c.certPath, jsonPath)
|
||||||
|
|
||||||
// Exit Signal Handle
|
// Exit Signal Handle
|
||||||
select {
|
select {
|
||||||
|
@ -214,7 +214,7 @@ func main() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func processFile(p string) bool {
|
func processFile(fp string, p string) bool {
|
||||||
// read corresponding json file
|
// read corresponding json file
|
||||||
dat, err := ioutil.ReadFile(p)
|
dat, err := ioutil.ReadFile(p)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -243,7 +243,7 @@ func processFile(p string) bool {
|
||||||
session := buildChain(&s)
|
session := buildChain(&s)
|
||||||
|
|
||||||
// Insert Certificates
|
// Insert Certificates
|
||||||
idc, err := insertCertificates(session)
|
idc, err := insertCertificates(fp, session)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(fmt.Sprintf("Insert Certificate into DB failed: %q", err))
|
log.Fatal(fmt.Sprintf("Insert Certificate into DB failed: %q", err))
|
||||||
}
|
}
|
||||||
|
@ -429,9 +429,9 @@ func unique(s []certMapElm) []certMapElm {
|
||||||
return list
|
return list
|
||||||
}
|
}
|
||||||
|
|
||||||
func insertCertificate(c certMapElm) (string, error) {
|
func insertCertificate(fp string, c certMapElm) (string, error) {
|
||||||
q := `INSERT INTO "certificate" (hash, "is_CA", "is_SS", issuer, subject, cert_chain, is_valid_chain, file_path) VALUES ($1, $2, $3, $4, $5, $6, $7, $8) ON CONFLICT DO NOTHING`
|
q := `INSERT INTO "certificate" (hash, "is_CA", "is_SS", issuer, subject, cert_chain, is_valid_chain, file_path) VALUES ($1, $2, $3, $4, $5, $6, $7, $8) ON CONFLICT DO NOTHING`
|
||||||
_, err := db.Exec(q, c.CertHash, c.Certificate.IsCA, c.chain.isSS, c.Certificate.Issuer.String(), c.Certificate.Subject.String(), c.chain.s, c.chain.isValid, getFullPath(c.CertHash))
|
_, err := db.Exec(q, c.CertHash, c.Certificate.IsCA, c.chain.isSS, c.Certificate.Issuer.String(), c.Certificate.Subject.String(), c.chain.s, c.chain.isValid, getFullPath(fp, c.CertHash))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return c.CertHash, err
|
return c.CertHash, err
|
||||||
}
|
}
|
||||||
|
@ -451,14 +451,14 @@ func insertCertificate(c certMapElm) (string, error) {
|
||||||
|
|
||||||
// getFullPath takes a certificate's hash and return the full path to
|
// getFullPath takes a certificate's hash and return the full path to
|
||||||
// its location on disk
|
// its location on disk
|
||||||
func getFullPath(h string) string {
|
func getFullPath(c string, h string) string {
|
||||||
return "TODO PATH"
|
return c+h
|
||||||
}
|
}
|
||||||
|
|
||||||
func insertCertificates(s *sessionRecord) ([]string, error) {
|
func insertCertificates(fp string, s *sessionRecord) ([]string, error) {
|
||||||
var inserted []string
|
var inserted []string
|
||||||
for _, certificate := range s.Certificates {
|
for _, certificate := range s.Certificates {
|
||||||
tmp, err := insertCertificate(certificate)
|
tmp, err := insertCertificate(fp, certificate)
|
||||||
inserted = append(inserted, tmp)
|
inserted = append(inserted, tmp)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return inserted, err
|
return inserted, err
|
||||||
|
|
Loading…
Reference in New Issue