forked from hanwen/go-fuse
/
tarfs.go
124 lines (106 loc) · 1.97 KB
/
tarfs.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package zipfs
import (
"archive/tar"
"bytes"
"compress/bzip2"
"compress/gzip"
"fmt"
"io"
"os"
"path/filepath"
"strings"
"syscall"
)
var _ = fmt.Println
// TODO - handle symlinks.
func HeaderToFileInfo(h *tar.Header) *os.FileInfo {
return &os.FileInfo{
Name: h.Name,
Mode: uint32(h.Mode),
Uid: h.Uid,
Gid: h.Gid,
Size: h.Size,
Mtime_ns: h.Mtime,
Atime_ns: h.Atime,
Ctime_ns: h.Ctime,
}
}
type TarFile struct {
data []byte
tar.Header
}
func (me *TarFile) Stat() *os.FileInfo {
fi := HeaderToFileInfo(&me.Header)
fi.Mode |= syscall.S_IFREG
return fi
}
func (me *TarFile) Data() []byte {
return me.data
}
func NewTarTree(r io.Reader) *MemTree {
tree := NewMemTree()
tr := tar.NewReader(r)
var longName *string
for {
hdr, err := tr.Next()
if err == os.EOF {
// end of tar archive
break
}
if err != nil {
// handle error
}
if hdr.Typeflag == 'L' {
buf := bytes.NewBuffer(make([]byte, 0, hdr.Size))
io.Copy(buf, tr)
s := buf.String()
longName = &s
continue
}
if longName != nil {
hdr.Name = *longName
longName = nil
}
comps := strings.Split(filepath.Clean(hdr.Name), "/", -1)
base := ""
if !strings.HasSuffix(hdr.Name, "/") {
base = comps[len(comps)-1]
comps = comps[:len(comps)-1]
}
parent := tree
for _, c := range comps {
parent = parent.FindDir(c)
}
buf := bytes.NewBuffer(make([]byte, 0, hdr.Size))
io.Copy(buf, tr)
if base != "" {
b := buf.Bytes()
parent.files[base] = &TarFile{
Header: *hdr,
data: b,
}
}
}
return tree
}
func NewTarCompressedTree(name string, format string) (*MemTree, os.Error) {
f, err := os.Open(name)
if err != nil {
return nil, err
}
defer f.Close()
var stream io.Reader
switch format {
case "gz":
unzip, err := gzip.NewReader(f)
if err != nil {
return nil, err
}
defer unzip.Close()
stream = unzip
case "bz2":
unzip := bzip2.NewReader(f)
stream = unzip
}
return NewTarTree(stream), nil
}