1 files changed, 67 insertions, 0 deletions

diff --git a/src/regress/lib/libtls/gotls/tls_test.go b/src/regress/lib/libtls/gotls/tls_test.go
index 2331ec0be6..3a430924a7 100644
--- a/src/regress/lib/libtls/gotls/tls_test.go
+++ b/src/regress/lib/libtls/gotls/tls_test.go
@@ -118,6 +118,73 @@ func TestTLSBasic(t *testing.T) {
         }
 }
 
+func TestTLSSingleByteReadWrite(t *testing.T) {
+        ts, u, caFile, err := newTestServer()
+        if err != nil {
+                t.Fatalf("Failed to start test server: %v", err)
+        }
+        defer os.Remove(caFile)
+        defer ts.Close()
+
+        if err := Init(); err != nil {
+                t.Fatal(err)
+        }
+
+        cfg, err := NewConfig()
+        if err != nil {
+                t.Fatal(err)
+        }
+        defer cfg.Free()
+        cfg.SetCAFile(caFile)
+
+        tls, err := NewClient(cfg)
+        if err != nil {
+                t.Fatal(err)
+        }
+        defer tls.Free()
+
+        t.Logf("Connecting to %s", u.Host)
+
+        if err := tls.Connect(u.Host, ""); err != nil {
+                t.Fatal(err)
+        }
+        defer func() {
+                if err := tls.Close(); err != nil {
+                        t.Fatalf("Close failed: %v", err)
+                }
+        }()
+
+        for _, b := range []byte("GET / HTTP/1.0\n\n") {
+                n, err := tls.Write([]byte{b})
+                if err != nil {
+                        t.Fatal(err)
+                }
+                if n != 1 {
+                        t.Fatalf("Wrote byte %v, got length %d, want 1", b, n)
+                }
+        }
+
+        var body []byte
+        for {
+                buf := make([]byte, 1)
+                n, err := tls.Read(buf)
+                if err != nil {
+                        t.Fatal(err)
+                }
+                if n == 0 {
+                        break
+                }
+                if n != 1 {
+                        t.Fatalf("Read single byte, got length %d, want 1", n)
+                }
+                body = append(body, buf...)
+        }
+
+        if !strings.Contains(string(body), httpContent) {
+                t.Errorf("Response does not contain %q", httpContent)
+        }
+}
+
 func TestTLSInfo(t *testing.T) {
         ts, u, caFile, err := newTestServer()
         if err != nil {