aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--misc/main.go5
-rw-r--r--tde/tde.go22
2 files changed, 12 insertions, 15 deletions
diff --git a/misc/main.go b/misc/main.go
index dcf9b98..fbc967b 100644
--- a/misc/main.go
+++ b/misc/main.go
@@ -5,7 +5,6 @@ import (
"flag"
"fmt"
"log"
- "io/ioutil"
"dim13.org/asn1/tde"
)
@@ -212,7 +211,7 @@ func Ask(c tde.TDE, out []byte) []byte {
log.Println("Ask")
dump(out)
c.Write(out)
- in, _ := ioutil.ReadAll(c)
+ in, _ := tde.ReadAll(c)
log.Printf("Packet length: 0x%.2X\n", len(in))
dump(in)
return in
@@ -220,7 +219,7 @@ func Ask(c tde.TDE, out []byte) []byte {
func Status(c tde.TDE) {
log.Println("Status")
- in, _ := ioutil.ReadAll(c)
+ in, _ := tde.ReadAll(c)
log.Printf("Packet length: 0x%.2X\n", len(in))
dump(in)
out := status
diff --git a/tde/tde.go b/tde/tde.go
index 44ee6e3..2b4e35b 100644
--- a/tde/tde.go
+++ b/tde/tde.go
@@ -17,10 +17,7 @@ func (t TDE) Write(p []byte) (n int, err error) {
if err := binary.Write(t.Conn, nbo, size); err != nil {
return 0, err
}
- if err := binary.Write(t.Conn, nbo, p); err != nil {
- return 0, err
- }
- return int(size), nil
+ return t.Conn.Write(p)
}
func (t TDE) Read(p []byte) (n int, err error) {
@@ -28,14 +25,15 @@ func (t TDE) Read(p []byte) (n int, err error) {
if err := binary.Read(t.Conn, nbo, &size); err != nil {
return 0, err
}
- if int(size) < len(p) {
- p = p[:size]
- err = io.EOF
- }
- if err := binary.Read(t.Conn, nbo, p); err != nil {
- return 0, err
- }
- return int(size), err
+ return t.Conn.Read(p[:size])
+}
+
+const maxLen = 240
+
+func ReadAll(r io.Reader) ([]byte, error) {
+ p := make([]byte, maxLen)
+ n, err := r.Read(p)
+ return p[:n], err
}
func Dial(service string) (TDE, error) {