mirror of
https://github.com/gocsaf/csaf.git
synced 2025-12-22 05:40:11 +01:00
Moved commonly use file operations to separate package.
This commit is contained in:
parent
9d0ed98a17
commit
2fb2dfda78
7 changed files with 151 additions and 127 deletions
|
|
@ -8,102 +8,9 @@ import (
|
|||
"hash"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"math/rand"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
func deepCopy(dst, src string) error {
|
||||
|
||||
stack := []string{dst, src}
|
||||
|
||||
for len(stack) > 0 {
|
||||
src = stack[len(stack)-1]
|
||||
dst = stack[len(stack)-2]
|
||||
stack = stack[:len(stack)-2]
|
||||
|
||||
if err := func() error {
|
||||
dir, err := os.Open(src)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer dir.Close()
|
||||
|
||||
// Use Readdir as we need no sorting.
|
||||
files, err := dir.Readdir(-1)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, f := range files {
|
||||
nsrc := filepath.Join(src, f.Name())
|
||||
ndst := filepath.Join(dst, f.Name())
|
||||
if f.IsDir() {
|
||||
// Create new sub dir
|
||||
if err := os.Mkdir(ndst, 0755); err != nil {
|
||||
return err
|
||||
}
|
||||
stack = append(stack, ndst, nsrc)
|
||||
} else if f.Mode().IsRegular() {
|
||||
// Create hard link.
|
||||
if err := os.Link(nsrc, ndst); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}(); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func mkUniqFile(prefix string) (string, *os.File, error) {
|
||||
var file *os.File
|
||||
name, err := mkUniq(prefix, func(name string) error {
|
||||
var err error
|
||||
file, err = os.OpenFile(name, os.O_CREATE|os.O_EXCL|os.O_WRONLY, 0644)
|
||||
return err
|
||||
})
|
||||
return name, file, err
|
||||
}
|
||||
|
||||
func mkUniqDir(prefix string) (string, error) {
|
||||
return mkUniq(prefix, func(name string) error { return os.Mkdir(name, 0755) })
|
||||
}
|
||||
|
||||
func mkUniq(prefix string, create func(string) error) (string, error) {
|
||||
now := time.Now()
|
||||
stamp := now.Format("-2006-01-02-150405")
|
||||
name := prefix + stamp
|
||||
err := create(name)
|
||||
if err == nil {
|
||||
return name, nil
|
||||
}
|
||||
if os.IsExist(err) {
|
||||
rnd := rand.New(rand.NewSource(now.Unix()))
|
||||
|
||||
for i := 0; i < 10000; i++ {
|
||||
nname := name + "-" + strconv.FormatUint(uint64(rnd.Uint32()&0xff_ffff), 16)
|
||||
err := create(nname)
|
||||
if err == nil {
|
||||
return nname, nil
|
||||
}
|
||||
if os.IsExist(err) {
|
||||
continue
|
||||
}
|
||||
return "", err
|
||||
}
|
||||
return "", &os.PathError{Op: "mkuniq", Path: name, Err: os.ErrExist}
|
||||
}
|
||||
|
||||
return "", err
|
||||
}
|
||||
|
||||
func writeHash(fname, name string, h hash.Hash, data []byte) error {
|
||||
|
||||
if _, err := io.Copy(h, bytes.NewReader(data)); err != nil {
|
||||
|
|
@ -137,16 +44,3 @@ func writeHashedFile(fname, name string, data []byte, armored string) error {
|
|||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func saveToFile(fname string, wt io.WriterTo) error {
|
||||
f, err1 := os.Create(fname)
|
||||
if err1 != nil {
|
||||
return err1
|
||||
}
|
||||
_, err1 = wt.WriteTo(f)
|
||||
err2 := f.Close()
|
||||
if err1 != nil {
|
||||
return err1
|
||||
}
|
||||
return err2
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue