support read kbin

Author: littlecxm

.gitignore

@@ -20,3 +20,7 @@
 *build/*
 *.exe
 *.bat
+
+# custom
+*.xml
+*.list

go.mod

@@ -2,4 +2,9 @@ module github.com/littlecxm/kcheck
 
 go 1.14
 
-require golang.org/x/net v0.0.0-20200513185701-a91f0712d120
+require (
+	github.com/beevik/etree v1.1.0
+	github.com/orcaman/writerseeker v0.0.0-20200621085525-1d3f536ff85e
+	golang.org/x/net v0.0.0-20200513185701-a91f0712d120
+	golang.org/x/text v0.3.0
+)

go.sum

@@ -1,3 +1,7 @@
+github.com/beevik/etree v1.1.0 h1:T0xke/WvNtMoCqgzPhkX2r4rjY3GDZFi+FjpRZY2Jbs=
+github.com/beevik/etree v1.1.0/go.mod h1:r8Aw8JqVegEf0w2fDnATrX9VpkMcyFeM0FhwO62wh+A=
+github.com/orcaman/writerseeker v0.0.0-20200621085525-1d3f536ff85e h1:s2RNOM/IGdY0Y6qfTeUKhDawdHDpK9RGBdx80qN4Ttw=
+github.com/orcaman/writerseeker v0.0.0-20200621085525-1d3f536ff85e/go.mod h1:nBdnFKj15wFbf94Rwfq4m30eAcyY9V/IyKAGQFtqkW0=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/net v0.0.0-20200513185701-a91f0712d120 h1:EZ3cVSzKOlJxAd8e8YAJ7no8nNypTxexh/YE/xW3ZEY=
 golang.org/x/net v0.0.0-20200513185701-a91f0712d120/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=

kbinxml/iterwalk.go

@@ -0,0 +1,46 @@
+package kbinxml
+
+import "github.com/beevik/etree"
+
+//IterWalker structure
+type IterWalker struct {
+	doc  *etree.Document
+	node *nodeWalker
+	eof  bool
+}
+
+type nodeWalker struct {
+	node              *etree.Element
+	currentChildIndex int
+	parent            *nodeWalker
+}
+
+// IterWalk through a xml document
+func IterWalk(doc *etree.Document) *IterWalker {
+	return &IterWalker{doc, &nodeWalker{doc.Root(), 0, nil}, false}
+}
+
+// Walk a step
+func (i *IterWalker) Walk() (node *etree.Element, event string) {
+	if i.eof {
+		return i.doc.Root(), "eof"
+	}
+	children := i.node.node.ChildElements()
+	if i.node.currentChildIndex == len(children) {
+		event = "end"
+		retNode := i.node.node
+		if i.node.parent == nil {
+			i.eof = true
+		} else {
+			i.node.parent.currentChildIndex++
+			i.node = i.node.parent
+		}
+		return retNode, event
+	}
+	newNode := &nodeWalker{children[i.node.currentChildIndex],
+		0,
+		i.node}
+	i.node = newNode
+	return i.node.node, "start"
+
+}