mirror of https://github.com/dnomd343/XProxy.git
Dnomd343
10 months ago
5 changed files with 130 additions and 88 deletions
@ -1,86 +0,0 @@ |
|||
package assets |
|||
|
|||
import ( |
|||
. "XProxy/next/logger" |
|||
"bytes" |
|||
"compress/bzip2" |
|||
"compress/gzip" |
|||
"fmt" |
|||
"github.com/ulikunitz/xz" |
|||
"io" |
|||
"os" |
|||
) |
|||
|
|||
const gzSample = "/root/XProxy/LICENSE.gz" |
|||
const xzSample = "/root/XProxy/LICENSE.xz" |
|||
const bz2Sample = "/root/XProxy/LICENSE.bz2" |
|||
|
|||
func gzipExtract(content io.Reader) ([]byte, error) { |
|||
Logger.Debugf("Start extracting gzip archive") |
|||
reader, err := gzip.NewReader(content) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to extract gzip archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
defer reader.Close() |
|||
|
|||
var buffer bytes.Buffer |
|||
size, err := io.Copy(&buffer, reader) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to handle gzip archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
Logger.Debugf("Successfully extracted gzip archive -> %d bytes", size) |
|||
return buffer.Bytes(), nil |
|||
} |
|||
|
|||
func bzip2Extract(content io.Reader) ([]byte, error) { |
|||
Logger.Debugf("Start extracting bzip2 archive") |
|||
reader := bzip2.NewReader(content) |
|||
|
|||
var buffer bytes.Buffer |
|||
size, err := io.Copy(&buffer, reader) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to extract bzip2 archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
Logger.Debugf("Successfully extracted bzip2 archive -> %d bytes", size) |
|||
return buffer.Bytes(), nil |
|||
} |
|||
|
|||
func xzExtract(content io.Reader) ([]byte, error) { |
|||
Logger.Debugf("Start extracting xz archive") |
|||
reader, err := xz.NewReader(content) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to extract xz archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
|
|||
var buffer bytes.Buffer |
|||
size, err := io.Copy(&buffer, reader) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to handle xz archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
Logger.Debugf("Successfully extracted xz archive -> %d bytes", size) |
|||
return buffer.Bytes(), nil |
|||
} |
|||
|
|||
func Demo() { |
|||
Logger.Infof("Assets demo begin") |
|||
|
|||
//fp, err := os.Open(gzSample)
|
|||
//fp, err := os.Open(bz2Sample)
|
|||
fp, err := os.Open(xzSample) |
|||
if err != nil { |
|||
fmt.Println("open failed") |
|||
} |
|||
defer fp.Close() |
|||
|
|||
//gzipExtract(fp)
|
|||
//bzip2Extract(fp)
|
|||
xzExtract(fp) |
|||
|
|||
//fp.Name()
|
|||
|
|||
} |
|||
@ -0,0 +1,106 @@ |
|||
package assets |
|||
|
|||
import ( |
|||
. "XProxy/next/logger" |
|||
"bytes" |
|||
"compress/bzip2" |
|||
"github.com/gabriel-vasile/mimetype" |
|||
"github.com/klauspost/compress/gzip" |
|||
"github.com/ulikunitz/xz" |
|||
"io" |
|||
) |
|||
|
|||
const ( |
|||
notArchive = iota |
|||
gzipArchive |
|||
bzip2Archive |
|||
xzArchive |
|||
) |
|||
|
|||
// gzipExtract use to extract independent gzip archive data.
|
|||
func gzipExtract(data []byte) ([]byte, error) { |
|||
Logger.Debugf("Start extracting gzip archive -> %d bytes", len(data)) |
|||
reader, err := gzip.NewReader(bytes.NewReader(data)) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to extract gzip archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
defer reader.Close() |
|||
|
|||
var buffer bytes.Buffer |
|||
size, err := io.Copy(&buffer, reader) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to handle gzip archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
Logger.Debugf("Extracted gzip archive successfully -> %d bytes", size) |
|||
return buffer.Bytes(), nil |
|||
} |
|||
|
|||
// bzip2Extract use to extract independent bzip2 archive data.
|
|||
func bzip2Extract(data []byte) ([]byte, error) { |
|||
Logger.Debugf("Start extracting bzip2 archive -> %d bytes", len(data)) |
|||
reader := bzip2.NewReader(bytes.NewReader(data)) |
|||
|
|||
var buffer bytes.Buffer |
|||
size, err := io.Copy(&buffer, reader) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to extract bzip2 archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
Logger.Debugf("Extracted bzip2 archive successfully -> %d bytes", size) |
|||
return buffer.Bytes(), nil |
|||
} |
|||
|
|||
// xzExtract use to extract independent xz archive data.
|
|||
func xzExtract(data []byte) ([]byte, error) { |
|||
Logger.Debugf("Start extracting xz archive -> %d bytes", len(data)) |
|||
reader, err := xz.NewReader(bytes.NewReader(data)) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to extract xz archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
|
|||
var buffer bytes.Buffer |
|||
size, err := io.Copy(&buffer, reader) |
|||
if err != nil { |
|||
Logger.Errorf("Failed to handle xz archive -> %v", err) |
|||
return nil, err |
|||
} |
|||
Logger.Debugf("Extracted xz archive successfully -> %d bytes", size) |
|||
return buffer.Bytes(), nil |
|||
} |
|||
|
|||
// archiveType use to determine the type of archive file.
|
|||
func archiveType(data []byte) uint { |
|||
mime := mimetype.Detect(data) |
|||
switch mime.String() { |
|||
case "application/gzip": |
|||
Logger.Debugf("Data detected as gzip format") |
|||
return gzipArchive |
|||
case "application/x-bzip2": |
|||
Logger.Debugf("Data detected as bzip2 format") |
|||
return bzip2Archive |
|||
case "application/x-xz": |
|||
Logger.Debugf("Data detected as xz format") |
|||
return xzArchive |
|||
default: |
|||
Logger.Debugf("Data detected as non-archive format -> %s", mime) |
|||
return notArchive |
|||
} |
|||
} |
|||
|
|||
// Extract will try to extract the data as a compressed format, and will
|
|||
// return the original data if it cannot be determined.
|
|||
func Extract(data []byte) ([]byte, error) { |
|||
switch archiveType(data) { |
|||
case gzipArchive: |
|||
return gzipExtract(data) |
|||
case bzip2Archive: |
|||
return bzip2Extract(data) |
|||
case xzArchive: |
|||
return xzExtract(data) |
|||
default: |
|||
return data, nil |
|||
} |
|||
} |
|||
Loading…
Reference in new issue