Skip to content

Commit f3000a3

Browse files
committed
refactoring for golang 1.11
1 parent f4d34b4 commit f3000a3

File tree

5 files changed

+24
-17
lines changed

5 files changed

+24
-17
lines changed

.gitignore

+1
Original file line numberDiff line numberDiff line change
@@ -22,3 +22,4 @@ _testmain.go
2222
*.exe
2323
*.test
2424
*.prof
25+
example/example

README.md

+3-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
# scp
22

3-
A Simple `go` SCP client library
3+
[![Go Report Card](https://goreportcard.com/badge/github.com/dtylman/scp)](https://goreportcard.com/report/github.com/dtylman/scp)
4+
5+
A Simple `go` SCP client library.
46

57
## Usage
68

example/main.go

+2-2
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@ import (
88
"path/filepath"
99
"strings"
1010
"time"
11-
12-
"github.com/dtylman/scp"
11+
12+
"github.com/dtylman/scp"
1313
"golang.org/x/crypto/ssh"
1414
)
1515

msg.go

+15-9
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,12 @@ import (
1010
)
1111

1212
const (
13-
CopyMessage = 'C'
13+
//CopyMessage Copy Message Opcode
14+
CopyMessage = 'C'
15+
//ErrorMessage Error OpCode
1416
ErrorMessage = 0x1
15-
WarnMessage = 0x2
17+
//WarnMessage Warning Opcode
18+
WarnMessage = 0x2
1619
)
1720

1821
//Message is scp control message
@@ -40,6 +43,7 @@ func (m *Message) readOpCode(reader io.Reader) error {
4043
return err
4144
}
4245

46+
//ReadError reads an error message
4347
func (m *Message) ReadError(reader io.Reader) error {
4448
msg, err := ioutil.ReadAll(reader)
4549
if err != nil {
@@ -83,31 +87,33 @@ func (m *Message) readCopy(reader io.Reader) error {
8387
return nil
8488
}
8589

86-
func (m *Message) ReadFrom(reader io.Reader) error {
90+
//ReadFrom reads message from reader
91+
func (m *Message) ReadFrom(reader io.Reader) (int64, error) {
8792
err := m.readOpCode(reader)
8893
if err != nil {
89-
return err
94+
return 0, err
9095
}
9196
switch m.Type {
9297
case CopyMessage:
9398
err = m.readCopy(reader)
9499
if err != nil {
95-
return err
100+
return 0, err
96101
}
97102
case ErrorMessage, WarnMessage:
98103
err = m.ReadError(reader)
99104
if err != nil {
100-
return err
105+
return 0, err
101106
}
102107
default:
103-
return errors.New(fmt.Sprintf("Unsupported opcode: %v", m.Type))
108+
return 0, fmt.Errorf("Unsupported opcode: %v", m.Type)
104109
}
105-
return nil
110+
return m.Size, nil
106111
}
107112

113+
//NewMessageFromReader constructs a new message from a data in reader
108114
func NewMessageFromReader(reader io.Reader) (*Message, error) {
109115
m := new(Message)
110-
err := m.ReadFrom(reader)
116+
_, err := m.ReadFrom(reader)
111117
if err != nil {
112118
return nil, err
113119
}

scp.go

+3-5
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,15 @@
11
package scp
22

33
import (
4+
"bytes"
5+
"errors"
46
"fmt"
57
"io"
68
"os"
79
"path/filepath"
810

9-
log "github.com/Sirupsen/logrus"
11+
log "github.com/sirupsen/logrus"
1012
"golang.org/x/crypto/ssh"
11-
12-
"errors"
13-
14-
"bytes"
1513
)
1614

1715
const (

0 commit comments

Comments
 (0)