feat: Add basic write support

This commit is contained in:
Felix Pojtinger
2021-11-14 19:57:59 +01:00
parent 725d716eeb
commit 004ef4ea3d
5 changed files with 131 additions and 3 deletions

View File

@@ -22,7 +22,6 @@ type Operation struct {
func main() {
file := flag.String("file", "/dev/nst0", "File of tape drive to open")
record := flag.Int("record", 0, "Record to seek too")
flag.Parse()
@@ -38,8 +37,7 @@ func main() {
MTIOCTOP,
uintptr(unsafe.Pointer(
&Operation{
Op: MTEOM,
Count: int32(*record),
Op: MTEOM,
},
)),
)

View File

@@ -97,6 +97,11 @@ func main() {
break
}
// Missing trailer (expected for concatenated tars)
if err == io.ErrUnexpectedEOF {
break
}
panic(err)
}

View File

@@ -60,6 +60,11 @@ func main() {
break
}
// Missing trailer (expected for concatenated tars)
if err == io.ErrUnexpectedEOF {
break
}
panic(err)
}

120
cmd/stfs-uvf/main.go Normal file
View File

@@ -0,0 +1,120 @@
package main
import (
"archive/tar"
"flag"
"io"
"io/fs"
"log"
"os"
"path/filepath"
"syscall"
"unsafe"
)
// See https://github.com/benmcclelland/mtio
const (
MTIOCTOP = 0x40086d01 // Do magnetic tape operation
MTEOM = 12 // Goto end of recorded media (for appending files)
)
// Operation is struct for MTIOCTOP
type Operation struct {
Op int16 // Operation ID
Pad int16 // Padding to match C structures
Count int32 // Operation count
}
func main() {
file := flag.String("file", "/dev/nst0", "File (tape drive or tar file) to open")
dir := flag.String("dir", ".", "Directory to add to the file")
flag.Parse()
isRegular := true
stat, err := os.Stat(*file)
if err == nil {
isRegular = stat.Mode().IsRegular()
} else {
if os.IsNotExist(err) {
isRegular = true
} else {
panic(err)
}
}
var f *os.File
if isRegular {
f, err = os.OpenFile(*file, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0600)
if err != nil {
panic(err)
}
// No need to go to end manually due to `os.O_APPEND`
} else {
// Go to end of file
syscall.Syscall(
syscall.SYS_IOCTL,
f.Fd(),
MTIOCTOP,
uintptr(unsafe.Pointer(
&Operation{
Op: MTEOM,
},
)),
)
f, err = os.OpenFile(*file, os.O_APPEND|os.O_WRONLY, os.ModeCharDevice)
if err != nil {
panic(err)
}
}
defer f.Close()
tw := tar.NewWriter(f) // We are not closing the tar writer to prevent writing the trailer
if err := filepath.Walk(*dir, func(path string, info fs.FileInfo, err error) error {
if err != nil {
return err
}
link := ""
if info.Mode()&os.ModeSymlink == os.ModeSymlink {
if link, err = os.Readlink(path); err != nil {
return err
}
}
hdr, err := tar.FileInfoHeader(info, link)
if err != nil {
return err
}
hdr.Name = path
log.Println(hdr)
if err := tw.WriteHeader(hdr); err != nil {
return err
}
// Skip writing non-regular files
if !info.Mode().IsRegular() {
return nil
}
file, err := os.Open(path)
if err != nil {
return err
}
defer file.Close()
if _, err := io.Copy(tw, file); err != nil {
return err
}
return nil
}); err != nil {
panic(err)
}
}