tcpsock_plan9.go 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. // Copyright 2009 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. package net
  5. import (
  6. "context"
  7. "io"
  8. "os"
  9. )
  10. func (c *TCPConn) readFrom(r io.Reader) (int64, error) {
  11. return genericReadFrom(c, r)
  12. }
  13. func (sd *sysDialer) dialTCP(ctx context.Context, laddr, raddr *TCPAddr) (*TCPConn, error) {
  14. if testHookDialTCP != nil {
  15. return testHookDialTCP(ctx, sd.network, laddr, raddr)
  16. }
  17. return sd.doDialTCP(ctx, laddr, raddr)
  18. }
  19. func (sd *sysDialer) doDialTCP(ctx context.Context, laddr, raddr *TCPAddr) (*TCPConn, error) {
  20. switch sd.network {
  21. case "tcp4":
  22. // Plan 9 doesn't complain about [::]:0->127.0.0.1, so it's up to us.
  23. if laddr != nil && len(laddr.IP) != 0 && laddr.IP.To4() == nil {
  24. return nil, &AddrError{Err: "non-IPv4 local address", Addr: laddr.String()}
  25. }
  26. case "tcp", "tcp6":
  27. default:
  28. return nil, UnknownNetworkError(sd.network)
  29. }
  30. if raddr == nil {
  31. return nil, errMissingAddress
  32. }
  33. fd, err := dialPlan9(ctx, sd.network, laddr, raddr)
  34. if err != nil {
  35. return nil, err
  36. }
  37. return newTCPConn(fd), nil
  38. }
  39. func (ln *TCPListener) ok() bool { return ln != nil && ln.fd != nil && ln.fd.ctl != nil }
  40. func (ln *TCPListener) accept() (*TCPConn, error) {
  41. fd, err := ln.fd.acceptPlan9()
  42. if err != nil {
  43. return nil, err
  44. }
  45. tc := newTCPConn(fd)
  46. if ln.lc.KeepAlive >= 0 {
  47. setKeepAlive(fd, true)
  48. ka := ln.lc.KeepAlive
  49. if ln.lc.KeepAlive == 0 {
  50. ka = defaultTCPKeepAlive
  51. }
  52. setKeepAlivePeriod(fd, ka)
  53. }
  54. return tc, nil
  55. }
  56. func (ln *TCPListener) close() error {
  57. if err := ln.fd.pfd.Close(); err != nil {
  58. return err
  59. }
  60. if _, err := ln.fd.ctl.WriteString("hangup"); err != nil {
  61. ln.fd.ctl.Close()
  62. return err
  63. }
  64. if err := ln.fd.ctl.Close(); err != nil {
  65. return err
  66. }
  67. return nil
  68. }
  69. func (ln *TCPListener) file() (*os.File, error) {
  70. f, err := ln.dup()
  71. if err != nil {
  72. return nil, err
  73. }
  74. return f, nil
  75. }
  76. func (sl *sysListener) listenTCP(ctx context.Context, laddr *TCPAddr) (*TCPListener, error) {
  77. fd, err := listenPlan9(ctx, sl.network, laddr)
  78. if err != nil {
  79. return nil, err
  80. }
  81. return &TCPListener{fd: fd, lc: sl.ListenConfig}, nil
  82. }