blob: dd805e56edbe3226c56871a8a43624066d024590 [file] [log] [blame]
Jed Estepa7b9a462015-12-15 16:04:53 -08001/*
2 * Copyright (C) 2015 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
Jed Estepa7b9a462015-12-15 16:04:53 -080017#include <map>
Jed Estepa7b9a462015-12-15 16:04:53 -080018
19#include <errno.h>
20#include <fcntl.h>
21#include <stdio.h>
22#include <sys/stat.h>
23#include <unistd.h>
24
Jed Estep39c1b5e2015-12-15 16:04:53 -080025#include "config.h"
Jed Estepa7b9a462015-12-15 16:04:53 -080026#include "ota_io.h"
27
Jed Estep39c1b5e2015-12-15 16:04:53 -080028static std::map<intptr_t, const char*> filename_cache;
29static std::string read_fault_file_name = "";
30static std::string write_fault_file_name = "";
31static std::string fsync_fault_file_name = "";
Tianjie Xu3c62b672016-02-05 18:25:58 -080032bool have_eio_error = false;
Jed Estep39c1b5e2015-12-15 16:04:53 -080033
34static bool get_hit_file(const char* cached_path, std::string ffn) {
35 return should_hit_cache()
36 ? !strncmp(cached_path, OTAIO_CACHE_FNAME, strlen(cached_path))
37 : !strncmp(cached_path, ffn.c_str(), strlen(cached_path));
38}
39
40void ota_set_fault_files() {
41 if (should_fault_inject(OTAIO_READ)) {
42 read_fault_file_name = fault_fname(OTAIO_READ);
43 }
44 if (should_fault_inject(OTAIO_WRITE)) {
45 write_fault_file_name = fault_fname(OTAIO_WRITE);
46 }
47 if (should_fault_inject(OTAIO_FSYNC)) {
48 fsync_fault_file_name = fault_fname(OTAIO_FSYNC);
49 }
50}
Jed Estepa7b9a462015-12-15 16:04:53 -080051
52int ota_open(const char* path, int oflags) {
Jed Estepa7b9a462015-12-15 16:04:53 -080053 // Let the caller handle errors; we do not care if open succeeds or fails
54 int fd = open(path, oflags);
Jed Estep39c1b5e2015-12-15 16:04:53 -080055 filename_cache[fd] = path;
Jed Estepa7b9a462015-12-15 16:04:53 -080056 return fd;
Jed Estepa7b9a462015-12-15 16:04:53 -080057}
58
59int ota_open(const char* path, int oflags, mode_t mode) {
Jed Estepa7b9a462015-12-15 16:04:53 -080060 int fd = open(path, oflags, mode);
Jed Estep39c1b5e2015-12-15 16:04:53 -080061 filename_cache[fd] = path;
62 return fd; }
Jed Estepa7b9a462015-12-15 16:04:53 -080063
64FILE* ota_fopen(const char* path, const char* mode) {
Jed Estepa7b9a462015-12-15 16:04:53 -080065 FILE* fh = fopen(path, mode);
Jed Estep39c1b5e2015-12-15 16:04:53 -080066 filename_cache[(intptr_t)fh] = path;
Jed Estepa7b9a462015-12-15 16:04:53 -080067 return fh;
Jed Estepa7b9a462015-12-15 16:04:53 -080068}
69
70int ota_close(int fd) {
Jed Estep39c1b5e2015-12-15 16:04:53 -080071 // descriptors can be reused, so make sure not to leave them in the cache
72 filename_cache.erase(fd);
Jed Estepa7b9a462015-12-15 16:04:53 -080073 return close(fd);
74}
75
76int ota_fclose(FILE* fh) {
Jed Estep39c1b5e2015-12-15 16:04:53 -080077 filename_cache.erase((intptr_t)fh);
Jed Estepa7b9a462015-12-15 16:04:53 -080078 return fclose(fh);
79}
80
81size_t ota_fread(void* ptr, size_t size, size_t nitems, FILE* stream) {
Jed Estep39c1b5e2015-12-15 16:04:53 -080082 if (should_fault_inject(OTAIO_READ)) {
83 auto cached = filename_cache.find((intptr_t)stream);
84 const char* cached_path = cached->second;
85 if (cached != filename_cache.end() &&
86 get_hit_file(cached_path, read_fault_file_name)) {
87 read_fault_file_name = "";
88 errno = EIO;
Tianjie Xu3c62b672016-02-05 18:25:58 -080089 have_eio_error = true;
Jed Estep39c1b5e2015-12-15 16:04:53 -080090 return 0;
91 }
Jed Estepa7b9a462015-12-15 16:04:53 -080092 }
Tianjie Xu3c62b672016-02-05 18:25:58 -080093 size_t status = fread(ptr, size, nitems, stream);
94 // If I/O error occurs, set the retry-update flag.
95 if (status != nitems && errno == EIO) {
96 have_eio_error = true;
97 }
98 return status;
Jed Estepa7b9a462015-12-15 16:04:53 -080099}
100
101ssize_t ota_read(int fd, void* buf, size_t nbyte) {
Jed Estep39c1b5e2015-12-15 16:04:53 -0800102 if (should_fault_inject(OTAIO_READ)) {
103 auto cached = filename_cache.find(fd);
104 const char* cached_path = cached->second;
105 if (cached != filename_cache.end()
106 && get_hit_file(cached_path, read_fault_file_name)) {
107 read_fault_file_name = "";
108 errno = EIO;
Tianjie Xu3c62b672016-02-05 18:25:58 -0800109 have_eio_error = true;
Jed Estep39c1b5e2015-12-15 16:04:53 -0800110 return -1;
111 }
Jed Estepa7b9a462015-12-15 16:04:53 -0800112 }
Tianjie Xu3c62b672016-02-05 18:25:58 -0800113 ssize_t status = read(fd, buf, nbyte);
114 if (status == -1 && errno == EIO) {
115 have_eio_error = true;
116 }
117 return status;
Jed Estepa7b9a462015-12-15 16:04:53 -0800118}
119
120size_t ota_fwrite(const void* ptr, size_t size, size_t count, FILE* stream) {
Jed Estep39c1b5e2015-12-15 16:04:53 -0800121 if (should_fault_inject(OTAIO_WRITE)) {
122 auto cached = filename_cache.find((intptr_t)stream);
123 const char* cached_path = cached->second;
124 if (cached != filename_cache.end() &&
125 get_hit_file(cached_path, write_fault_file_name)) {
126 write_fault_file_name = "";
127 errno = EIO;
Tianjie Xu3c62b672016-02-05 18:25:58 -0800128 have_eio_error = true;
Jed Estep39c1b5e2015-12-15 16:04:53 -0800129 return 0;
130 }
Jed Estepa7b9a462015-12-15 16:04:53 -0800131 }
Tianjie Xu3c62b672016-02-05 18:25:58 -0800132 size_t status = fwrite(ptr, size, count, stream);
133 if (status != count && errno == EIO) {
134 have_eio_error = true;
135 }
136 return status;
Jed Estepa7b9a462015-12-15 16:04:53 -0800137}
138
139ssize_t ota_write(int fd, const void* buf, size_t nbyte) {
Jed Estep39c1b5e2015-12-15 16:04:53 -0800140 if (should_fault_inject(OTAIO_WRITE)) {
141 auto cached = filename_cache.find(fd);
142 const char* cached_path = cached->second;
143 if (cached != filename_cache.end() &&
144 get_hit_file(cached_path, write_fault_file_name)) {
145 write_fault_file_name = "";
146 errno = EIO;
Tianjie Xu3c62b672016-02-05 18:25:58 -0800147 have_eio_error = true;
Jed Estep39c1b5e2015-12-15 16:04:53 -0800148 return -1;
149 }
Jed Estepa7b9a462015-12-15 16:04:53 -0800150 }
Tianjie Xu3c62b672016-02-05 18:25:58 -0800151 ssize_t status = write(fd, buf, nbyte);
152 if (status == -1 && errno == EIO) {
153 have_eio_error = true;
154 }
155 return status;
Jed Estepa7b9a462015-12-15 16:04:53 -0800156}
157
158int ota_fsync(int fd) {
Jed Estep39c1b5e2015-12-15 16:04:53 -0800159 if (should_fault_inject(OTAIO_FSYNC)) {
160 auto cached = filename_cache.find(fd);
161 const char* cached_path = cached->second;
162 if (cached != filename_cache.end() &&
163 get_hit_file(cached_path, fsync_fault_file_name)) {
164 fsync_fault_file_name = "";
165 errno = EIO;
Tianjie Xu3c62b672016-02-05 18:25:58 -0800166 have_eio_error = true;
Jed Estep39c1b5e2015-12-15 16:04:53 -0800167 return -1;
168 }
Jed Estepa7b9a462015-12-15 16:04:53 -0800169 }
Tianjie Xu3c62b672016-02-05 18:25:58 -0800170 int status = fsync(fd);
171 if (status == -1 && errno == EIO) {
172 have_eio_error = true;
173 }
174 return status;
Jed Estepa7b9a462015-12-15 16:04:53 -0800175}
Jed Estep39c1b5e2015-12-15 16:04:53 -0800176