tests/fcntl_test.cpp

   1 /*
   2  * Copyright (C) 2013 The Android Open Source Project
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <gtest/gtest.h>
  18
  19 #include <errno.h>
  20 #include <fcntl.h>
  21
  22 #include "TemporaryFile.h"
  23
  24 TEST(fcntl, fcntl_smoke) {
  25   int fd = open("/proc/version", O_RDONLY);
  26   ASSERT_TRUE(fd != -1);
  27
  28   int flags = fcntl(fd, F_GETFD);
  29   ASSERT_TRUE(flags != -1);
  30   ASSERT_EQ(0, flags & FD_CLOEXEC);
  31
  32   int rc = fcntl(fd, F_SETFD, FD_CLOEXEC);
  33   ASSERT_EQ(0, rc);
  34
  35   flags = fcntl(fd, F_GETFD);
  36   ASSERT_TRUE(flags != -1);
  37   ASSERT_EQ(FD_CLOEXEC, flags & FD_CLOEXEC);
  38
  39   close(fd);
  40 }
  41
  42 TEST(fcntl, open_open64) {
  43   int fd;
  44
  45   fd = open("/proc/version", O_RDONLY);
  46   ASSERT_TRUE(fd != -1);
  47   close(fd);
  48
  49   fd = open64("/proc/version", O_RDONLY);
  50   ASSERT_TRUE(fd != -1);
  51   close(fd);
  52 }
  53
  54 TEST(fcntl, openat_openat64) {
  55   int fd;
  56
  57   fd = openat(AT_FDCWD, "/proc/version", O_RDONLY);
  58   ASSERT_TRUE(fd != -1);
  59   close(fd);
  60
  61   fd = openat64(AT_FDCWD, "/proc/version", O_RDONLY);
  62   ASSERT_TRUE(fd != -1);
  63   close(fd);
  64 }
  65
  66 TEST(fcntl, creat_creat64) {
  67   ASSERT_EQ(-1, creat("", 0666));
  68   ASSERT_EQ(ENOENT, errno);
  69   ASSERT_EQ(-1, creat64("", 0666));
  70   ASSERT_EQ(ENOENT, errno);
  71 }
  72
  73 TEST(fcntl, posix_fadvise) {
  74   TemporaryFile tf;
  75   errno = 0;
  76
  77   EXPECT_EQ(EBADF, posix_fadvise(-1, 0, 0, POSIX_FADV_NORMAL));
  78   EXPECT_EQ(0, errno);
  79
  80   EXPECT_EQ(EBADF, posix_fadvise64(-1, 0, 0, POSIX_FADV_NORMAL));
  81   EXPECT_EQ(0, errno);
  82
  83   EXPECT_EQ(EINVAL, posix_fadvise(tf.fd, 0, 0, -1));
  84   EXPECT_EQ(0, errno);
  85
  86   EXPECT_EQ(EINVAL, posix_fadvise64(tf.fd, 0, 0, -1));
  87   EXPECT_EQ(0, errno);
  88
  89   EXPECT_EQ(0, posix_fadvise(tf.fd, 0, 0, POSIX_FADV_NORMAL));
  90   EXPECT_EQ(0, posix_fadvise64(tf.fd, 0, 0, POSIX_FADV_NORMAL));
  91 }
  92
  93 TEST(fcntl, fallocate_EINVAL) {
  94   TemporaryFile tf;
  95
  96   // fallocate/fallocate64 set errno.
  97   // posix_fallocate/posix_fallocate64 return an errno value.
  98
  99   errno = 0;
 100   ASSERT_EQ(-1, fallocate(tf.fd, 0, 0, -1));
 101   ASSERT_EQ(EINVAL, errno);
 102
 103   errno = 0;
 104   ASSERT_EQ(-1, fallocate64(tf.fd, 0, 0, -1));
 105   ASSERT_EQ(EINVAL, errno);
 106
 107   errno = 0;
 108   ASSERT_EQ(EINVAL, posix_fallocate(tf.fd, 0, -1));
 109   ASSERT_EQ(0, errno);
 110
 111   errno = 0;
 112   ASSERT_EQ(EINVAL, posix_fallocate64(tf.fd, 0, -1));
 113   ASSERT_EQ(0, errno);
 114 }
 115
 116 TEST(fcntl, fallocate) {
 117   TemporaryFile tf;
 118   struct stat sb;
 119   ASSERT_EQ(0, fstat(tf.fd, &sb));
 120   ASSERT_EQ(0, sb.st_size);
 121
 122 #if defined(__BIONIC__)
 123   ASSERT_EQ(0, fallocate(tf.fd, 0, 0, 1));
 124   ASSERT_EQ(0, fstat(tf.fd, &sb));
 125   ASSERT_EQ(1, sb.st_size);
 126
 127   ASSERT_EQ(0, fallocate64(tf.fd, 0, 0, 2));
 128   ASSERT_EQ(0, fstat(tf.fd, &sb));
 129   ASSERT_EQ(2, sb.st_size);
 130 #endif
 131
 132   ASSERT_EQ(0, posix_fallocate(tf.fd, 0, 3));
 133   ASSERT_EQ(0, fstat(tf.fd, &sb));
 134   ASSERT_EQ(3, sb.st_size);
 135
 136   ASSERT_EQ(0, posix_fallocate64(tf.fd, 0, 4));
 137   ASSERT_EQ(0, fstat(tf.fd, &sb));
 138   ASSERT_EQ(4, sb.st_size);
 139 }
 140
 141 TEST(fcntl, f_getlk64) {
 142   int fd = open64("/proc/version", O_RDONLY);
 143   ASSERT_TRUE(fd != -1);
 144
 145   struct flock64 check_lock;
 146   check_lock.l_type = F_WRLCK;
 147   check_lock.l_start = 0;
 148   check_lock.l_whence = SEEK_SET;
 149   check_lock.l_len = 0;
 150
 151   int rc = fcntl(fd, F_GETLK64, &check_lock);
 152   ASSERT_EQ(0, rc);
 153
 154   close(fd);
 155 }
 156
 157 TEST(fcntl, splice) {
 158   int pipe_fds[2];
 159   ASSERT_EQ(0, pipe(pipe_fds));
 160
 161   int in = open("/proc/cpuinfo", O_RDONLY);
 162   ASSERT_NE(in, -1);
 163
 164   TemporaryFile tf;
 165
 166   ssize_t bytes_read = splice(in, 0, pipe_fds[1], NULL, 8*1024, SPLICE_F_MORE | SPLICE_F_MOVE);
 167   ASSERT_NE(bytes_read, -1);
 168
 169   ssize_t bytes_written = splice(pipe_fds[0], NULL, tf.fd, 0, bytes_read, SPLICE_F_MORE | SPLICE_F_MOVE);
 170   ASSERT_EQ(bytes_read, bytes_written);
 171
 172   close(pipe_fds[0]);
 173   close(pipe_fds[1]);
 174   close(in);
 175 }
 176
 177 TEST(fcntl, vmsplice) {
 178   int pipe_fds[2];
 179   ASSERT_EQ(0, pipe(pipe_fds));
 180
 181   iovec v[2];
 182   v[0].iov_base = const_cast<char*>("hello ");
 183   v[0].iov_len = 6;
 184   v[1].iov_base = const_cast<char*>("world\n");
 185   v[1].iov_len = 6;
 186   ssize_t bytes_written = vmsplice(pipe_fds[1], v, sizeof(v)/sizeof(iovec), 0);
 187   ASSERT_EQ(v[0].iov_len + v[1].iov_len, static_cast<size_t>(bytes_written));
 188   close(pipe_fds[1]);
 189
 190   char buf[BUFSIZ];
 191   FILE* fp = fdopen(pipe_fds[0], "r");
 192   ASSERT_TRUE(fp != NULL);
 193   ASSERT_TRUE(fgets(buf, sizeof(buf), fp) != NULL);
 194   fclose(fp);
 195   ASSERT_STREQ("hello world\n", buf);
 196 }
 197
 198 TEST(fcntl, tee) {
 199   char expected[256];
 200   FILE* expected_fp = fopen("/proc/version", "r");
 201   ASSERT_TRUE(expected_fp != NULL);
 202   ASSERT_TRUE(fgets(expected, sizeof(expected), expected_fp) != NULL);
 203   fclose(expected_fp);
 204
 205   int pipe1[2];
 206   ASSERT_EQ(0, pipe(pipe1));
 207
 208   int pipe2[2];
 209   ASSERT_EQ(0, pipe(pipe2));
 210
 211   int in = open("/proc/version", O_RDONLY);
 212   ASSERT_NE(in, -1);
 213
 214   // Write /proc/version into pipe1.
 215   ssize_t bytes_read = splice(in, 0, pipe1[1], NULL, 8*1024, SPLICE_F_MORE | SPLICE_F_MOVE);
 216   ASSERT_NE(bytes_read, -1);
 217   close(pipe1[1]);
 218
 219   // Tee /proc/version from pipe1 into pipe2.
 220   ssize_t bytes_teed = tee(pipe1[0], pipe2[1], SIZE_MAX, 0);
 221   ASSERT_EQ(bytes_read, bytes_teed);
 222   close(pipe2[1]);
 223
 224   // The out fds of both pipe1 and pipe2 should now contain /proc/version.
 225   char buf1[BUFSIZ];
 226   FILE* fp1 = fdopen(pipe1[0], "r");
 227   ASSERT_TRUE(fp1 != NULL);
 228   ASSERT_TRUE(fgets(buf1, sizeof(buf1), fp1) != NULL);
 229   fclose(fp1);
 230
 231   char buf2[BUFSIZ];
 232   FILE* fp2 = fdopen(pipe2[0], "r");
 233   ASSERT_TRUE(fp2 != NULL);
 234   ASSERT_TRUE(fgets(buf2, sizeof(buf2), fp2) != NULL);
 235   fclose(fp2);
 236
 237   ASSERT_STREQ(expected, buf1);
 238   ASSERT_STREQ(expected, buf2);
 239 }