libgo/go/net/fd_unix_test.go

   1 // Copyright 2012 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
   5 // +build darwin freebsd linux netbsd openbsd
   6
   7 package net
   8
   9 import (
  10         "io"
  11         "syscall"
  12         "testing"
  13 )
  14
  15 // Issue 3590. netFd.AddFD should return an error
  16 // from the underlying pollster rather than panicing.
  17 func TestAddFDReturnsError(t *testing.T) {
  18         ln := newLocalListener(t).(*TCPListener)
  19         defer ln.Close()
  20         connected := make(chan bool)
  21         go func() {
  22                 for {
  23                         c, err := ln.Accept()
  24                         if err != nil {
  25                                 return
  26                         }
  27                         connected <- true
  28                         defer c.Close()
  29                 }
  30         }()
  31
  32         c, err := DialTCP("tcp", nil, ln.Addr().(*TCPAddr))
  33         if err != nil {
  34                 t.Fatal(err)
  35         }
  36         defer c.Close()
  37         <-connected
  38
  39         // replace c's pollServer with a closed version.
  40         ps, err := newPollServer()
  41         if err != nil {
  42                 t.Fatal(err)
  43         }
  44         ps.poll.Close()
  45         c.conn.fd.pollServer = ps
  46
  47         var b [1]byte
  48         _, err = c.Read(b[:])
  49         if err, ok := err.(*OpError); ok {
  50                 if err.Op == "addfd" {
  51                         return
  52                 }
  53                 if err, ok := err.Err.(*OpError); ok {
  54                         // the err is sometimes wrapped by another OpError
  55                         if err.Op == "addfd" {
  56                                 return
  57                         }
  58                 }
  59         }
  60         t.Error("unexpected error:", err)
  61 }
  62
  63 var chkReadErrTests = []struct {
  64         n        int
  65         err      error
  66         fd       *netFD
  67         expected error
  68 }{
  69
  70         {100, nil, &netFD{sotype: syscall.SOCK_STREAM}, nil},
  71         {100, io.EOF, &netFD{sotype: syscall.SOCK_STREAM}, io.EOF},
  72         {100, errClosing, &netFD{sotype: syscall.SOCK_STREAM}, errClosing},
  73         {0, nil, &netFD{sotype: syscall.SOCK_STREAM}, io.EOF},
  74         {0, io.EOF, &netFD{sotype: syscall.SOCK_STREAM}, io.EOF},
  75         {0, errClosing, &netFD{sotype: syscall.SOCK_STREAM}, errClosing},
  76
  77         {100, nil, &netFD{sotype: syscall.SOCK_DGRAM}, nil},
  78         {100, io.EOF, &netFD{sotype: syscall.SOCK_DGRAM}, io.EOF},
  79         {100, errClosing, &netFD{sotype: syscall.SOCK_DGRAM}, errClosing},
  80         {0, nil, &netFD{sotype: syscall.SOCK_DGRAM}, nil},
  81         {0, io.EOF, &netFD{sotype: syscall.SOCK_DGRAM}, io.EOF},
  82         {0, errClosing, &netFD{sotype: syscall.SOCK_DGRAM}, errClosing},
  83
  84         {100, nil, &netFD{sotype: syscall.SOCK_SEQPACKET}, nil},
  85         {100, io.EOF, &netFD{sotype: syscall.SOCK_SEQPACKET}, io.EOF},
  86         {100, errClosing, &netFD{sotype: syscall.SOCK_SEQPACKET}, errClosing},
  87         {0, nil, &netFD{sotype: syscall.SOCK_SEQPACKET}, io.EOF},
  88         {0, io.EOF, &netFD{sotype: syscall.SOCK_SEQPACKET}, io.EOF},
  89         {0, errClosing, &netFD{sotype: syscall.SOCK_SEQPACKET}, errClosing},
  90
  91         {100, nil, &netFD{sotype: syscall.SOCK_RAW}, nil},
  92         {100, io.EOF, &netFD{sotype: syscall.SOCK_RAW}, io.EOF},
  93         {100, errClosing, &netFD{sotype: syscall.SOCK_RAW}, errClosing},
  94         {0, nil, &netFD{sotype: syscall.SOCK_RAW}, nil},
  95         {0, io.EOF, &netFD{sotype: syscall.SOCK_RAW}, io.EOF},
  96         {0, errClosing, &netFD{sotype: syscall.SOCK_RAW}, errClosing},
  97 }
  98
  99 func TestChkReadErr(t *testing.T) {
 100         for _, tt := range chkReadErrTests {
 101                 actual := chkReadErr(tt.n, tt.err, tt.fd)
 102                 if actual != tt.expected {
 103                         t.Errorf("chkReadError(%v, %v, %v): expected %v, actual %v", tt.n, tt.err, tt.fd.sotype, tt.expected, actual)
 104                 }
 105         }
 106 }