Serenity Operating System
at hosted 342 lines 8.5 kB view raw
1/* 2 * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright notice, this 9 * list of conditions and the following disclaimer. 10 * 11 * 2. Redistributions in binary form must reproduce the above copyright notice, 12 * this list of conditions and the following disclaimer in the documentation 13 * and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 18 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 21 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 22 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 23 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 24 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27#include <AK/BufferStream.h> 28#include <Kernel/Devices/BlockDevice.h> 29#include <Kernel/Devices/CharacterDevice.h> 30#include <Kernel/FileSystem/Custody.h> 31#include <Kernel/FileSystem/FIFO.h> 32#include <Kernel/FileSystem/FileDescription.h> 33#include <Kernel/FileSystem/FileSystem.h> 34#include <Kernel/FileSystem/InodeFile.h> 35#include <Kernel/Net/Socket.h> 36#include <Kernel/Process.h> 37#include <Kernel/TTY/MasterPTY.h> 38#include <Kernel/TTY/TTY.h> 39#include <Kernel/UnixTypes.h> 40#include <Kernel/VM/MemoryManager.h> 41#include <LibC/errno_numbers.h> 42 43namespace Kernel { 44 45NonnullRefPtr<FileDescription> FileDescription::create(Custody& custody) 46{ 47 auto description = adopt(*new FileDescription(InodeFile::create(custody.inode()))); 48 description->m_custody = custody; 49 return description; 50} 51 52NonnullRefPtr<FileDescription> FileDescription::create(File& file) 53{ 54 return adopt(*new FileDescription(file)); 55} 56 57FileDescription::FileDescription(File& file) 58 : m_file(file) 59{ 60 if (file.is_inode()) 61 m_inode = static_cast<InodeFile&>(file).inode(); 62 if (is_socket()) 63 socket()->attach(*this); 64 m_is_directory = metadata().is_directory(); 65} 66 67FileDescription::~FileDescription() 68{ 69 if (is_socket()) 70 socket()->detach(*this); 71 if (is_fifo()) 72 static_cast<FIFO*>(m_file.ptr())->detach(m_fifo_direction); 73 m_file->close(); 74 m_inode = nullptr; 75} 76 77KResult FileDescription::fstat(stat& buffer) 78{ 79 if (is_fifo()) { 80 memset(&buffer, 0, sizeof(buffer)); 81 buffer.st_mode = 001000; 82 return KSuccess; 83 } 84 if (is_socket()) { 85 memset(&buffer, 0, sizeof(buffer)); 86 buffer.st_mode = 0140000; 87 return KSuccess; 88 } 89 90 if (!m_inode) 91 return KResult(-EBADF); 92 return metadata().stat(buffer); 93} 94 95off_t FileDescription::seek(off_t offset, int whence) 96{ 97 LOCKER(m_lock); 98 if (!m_file->is_seekable()) 99 return -EINVAL; 100 101 off_t new_offset; 102 103 switch (whence) { 104 case SEEK_SET: 105 new_offset = offset; 106 break; 107 case SEEK_CUR: 108 new_offset = m_current_offset + offset; 109 break; 110 case SEEK_END: 111 if (!metadata().is_valid()) 112 return -EIO; 113 new_offset = metadata().size; 114 break; 115 default: 116 return -EINVAL; 117 } 118 119 if (new_offset < 0) 120 return -EINVAL; 121 // FIXME: Return -EINVAL if attempting to seek past the end of a seekable device. 122 123 m_current_offset = new_offset; 124 return m_current_offset; 125} 126 127ssize_t FileDescription::read(u8* buffer, ssize_t count) 128{ 129 LOCKER(m_lock); 130 if ((m_current_offset + count) < 0) 131 return -EOVERFLOW; 132 SmapDisabler disabler; 133 int nread = m_file->read(*this, buffer, count); 134 if (nread > 0 && m_file->is_seekable()) 135 m_current_offset += nread; 136 return nread; 137} 138 139ssize_t FileDescription::write(const u8* data, ssize_t size) 140{ 141 LOCKER(m_lock); 142 if ((m_current_offset + size) < 0) 143 return -EOVERFLOW; 144 SmapDisabler disabler; 145 int nwritten = m_file->write(*this, data, size); 146 if (nwritten > 0 && m_file->is_seekable()) 147 m_current_offset += nwritten; 148 return nwritten; 149} 150 151bool FileDescription::can_write() const 152{ 153 return m_file->can_write(*this); 154} 155 156bool FileDescription::can_read() const 157{ 158 return m_file->can_read(*this); 159} 160 161ByteBuffer FileDescription::read_entire_file() 162{ 163 // HACK ALERT: (This entire function) 164 ASSERT(m_file->is_inode()); 165 ASSERT(m_inode); 166 return m_inode->read_entire(this); 167} 168 169ssize_t FileDescription::get_dir_entries(u8* buffer, ssize_t size) 170{ 171 LOCKER(m_lock); 172 if (!is_directory()) 173 return -ENOTDIR; 174 175 auto metadata = this->metadata(); 176 if (!metadata.is_valid()) 177 return -EIO; 178 179 if (size < 0) 180 return -EINVAL; 181 182 size_t size_to_allocate = max(PAGE_SIZE, metadata.size); 183 184 auto temp_buffer = ByteBuffer::create_uninitialized(size_to_allocate); 185 BufferStream stream(temp_buffer); 186 VFS::the().traverse_directory_inode(*m_inode, [&stream](auto& entry) { 187 stream << (u32)entry.inode.index(); 188 stream << (u8)entry.file_type; 189 stream << (u32)entry.name_length; 190 stream << entry.name; 191 return true; 192 }); 193 stream.snip(); 194 195 if (static_cast<size_t>(size) < temp_buffer.size()) 196 return -EINVAL; 197 198 copy_to_user(buffer, temp_buffer.data(), temp_buffer.size()); 199 return stream.offset(); 200} 201 202bool FileDescription::is_device() const 203{ 204 return m_file->is_device(); 205} 206 207const Device* FileDescription::device() const 208{ 209 if (!is_device()) 210 return nullptr; 211 return static_cast<const Device*>(m_file.ptr()); 212} 213 214Device* FileDescription::device() 215{ 216 if (!is_device()) 217 return nullptr; 218 return static_cast<Device*>(m_file.ptr()); 219} 220 221bool FileDescription::is_tty() const 222{ 223 return m_file->is_tty(); 224} 225 226const TTY* FileDescription::tty() const 227{ 228 if (!is_tty()) 229 return nullptr; 230 return static_cast<const TTY*>(m_file.ptr()); 231} 232 233TTY* FileDescription::tty() 234{ 235 if (!is_tty()) 236 return nullptr; 237 return static_cast<TTY*>(m_file.ptr()); 238} 239 240bool FileDescription::is_master_pty() const 241{ 242 return m_file->is_master_pty(); 243} 244 245const MasterPTY* FileDescription::master_pty() const 246{ 247 if (!is_master_pty()) 248 return nullptr; 249 return static_cast<const MasterPTY*>(m_file.ptr()); 250} 251 252MasterPTY* FileDescription::master_pty() 253{ 254 if (!is_master_pty()) 255 return nullptr; 256 return static_cast<MasterPTY*>(m_file.ptr()); 257} 258 259int FileDescription::close() 260{ 261 return 0; 262} 263 264String FileDescription::absolute_path() const 265{ 266 if (m_custody) 267 return m_custody->absolute_path(); 268 return m_file->absolute_path(*this); 269} 270 271InodeMetadata FileDescription::metadata() const 272{ 273 if (m_inode) 274 return m_inode->metadata(); 275 return {}; 276} 277 278KResultOr<Region*> FileDescription::mmap(Process& process, VirtualAddress vaddr, size_t offset, size_t size, int prot, bool shared) 279{ 280 LOCKER(m_lock); 281 return m_file->mmap(process, *this, vaddr, offset, size, prot, shared); 282} 283 284KResult FileDescription::truncate(u64 length) 285{ 286 LOCKER(m_lock); 287 return m_file->truncate(length); 288} 289 290bool FileDescription::is_fifo() const 291{ 292 return m_file->is_fifo(); 293} 294 295FIFO* FileDescription::fifo() 296{ 297 if (!is_fifo()) 298 return nullptr; 299 return static_cast<FIFO*>(m_file.ptr()); 300} 301 302bool FileDescription::is_socket() const 303{ 304 return m_file->is_socket(); 305} 306 307Socket* FileDescription::socket() 308{ 309 if (!is_socket()) 310 return nullptr; 311 return static_cast<Socket*>(m_file.ptr()); 312} 313 314const Socket* FileDescription::socket() const 315{ 316 if (!is_socket()) 317 return nullptr; 318 return static_cast<const Socket*>(m_file.ptr()); 319} 320 321void FileDescription::set_file_flags(u32 flags) 322{ 323 LOCKER(m_lock); 324 m_is_blocking = !(flags & O_NONBLOCK); 325 m_should_append = flags & O_APPEND; 326 m_direct = flags & O_DIRECT; 327 m_file_flags = flags; 328} 329 330KResult FileDescription::chmod(mode_t mode) 331{ 332 LOCKER(m_lock); 333 return m_file->chmod(mode); 334} 335 336KResult FileDescription::chown(uid_t uid, gid_t gid) 337{ 338 LOCKER(m_lock); 339 return m_file->chown(uid, gid); 340} 341 342}