The zero_len fixture injects raw pre-encrypted records over a socket
carrying a TLS_RX key only, and its record table already holds
zero-length application_data records. Every variant reads them back
with a plain recv(), so neither the splice path nor MSG_PEEK is
exercised against a record that decrypts to no payload.

Add zero_len_splice. An empty data record splices zero bytes, and a
zero return from splice() means EOF, so a peer that sends one ends a
connection that is still live. Its variants expect the payload's
length when one sits behind a run of empty records, EAGAIN when
nothing does, and EINVAL when a control record does; an unfixed
kernel reports 0 for all three. Add zero_len_peek, which checks that
a peek reaches the payload behind such a run and leaves it in place
for the read that follows. That one reproduces no failure: the
unbounded rx_list growth it accompanies needs a sustained flood that
three fixed-sequence records cannot supply.

Signed-off-by: Chuck Lever <[email protected]>
---
 tools/testing/selftests/net/tls.c | 213 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 213 insertions(+)

diff --git a/tools/testing/selftests/net/tls.c 
b/tools/testing/selftests/net/tls.c
index cbdd3ea28b99..089e940ac43d 100644
--- a/tools/testing/selftests/net/tls.c
+++ b/tools/testing/selftests/net/tls.c
@@ -2573,6 +2573,219 @@ TEST_F(zero_len, test)
        }
 };
 
+static void zero_len_sock_pair(struct __test_metadata *_metadata,
+                              int *fd, int *cfd, bool *notls)
+{
+       struct tls_crypto_info_keys tls12;
+       int ret;
+
+       tls_crypto_info_init(TLS_1_2_VERSION, TLS_CIPHER_AES_CCM_128,
+                            &tls12, 0);
+
+       ulp_sock_pair(_metadata, fd, cfd, notls);
+       if (*notls)
+               return;
+
+       /* fd stays keyless; these fixtures send raw records over it */
+       ret = setsockopt(*cfd, SOL_TLS, TLS_RX, &tls12, tls12.len);
+       ASSERT_EQ(ret, 0);
+}
+
+/* Send a variant's records; return the last one carrying payload */
+static const struct raw_rec *
+zero_len_send_recs(struct __test_metadata *_metadata, int fd,
+                  const struct raw_rec *const *recs)
+{
+       const struct raw_rec *payload = NULL;
+       int i;
+
+       for (i = 0; i < 4 && recs[i]; i++) {
+               EXPECT_EQ(send(fd, recs[i]->cipher_data, recs[i]->cipher_len, 
0),
+                         recs[i]->cipher_len);
+               if (recs[i]->plain_len)
+                       payload = recs[i];
+       }
+
+       return payload;
+}
+
+FIXTURE(zero_len_peek)
+{
+       int fd, cfd;
+       bool notls;
+};
+
+FIXTURE_VARIANT(zero_len_peek)
+{
+       const struct raw_rec *recs[4];
+       ssize_t peek_ret;
+};
+
+FIXTURE_VARIANT_ADD(zero_len_peek, 0data_0data_data)
+{
+       .recs = { &id0_data_l0, &id1_data_l0, &id2_data_l11, },
+       .peek_ret = 11,
+};
+
+FIXTURE_VARIANT_ADD(zero_len_peek, 0data_0data_0data)
+{
+       .recs = { &id0_data_l0, &id1_data_l0, &id2_data_l0, },
+       .peek_ret = -EAGAIN,
+};
+
+FIXTURE_SETUP(zero_len_peek)
+{
+       zero_len_sock_pair(_metadata, &self->fd, &self->cfd, &self->notls);
+}
+
+FIXTURE_TEARDOWN(zero_len_peek)
+{
+       close(self->fd);
+       close(self->cfd);
+}
+
+/* Peeking past a run of empty data records must reach the payload
+ * behind them, and a run with no payload behind it must report EAGAIN
+ * rather than the zero return that means EOF.
+ */
+TEST_F(zero_len_peek, test)
+{
+       const struct raw_rec *payload;
+       unsigned char buf[128];
+       ssize_t ret;
+
+       if (self->notls)
+               SKIP(return, "no TLS support");
+
+       payload = zero_len_send_recs(_metadata, self->fd, variant->recs);
+
+       if (variant->peek_ret < 0) {
+               ret = recv(self->cfd, buf, sizeof(buf),
+                          MSG_DONTWAIT | MSG_PEEK);
+               EXPECT_EQ(ret, -1);
+               if (ret == -1)
+                       EXPECT_EQ(errno, -variant->peek_ret);
+               return;
+       }
+
+       ret = recv(self->cfd, buf, sizeof(buf), MSG_DONTWAIT | MSG_PEEK);
+       EXPECT_EQ(ret, variant->peek_ret);
+       if (ret == variant->peek_ret)
+               EXPECT_EQ(memcmp(buf, payload->plain_data,
+                                variant->peek_ret), 0);
+
+       /* Peeking left the payload in place for the read that follows */
+       ret = recv(self->cfd, buf, sizeof(buf), MSG_DONTWAIT);
+       EXPECT_EQ(ret, variant->peek_ret);
+       if (ret == variant->peek_ret)
+               EXPECT_EQ(memcmp(buf, payload->plain_data,
+                                variant->peek_ret), 0);
+
+       ret = recv(self->cfd, buf, sizeof(buf), MSG_DONTWAIT);
+       EXPECT_EQ(ret, -1);
+       if (ret == -1)
+               EXPECT_EQ(errno, EAGAIN);
+}
+
+FIXTURE(zero_len_splice)
+{
+       int fd, cfd;
+       bool notls;
+};
+
+FIXTURE_VARIANT(zero_len_splice)
+{
+       const struct raw_rec *recs[4];
+       ssize_t splice_ret;
+};
+
+FIXTURE_VARIANT_ADD(zero_len_splice, 0data_data)
+{
+       .recs = { &id0_data_l0, &id1_data_l11, },
+       .splice_ret = 11,
+};
+
+FIXTURE_VARIANT_ADD(zero_len_splice, 0data_0data_data)
+{
+       .recs = { &id0_data_l0, &id1_data_l0, &id2_data_l11, },
+       .splice_ret = 11,
+};
+
+FIXTURE_VARIANT_ADD(zero_len_splice, 0data_0data_0data)
+{
+       .recs = { &id0_data_l0, &id1_data_l0, &id2_data_l0, },
+       .splice_ret = -EAGAIN,
+};
+
+FIXTURE_VARIANT_ADD(zero_len_splice, 0data_0ctrl)
+{
+       .recs = { &id0_data_l0, &id1_ctrl_l0, },
+       .splice_ret = -EINVAL,
+};
+
+FIXTURE_SETUP(zero_len_splice)
+{
+       zero_len_sock_pair(_metadata, &self->fd, &self->cfd, &self->notls);
+}
+
+FIXTURE_TEARDOWN(zero_len_splice)
+{
+       close(self->fd);
+       close(self->cfd);
+}
+
+/* An empty data record splices zero bytes, which a splice caller reads
+ * as EOF. Splicing must skip past such a record to the payload behind
+ * it, and report EAGAIN when a run of them has no payload behind it.
+ * A control record behind the run reports EINVAL, the error splice
+ * already reports for a control record it meets first.
+ */
+TEST_F(zero_len_splice, test)
+{
+       const struct raw_rec *payload;
+       unsigned char buf[128];
+       ssize_t ret;
+       int p[2];
+
+       if (self->notls)
+               SKIP(return, "no TLS support");
+
+       ASSERT_GE(pipe(p), 0);
+
+       payload = zero_len_send_recs(_metadata, self->fd, variant->recs);
+
+       if (variant->splice_ret < 0) {
+               ret = splice(self->cfd, NULL, p[1], NULL, sizeof(buf),
+                            SPLICE_F_NONBLOCK);
+               EXPECT_EQ(ret, -1);
+               if (ret == -1)
+                       EXPECT_EQ(errno, -variant->splice_ret);
+       } else {
+               /* Assert: a zero return, which is what an unfixed kernel
+                * gives here, leaves the pipe empty, and the read below
+                * would then block until the harness timeout.
+                */
+               ASSERT_EQ(splice(self->cfd, NULL, p[1], NULL, sizeof(buf),
+                                SPLICE_F_NONBLOCK), variant->splice_ret);
+               ret = read(p[0], buf, sizeof(buf));
+               EXPECT_EQ(ret, variant->splice_ret);
+               if (ret == variant->splice_ret)
+                       EXPECT_EQ(memcmp(buf, payload->plain_data,
+                                        variant->splice_ret), 0);
+
+               /* Reaching the payload consumed the empty records ahead
+                * of it rather than leaving them on the receive queue
+                */
+               ret = recv(self->cfd, buf, sizeof(buf), MSG_DONTWAIT);
+               EXPECT_EQ(ret, -1);
+               if (ret == -1)
+                       EXPECT_EQ(errno, EAGAIN);
+       }
+
+       close(p[0]);
+       close(p[1]);
+}
+
 FIXTURE(tls_err)
 {
        int fd, cfd;

-- 
2.54.0


Reply via email to