| OLD | NEW |
| (Empty) |
| 1 /* Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 * Use of this source code is governed by a BSD-style license that can be | |
| 3 * found in the LICENSE file. | |
| 4 */ | |
| 5 | |
| 6 #include <errno.h> | |
| 7 #include <fcntl.h> | |
| 8 #include <string.h> | |
| 9 #include <string> | |
| 10 #include <sys/stat.h> | |
| 11 | |
| 12 #include "nacl_mounts/mount.h" | |
| 13 #include "nacl_mounts/mount_dev.h" | |
| 14 #include "nacl_mounts/mount_mem.h" | |
| 15 #include "nacl_mounts/osdirent.h" | |
| 16 | |
| 17 #include "gtest/gtest.h" | |
| 18 | |
| 19 namespace { | |
| 20 | |
| 21 class MountMemMock : public MountMem { | |
| 22 public: | |
| 23 MountMemMock() { | |
| 24 StringMap_t map; | |
| 25 Init(1, map, NULL); | |
| 26 }; | |
| 27 | |
| 28 int num_nodes() { | |
| 29 return (int) inode_pool_.size(); | |
| 30 } | |
| 31 }; | |
| 32 | |
| 33 class MountDevMock : public MountDev { | |
| 34 public: | |
| 35 MountDevMock() { | |
| 36 StringMap_t map; | |
| 37 Init(1, map, NULL); | |
| 38 } | |
| 39 int num_nodes() { | |
| 40 return (int) inode_pool_.size(); | |
| 41 } | |
| 42 }; | |
| 43 | |
| 44 } // namespace | |
| 45 | |
| 46 | |
| 47 #define NULL_NODE ((MountNode *) NULL) | |
| 48 | |
| 49 TEST(MountTest, Sanity) { | |
| 50 MountMemMock* mnt = new MountMemMock(); | |
| 51 MountNode* file; | |
| 52 MountNode* root; | |
| 53 | |
| 54 char buf1[1024]; | |
| 55 | |
| 56 // A memory mount starts with one directory node: the root. | |
| 57 EXPECT_EQ(1, mnt->num_nodes()); | |
| 58 | |
| 59 // Fail to open non existant file | |
| 60 EXPECT_EQ(NULL_NODE, mnt->Open(Path("/foo"), O_RDWR)); | |
| 61 EXPECT_EQ(errno, ENOENT); | |
| 62 | |
| 63 // Create a file | |
| 64 file = mnt->Open(Path("/foo"), O_RDWR | O_CREAT); | |
| 65 EXPECT_NE(NULL_NODE, file); | |
| 66 if (file == NULL) return; | |
| 67 EXPECT_EQ(2, file->RefCount()); | |
| 68 EXPECT_EQ(2, mnt->num_nodes()); | |
| 69 | |
| 70 // Open the root directory | |
| 71 root = mnt->Open(Path("/"), O_RDWR); | |
| 72 EXPECT_NE(NULL_NODE, root); | |
| 73 if (NULL != root) { | |
| 74 struct dirent dirs[2]; | |
| 75 int len = root->GetDents(0, dirs, sizeof(dirs)); | |
| 76 EXPECT_EQ(sizeof(struct dirent), len); | |
| 77 } | |
| 78 | |
| 79 // Fail to re-create the same file | |
| 80 EXPECT_EQ(NULL_NODE, mnt->Open(Path("/foo"), O_RDWR | O_CREAT | O_EXCL)); | |
| 81 EXPECT_EQ(errno, EEXIST); | |
| 82 EXPECT_EQ(2, mnt->num_nodes()); | |
| 83 | |
| 84 // Fail to create a directory with the same name | |
| 85 EXPECT_EQ(-1, mnt->Mkdir(Path("/foo"), O_RDWR)); | |
| 86 EXPECT_EQ(errno, EEXIST); | |
| 87 | |
| 88 // Attempt to READ/WRITE | |
| 89 EXPECT_EQ(0, file->GetSize()); | |
| 90 EXPECT_EQ(sizeof(buf1), file->Write(0, buf1, sizeof(buf1))); | |
| 91 EXPECT_EQ(sizeof(buf1), file->GetSize()); | |
| 92 EXPECT_EQ(sizeof(buf1), file->Read(0, buf1, sizeof(buf1))); | |
| 93 | |
| 94 // Attempt to open the same file | |
| 95 EXPECT_EQ(file, mnt->Open(Path("/foo"), O_RDWR | O_CREAT)); | |
| 96 EXPECT_EQ(sizeof(buf1), file->GetSize()); | |
| 97 EXPECT_EQ(3, file->RefCount()); | |
| 98 EXPECT_EQ(2, mnt->num_nodes()); | |
| 99 | |
| 100 // Attempt to close and delete the file | |
| 101 mnt->ReleaseNode(file); | |
| 102 EXPECT_EQ(2, mnt->num_nodes()); | |
| 103 EXPECT_EQ(0, mnt->Unlink(Path("/foo"))); | |
| 104 EXPECT_EQ(2, mnt->num_nodes()); | |
| 105 EXPECT_EQ(-1, mnt->Unlink(Path("/foo"))); | |
| 106 EXPECT_EQ(2, mnt->num_nodes()); | |
| 107 EXPECT_EQ(errno, ENOENT); | |
| 108 mnt->ReleaseNode(file); | |
| 109 EXPECT_EQ(1, mnt->num_nodes()); | |
| 110 | |
| 111 // Recreate foo as a directory | |
| 112 EXPECT_EQ(0, mnt->Mkdir(Path("/foo"), O_RDWR)); | |
| 113 | |
| 114 // Create a file (exclusively) | |
| 115 file = mnt->Open(Path("/foo/bar"), O_RDWR | O_CREAT | O_EXCL); | |
| 116 EXPECT_NE(NULL_NODE, file); | |
| 117 if (NULL == file) return; | |
| 118 | |
| 119 // Attempt to delete the directory | |
| 120 EXPECT_EQ(-1, mnt->Rmdir(Path("/foo"))); | |
| 121 EXPECT_EQ(errno, ENOTEMPTY); | |
| 122 | |
| 123 // Unlink the file, then delete the directory | |
| 124 EXPECT_EQ(0, mnt->Unlink(Path("/foo/bar"))); | |
| 125 EXPECT_EQ(0, mnt->Rmdir(Path("/foo"))); | |
| 126 EXPECT_EQ(2, mnt->num_nodes()); | |
| 127 mnt->ReleaseNode(file); | |
| 128 EXPECT_EQ(1, mnt->num_nodes()); | |
| 129 | |
| 130 // Verify the directory is gone | |
| 131 file = mnt->Open(Path("/foo"), O_RDWR); | |
| 132 EXPECT_EQ(NULL_NODE, file); | |
| 133 EXPECT_EQ(errno, ENOENT); | |
| 134 } | |
| 135 | |
| 136 TEST(MountTest, MemMountRemove) { | |
| 137 MountMemMock* mnt = new MountMemMock(); | |
| 138 MountNode* file; | |
| 139 | |
| 140 EXPECT_EQ(0, mnt->Mkdir(Path("/dir"), O_RDWR)); | |
| 141 file = mnt->Open(Path("/file"), O_RDWR | O_CREAT | O_EXCL); | |
| 142 EXPECT_NE(NULL_NODE, file); | |
| 143 mnt->ReleaseNode(file); | |
| 144 | |
| 145 EXPECT_EQ(0, mnt->Remove(Path("/dir"))); | |
| 146 EXPECT_EQ(0, mnt->Remove(Path("/file"))); | |
| 147 | |
| 148 EXPECT_EQ(NULL_NODE, mnt->Open(Path("/dir/foo"), O_CREAT | O_RDWR)); | |
| 149 EXPECT_EQ(ENOENT, errno); | |
| 150 EXPECT_EQ(NULL_NODE, mnt->Open(Path("/file"), O_RDONLY)); | |
| 151 EXPECT_EQ(ENOENT, errno); | |
| 152 } | |
| 153 | |
| 154 TEST(MountTest, DevNull) { | |
| 155 MountDevMock* mnt = new MountDevMock(); | |
| 156 MountNode* dev_null = mnt->Open(Path("/null"), O_RDWR); | |
| 157 ASSERT_NE(NULL_NODE, dev_null); | |
| 158 | |
| 159 // Writing to /dev/null should write everything. | |
| 160 const char msg[] = "Dummy test message."; | |
| 161 EXPECT_EQ(strlen(msg), dev_null->Write(0, &msg[0], strlen(msg))); | |
| 162 | |
| 163 // Reading from /dev/null should read nothing. | |
| 164 const int kBufferLength = 100; | |
| 165 char buffer[kBufferLength]; | |
| 166 EXPECT_EQ(0, dev_null->Read(0, &buffer[0], kBufferLength)); | |
| 167 mnt->ReleaseNode(dev_null); | |
| 168 } | |
| 169 | |
| 170 TEST(MountTest, DevZero) { | |
| 171 MountDevMock* mnt = new MountDevMock(); | |
| 172 MountNode* dev_zero = mnt->Open(Path("/zero"), O_RDWR); | |
| 173 ASSERT_NE(NULL_NODE, dev_zero); | |
| 174 | |
| 175 // Writing to /dev/zero should write everything. | |
| 176 const char msg[] = "Dummy test message."; | |
| 177 EXPECT_EQ(strlen(msg), dev_zero->Write(0, &msg[0], strlen(msg))); | |
| 178 | |
| 179 // Reading from /dev/zero should read all zeroes. | |
| 180 const int kBufferLength = 100; | |
| 181 char buffer[kBufferLength]; | |
| 182 // First fill with all 1s. | |
| 183 memset(&buffer[0], 0x1, kBufferLength); | |
| 184 EXPECT_EQ(kBufferLength, dev_zero->Read(0, &buffer[0], kBufferLength)); | |
| 185 | |
| 186 char zero_buffer[kBufferLength]; | |
| 187 memset(&zero_buffer[0], 0, kBufferLength); | |
| 188 EXPECT_EQ(0, memcmp(&buffer[0], &zero_buffer[0], kBufferLength)); | |
| 189 mnt->ReleaseNode(dev_zero); | |
| 190 } | |
| 191 | |
| 192 TEST(MountTest, DevUrandom) { | |
| 193 MountDevMock* mnt = new MountDevMock(); | |
| 194 MountNode* dev_urandom = mnt->Open(Path("/urandom"), O_RDWR); | |
| 195 ASSERT_NE(NULL_NODE, dev_urandom); | |
| 196 | |
| 197 // Writing to /dev/zero should write everything. | |
| 198 const char msg[] = "Dummy test message."; | |
| 199 EXPECT_EQ(strlen(msg), dev_urandom->Write(0, &msg[0], strlen(msg))); | |
| 200 | |
| 201 // Reading from /dev/urandom should read random bytes. | |
| 202 const int kSampleBatches = 1000; | |
| 203 const int kSampleBatchSize = 1000; | |
| 204 const int kTotalSamples = kSampleBatches * kSampleBatchSize; | |
| 205 | |
| 206 int byte_count[256] = {0}; | |
| 207 | |
| 208 unsigned char buffer[kSampleBatchSize]; | |
| 209 for (int batch = 0; batch < kSampleBatches; ++batch) { | |
| 210 int bytes_read = dev_urandom->Read(0, &buffer[0], kSampleBatchSize); | |
| 211 EXPECT_EQ(kSampleBatchSize, bytes_read); | |
| 212 | |
| 213 for (int i = 0; i < bytes_read; ++i) { | |
| 214 byte_count[buffer[i]]++; | |
| 215 } | |
| 216 } | |
| 217 | |
| 218 double expected_count = kTotalSamples / 256.; | |
| 219 double chi_squared = 0; | |
| 220 for (int i = 0; i < 256; ++i) { | |
| 221 double difference = byte_count[i] - expected_count; | |
| 222 chi_squared += difference * difference / expected_count; | |
| 223 } | |
| 224 | |
| 225 // Approximate chi-squared value for p-value 0.05, 255 degrees-of-freedom. | |
| 226 EXPECT_LE(chi_squared, 293.24); | |
| 227 } | |
| OLD | NEW |