packagr-io-beta
87 lines
1.7 KiB
Go
87 lines
1.7 KiB
Go
//
|
|
// Use and distribution licensed under the Apache license version 2.
|
|
//
|
|
// See the COPYING file in the root project directory for full text.
|
|
//
|
|
|
|
package pcidb
|
|
|
|
import (
|
|
"bufio"
|
|
"compress/gzip"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
const (
|
|
PCIIDS_URI = "https://pci-ids.ucw.cz/v2.2/pci.ids.gz"
|
|
)
|
|
|
|
func (db *PCIDB) load(ctx *context) error {
|
|
var foundPath string
|
|
for _, fp := range ctx.searchPaths {
|
|
if _, err := os.Stat(fp); err == nil {
|
|
foundPath = fp
|
|
break
|
|
}
|
|
}
|
|
if foundPath == "" {
|
|
if ctx.disableNetworkFetch {
|
|
return ERR_NO_DB
|
|
}
|
|
// OK, so we didn't find any host-local copy of the pci-ids DB file. Let's
|
|
// try fetching it from the network and storing it
|
|
if err := cacheDBFile(ctx.cachePath); err != nil {
|
|
return err
|
|
}
|
|
foundPath = ctx.cachePath
|
|
}
|
|
f, err := os.Open(foundPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
scanner := bufio.NewScanner(f)
|
|
return parseDBFile(db, scanner)
|
|
}
|
|
|
|
func ensureDir(fp string) error {
|
|
fpDir := filepath.Dir(fp)
|
|
if _, err := os.Stat(fpDir); os.IsNotExist(err) {
|
|
err = os.MkdirAll(fpDir, os.ModePerm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Pulls down the latest copy of the pci-ids file from the network and stores
|
|
// it in the local host filesystem
|
|
func cacheDBFile(cacheFilePath string) error {
|
|
ensureDir(cacheFilePath)
|
|
|
|
response, err := http.Get(PCIIDS_URI)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer response.Body.Close()
|
|
f, err := os.Create(cacheFilePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
// write the gunzipped contents to our local cache file
|
|
zr, err := gzip.NewReader(response.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer zr.Close()
|
|
if _, err = io.Copy(f, zr); err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|