Merge branch 'admin' of git.daplie.com:Daplie/go-rvpn-server into admin

# Conflicts:
#	rvpn/genericlistener/conn_wedge.go
This commit is contained in:
Henry Camacho 2017-03-24 17:50:55 -05:00
commit 55e49fb966
2 changed files with 9 additions and 14 deletions

12
main.go
View File

@ -35,10 +35,6 @@ var (
serverName string
)
func init() {
}
//Main -- main entry point
func main() {
flag.Parse()
@ -56,10 +52,10 @@ func main() {
wssHostName = viper.Get("rvpn.wssdomain").(string)
adminHostName = viper.Get("rvpn.admindomain").(string)
argGenericBinding = viper.GetInt("rvpn.genericlistener")
deadtime := viper.Get("rvpn.deadtime")
idle = deadtime.(map[string]interface{})["idle"].(int)
dwell = deadtime.(map[string]interface{})["dwell"].(int)
cancelcheck = deadtime.(map[string]interface{})["cancelcheck"].(int)
deadtime := viper.Get("rvpn.deadtime").(map[string]interface{})
idle = deadtime["idle"].(int)
dwell = deadtime["dwell"].(int)
cancelcheck = deadtime["cancelcheck"].(int)
lbDefaultMethod = viper.Get("rvpn.loadbalancing.defaultmethod").(string)
serverName = viper.Get("rvpn.serverName").(string)

View File

@ -55,13 +55,12 @@ func (w *WedgeConn) Buffered() int {
//PeekAll --
// - get all the chars available
// - pass then back
func (w *WedgeConn) PeekAll() (buf []byte, err error) {
_, err = w.Peek(1)
if err != nil {
func (w *WedgeConn) PeekAll() ([]byte, error) {
// We first peek with 1 so that if there is no buffered data the reader will
// fill the buffer before we read how much data is buffered.
if _, err := w.Peek(1); err != nil {
return nil, err
}
buf, err = w.Peek(w.Buffered())
return
return w.Peek(w.Buffered())
}