blob: 46c6b86b555869ba6e82d26c5a5d17e7fa0e65cf [file] [log] [blame]
// Copyright 2021 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package main
import (
"encoding/json"
"fmt"
"io/ioutil"
"os"
"path/filepath"
"strings"
"github.com/google/go-cmp/cmp"
"golang.org/x/vulndb/internal"
"golang.org/x/vulndb/osv"
)
func loadDB(dbPath string) (osv.DBIndex, map[string][]osv.Entry, error) {
index := osv.DBIndex{}
dbMap := map[string][]osv.Entry{}
var loadDir func(string) error
loadDir = func(path string) error {
dir, err := ioutil.ReadDir(path)
if err != nil {
return err
}
for _, f := range dir {
fpath := filepath.Join(path, f.Name())
if f.IsDir() {
if err := loadDir(fpath); err != nil {
return err
}
continue
}
content, err := ioutil.ReadFile(fpath)
if err != nil {
return err
}
if path == dbPath && f.Name() == "index.json" {
if err := json.Unmarshal(content, &index); err != nil {
return fmt.Errorf("unable to parse %q: %s", fpath, err)
}
} else if path == filepath.Join(dbPath, internal.IDDirectory) {
var entry osv.Entry
if err := json.Unmarshal(content, &entry); err != nil {
return fmt.Errorf("unable to parse %q: %s", fpath, err)
}
fname := strings.TrimPrefix(fpath, dbPath)
dbMap[fname] = []osv.Entry{entry}
} else {
var entries []osv.Entry
if err := json.Unmarshal(content, &entries); err != nil {
return fmt.Errorf("unable to parse %q: %s", fpath, err)
}
module := strings.TrimPrefix(fpath, dbPath)
dbMap[module] = entries
}
}
return nil
}
if err := loadDir(dbPath); err != nil {
return nil, nil, err
}
return index, dbMap, nil
}
func main() {
if len(os.Args) != 3 {
fmt.Fprintln(os.Stderr, "usage: dbdiff db-a db-b")
os.Exit(1)
}
indexA, dbA, err := loadDB(os.Args[1])
if err != nil {
fmt.Fprintf(os.Stderr, "unable to load %q: %s\n", os.Args[1], err)
os.Exit(1)
}
indexB, dbB, err := loadDB(os.Args[2])
if err != nil {
fmt.Fprintf(os.Stderr, "unable to load %q: %s\n", os.Args[2], err)
os.Exit(1)
}
indexDiff := cmp.Diff(indexA, indexB)
if indexDiff == "" {
indexDiff = "(no change)"
}
dbDiff := cmp.Diff(dbA, dbB)
if dbDiff == "" {
dbDiff = "(no change)"
}
fmt.Printf("# index\n%s\n\n# db\n%s\n", indexDiff, dbDiff)
}