socks5.go 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. // Copyright 2017 fatedier, fatedier@gmail.com
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package plugin
  15. import (
  16. "io"
  17. "log"
  18. "net"
  19. gosocks5 "github.com/armon/go-socks5"
  20. frpNet "github.com/fatedier/frp/pkg/util/net"
  21. )
  22. const PluginSocks5 = "socks5"
  23. func init() {
  24. Register(PluginSocks5, NewSocks5Plugin)
  25. }
  26. type Socks5Plugin struct {
  27. Server *gosocks5.Server
  28. }
  29. func NewSocks5Plugin(params map[string]string) (p Plugin, err error) {
  30. user := params["plugin_user"]
  31. passwd := params["plugin_passwd"]
  32. cfg := &gosocks5.Config{
  33. Logger: log.New(io.Discard, "", log.LstdFlags),
  34. }
  35. if user != "" || passwd != "" {
  36. cfg.Credentials = gosocks5.StaticCredentials(map[string]string{user: passwd})
  37. }
  38. sp := &Socks5Plugin{}
  39. sp.Server, err = gosocks5.New(cfg)
  40. p = sp
  41. return
  42. }
  43. func (sp *Socks5Plugin) Handle(conn io.ReadWriteCloser, realConn net.Conn, extraBufToLocal []byte) {
  44. defer conn.Close()
  45. wrapConn := frpNet.WrapReadWriteCloserToConn(conn, realConn)
  46. _ = sp.Server.ServeConn(wrapConn)
  47. }
  48. func (sp *Socks5Plugin) Name() string {
  49. return PluginSocks5
  50. }
  51. func (sp *Socks5Plugin) Close() error {
  52. return nil
  53. }