lazarus/formats/mpq/mpq.go

187 lines
4.1 KiB
Go
Raw Normal View History

2018-12-15 21:30:18 +00:00
package mpq
2018-12-28 15:44:29 +00:00
// #ifdef _MPQ_WINDOWS
// #include <windows.h>
// #include <stdlib.h>
// #endif
// #ifdef _MPQ_LINUX
// #include <stdlib.h>
// #define WINAPI
// DWORD GetLastError();
// #endif
//
2018-12-16 22:08:38 +00:00
// #define DWORD unsigned int
// #define LPDWORD unsigned int *
2018-12-15 21:30:18 +00:00
// #define LPOVERLAPPED void *
2018-12-16 22:08:38 +00:00
// #define TCHAR char
// #define HANDLE void *
2018-12-16 22:08:38 +00:00
// #define bool unsigned char
// #define LONG int
2018-12-28 15:44:29 +00:00
//
// bool WINAPI SFileOpenArchive(const TCHAR * szMpqName, DWORD dwPriority, DWORD dwFlags, HANDLE * phMpq);
// bool WINAPI SFileCloseArchive(HANDLE hMpq);
// bool WINAPI SFileOpenFileEx(HANDLE hMpq, const char * szFileName, DWORD dwSearchScope, HANDLE * phFile);
// DWORD WINAPI SFileSetFilePointer(HANDLE hFile, LONG lFilePos, LONG * plFilePosHigh, DWORD dwMoveMethod);
2018-12-28 15:44:29 +00:00
// bool WINAPI SFileReadFile(HANDLE hFile, void * lpBuffer, DWORD dwToRead, LPDWORD pdwRead, LPOVERLAPPED lpOverlapped);
// bool WINAPI SFileCloseFile(HANDLE hFile);
//
2018-12-15 21:30:18 +00:00
import "C"
import (
"bytes"
"fmt"
"io"
"math"
"os"
2018-12-15 21:30:18 +00:00
"strings"
"unsafe"
)
type File interface {
Read(data []byte) (int, error)
Seek(offset int64, whence int) (int64, error)
Close() error
2018-12-15 21:30:18 +00:00
}
type Archive interface {
OpenFile(path string) (File, error)
2018-12-16 04:48:20 +00:00
GetPaths() []string
2018-12-15 21:30:18 +00:00
Close() error
}
2018-12-16 04:48:20 +00:00
func NewFromFile(path string) (Archive, error) {
2018-12-15 21:30:18 +00:00
cs := C.CString(path)
defer C.free(unsafe.Pointer(cs))
a := new(archive)
if result := C.SFileOpenArchive(cs, 0, 0, &a.handle); result == 0 {
return nil, fmt.Errorf("failed to open archive (%d)", getLastError())
}
2018-12-16 04:48:20 +00:00
if err := a.buildPathMap(); err != nil {
a.Close()
return nil, err
}
2018-12-15 21:30:18 +00:00
return a, nil
}
type file struct {
handle unsafe.Pointer
}
func (f *file) Read(data []byte) (int, error) {
var bytesRead int
if result := C.SFileReadFile(f.handle, unsafe.Pointer(&data[0]), C.uint(len(data)), (*C.uint)(unsafe.Pointer(&bytesRead)), nil); result == 0 {
lastError := getLastError()
if lastError == sysEOF { // ERROR_HANDLE_EOF
2018-12-18 01:26:10 +00:00
return bytesRead, io.EOF
}
return 0, fmt.Errorf("failed to read file (%d)", lastError)
2018-12-15 21:30:18 +00:00
}
return bytesRead, nil
}
func (f *file) Seek(offset int64, whence int) (int64, error) {
var method uint
switch whence {
case io.SeekStart:
method = 0 // FILE_BEGIN
case io.SeekCurrent:
method = 1 // FILE_CURRENT
case io.SeekEnd:
method = 2 // FILE_END
}
result := C.SFileSetFilePointer(f.handle, C.int(offset), nil, C.uint(method))
if result == math.MaxUint32 { // SFILE_INVALID_SIZE
return 0, fmt.Errorf("failed to set file pointer (%d)", getLastError())
}
return int64(result), nil
2018-12-15 21:30:18 +00:00
}
func (f *file) Close() error {
if result := C.SFileCloseFile(f.handle); result == 0 {
return fmt.Errorf("failed to close file (%d)", getLastError())
}
f.handle = nil
2018-12-16 04:48:20 +00:00
return nil
}
2018-12-15 21:30:18 +00:00
type archive struct {
handle unsafe.Pointer
2018-12-16 04:48:20 +00:00
paths map[string]string
2018-12-15 21:30:18 +00:00
}
func (a *archive) Close() error {
if result := C.SFileCloseArchive(a.handle); result == 0 {
return fmt.Errorf("failed to close archive (%d)", getLastError())
}
a.handle = nil
a.paths = nil
return nil
}
func (a *archive) OpenFile(path string) (File, error) {
2018-12-16 04:48:20 +00:00
if pathInt, ok := a.paths[path]; ok {
path = pathInt
}
cs := C.CString(path)
2018-12-15 21:30:18 +00:00
defer C.free(unsafe.Pointer(cs))
file := new(file)
2018-12-15 21:30:18 +00:00
if result := C.SFileOpenFileEx(a.handle, cs, 0, &file.handle); result == 0 {
return nil, fmt.Errorf("failed to open file (%d)", getLastError())
}
return file, nil
}
2018-12-16 04:48:20 +00:00
func (a *archive) GetPaths() []string {
var extPaths []string
for extPath := range a.paths {
extPaths = append(extPaths, extPath)
2018-12-15 21:30:18 +00:00
}
2018-12-16 04:48:20 +00:00
return extPaths
}
func (a *archive) buildPathMap() error {
2018-12-15 21:30:18 +00:00
f, err := a.OpenFile("(listfile)")
if err != nil {
2018-12-16 04:48:20 +00:00
return err
2018-12-15 21:30:18 +00:00
}
defer f.Close()
2018-12-15 21:30:18 +00:00
var buff bytes.Buffer
if _, err := io.Copy(&buff, f); err != nil {
2018-12-16 04:48:20 +00:00
return err
2018-12-15 21:30:18 +00:00
}
2018-12-16 04:48:20 +00:00
a.paths = make(map[string]string)
lines := strings.Split(string(buff.Bytes()), "\r\n")
for _, line := range lines {
pathInt := strings.TrimSpace(line)
if len(pathInt) > 0 {
2018-12-28 15:44:29 +00:00
pathExt := sanitizePath(pathInt)
2018-12-16 04:48:20 +00:00
a.paths[pathExt] = pathInt
2018-12-15 21:30:18 +00:00
}
}
2018-12-16 04:48:20 +00:00
return nil
}
2018-12-28 15:44:29 +00:00
func sanitizePath(path string) string {
2018-12-16 04:48:20 +00:00
return strings.ToLower(strings.Replace(path, "\\", string(os.PathSeparator), -1))
2018-12-15 21:30:18 +00:00
}
func getLastError() uint {
return uint(C.GetLastError())
}