mcadump

cmd/mcadump/mcadump.go

@@ -0,0 +1,117 @@
+package main
+
+import (
+	"bytes"
+	"compress/gzip"
+	"compress/zlib"
+	"flag"
+	"fmt"
+	"github.com/Tnze/go-mc/save/region"
+	"io"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+)
+
+var decomp = flag.Bool("x", false, "decompress each chunk to NBT format")
+var repack = flag.Bool("p", false, "repack .mcc file to .mca")
+
+func main() {
+	flag.Usage = usage
+	flag.Parse()
+	args := flag.Args()
+
+	var f, o string
+	switch len(args) {
+	default:
+		usage()
+	case 1:
+		f, o = args[0], "."
+	case 2:
+		f, o = args[0], args[1]
+	}
+
+	if *repack {
+		pack(f, o)
+		return
+	}
+
+	var x, z int
+	rn := filepath.Base(f)
+	_, err := fmt.Sscanf(rn, "r.%d.%d.mca", &x, &z)
+	if err != nil {
+		checkerr(fmt.Errorf("cannot use %s as mca file name: %v", rn, err))
+	}
+
+	r, err := region.Open(f)
+	checkerr(err)
+	defer r.Close()
+
+	for i := 0; i < 32; i++ {
+		for j := 0; j < 32; j++ {
+			if !r.ExistSector(i, j) {
+				continue
+			}
+
+			data, err := r.ReadSector(i, j)
+			checkerr(err)
+			var r io.Reader = bytes.NewReader(data[1:])
+
+			fn := fmt.Sprintf("c.%d.%d.mcc", x+i, z+j)
+			if *decomp {
+				fn += ".nbt" //解压后就是一个标准的NBT文件，可以加个.nbt后缀
+				switch data[0] {
+				default:
+					err = fmt.Errorf("unknown compression type 0x%02x", data[0])
+				case 1:
+					r, err = gzip.NewReader(r)
+				case 2:
+					r, err = zlib.NewReader(r)
+				}
+				checkerr(err)
+			}
+
+			cf, err := os.OpenFile(filepath.Join(o, fn), os.O_CREATE|os.O_RDWR|os.O_EXCL, 0666)
+			checkerr(err)
+
+			_, err = io.Copy(cf, r)
+			checkerr(err)
+		}
+	}
+}
+
+func usage() {
+	_, _ = fmt.Fprintf(os.Stderr, "usage: %s [-x] [-r] r.<X>.<Z>.mca [outdir]\n", flag.Arg(0))
+	os.Exit(1)
+}
+
+func checkerr(err error) {
+	if err != nil {
+		_, _ = fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+}
+
+func pack(f, o string) {
+	var x, z int
+	rn := filepath.Base(f)
+	_, err := fmt.Sscanf(rn, "c.%d.%d.mcc", &x, &z)
+	if err != nil {
+		checkerr(fmt.Errorf("cannot use %s as mca file name: %v", rn, err))
+	}
+
+	fn := fmt.Sprintf("r.%d.%d.mca", x/32, z/32)
+	r, err := region.Open(fn)
+	if err != nil && os.IsNotExist(err) {
+		r, err = region.Create(filepath.Join(o, fn))
+	}
+	checkerr(err)
+	defer r.Close()
+
+	mcc, err := ioutil.ReadFile(f)
+	checkerr(err)
+
+	rx, rz := region.In(x, z)
+	err = r.WriteSector(rx, rz, mcc)
+	checkerr(err)
+}

save/region/mca.go

@@ -68,6 +68,35 @@ func Open(name string) (r *Region, err error) {
 	return r, nil
 }
 
+// Create open .mca file with os.O_CREATE|os. O_EXCL, and init the region
+func Create(name string) (r *Region, err error) {
+	r = new(Region)
+	r.sectors = make(map[int32]bool)
+
+	r.f, err = os.OpenFile(name, os.O_CREATE|os.O_RDWR|os.O_EXCL, 0666)
+	if err != nil {
+		return nil, err
+	}
+
+	// read the offsets
+	err = binary.Write(r.f, binary.BigEndian, &r.offsets)
+	if err != nil {
+		_ = r.f.Close()
+		return nil, err
+	}
+	r.sectors[0] = true
+
+	// read the timestamps
+	err = binary.Write(r.f, binary.BigEndian, &r.timestamps)
+	if err != nil {
+		_ = r.f.Close()
+		return nil, err
+	}
+	r.sectors[1] = true
+
+	return r, nil
+}
+
 // Close close the region file
 func (r *Region) Close() error {
 	return r.f.Close()