| /* |
| * Copyright (C) 2023 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| #include "client.h" |
| |
| #include <inttypes.h> |
| #include <stddef.h> |
| #include <stdint.h> |
| #include <stdlib.h> |
| |
| #include <interface/storage/storage.h> |
| #include <lib/system_state/system_state.h> |
| #include <lk/list.h> |
| |
| #include "client_session.h" |
| #include "file.h" |
| #include "session.h" |
| #include "storage_limits.h" |
| |
| // macros to help manage debug output |
| #define SS_ERR(args...) fprintf(stderr, "ss: " args) |
| |
| #if 0 |
| // this can generate a lot of spew on debug builds |
| #define SS_INFO(args...) fprintf(stderr, "ss: " args) |
| #else |
| #define SS_INFO(args...) \ |
| do { \ |
| } while (0) |
| #endif |
| |
| /** |
| * checkpoint_update_allowed - Is checkpoint modification currently allowed? |
| * @transaction: Transaction object. |
| * |
| * Check if the given transaction is allowed to create a new or update the |
| * existing checkpoint for its file system. Checkpoint updates may only be |
| * requested by client while the device is in provisioning mode. |
| * |
| * Returns %true if a transaction may update the current checkpoint state, |
| * %false otherwise. |
| */ |
| static bool checkpoint_update_allowed(struct transaction* tr) { |
| return system_state_provisioning_allowed(); |
| } |
| |
| /* |
| * Legal secure storage directory and file names contain only |
| * characters from the following set: [a-z][A-Z][0-9][.-_] |
| * |
| * It is not null terminated. |
| */ |
| static int is_valid_name(const char* name, size_t name_len) { |
| size_t i; |
| |
| if (!name_len) |
| return 0; |
| |
| for (i = 0; i < name_len; i++) { |
| if ((name[i] >= 'a') && (name[i] <= 'z')) |
| continue; |
| if ((name[i] >= 'A') && (name[i] <= 'Z')) |
| continue; |
| if ((name[i] >= '0') && (name[i] <= '9')) |
| continue; |
| if ((name[i] == '.') || (name[i] == '-') || (name[i] == '_')) |
| continue; |
| |
| // not a legal character so reject this name |
| return 0; |
| } |
| |
| return 1; |
| } |
| |
| static int get_path(char* path_out, |
| size_t path_out_size, |
| const uuid_t* uuid, |
| const char* file_name, |
| size_t file_name_len) { |
| unsigned int rc; |
| |
| rc = snprintf(path_out, path_out_size, |
| "%08x%04x%04x%02x%02x%02x%02x%02x%02x%02x%02x/", |
| uuid->time_low, uuid->time_mid, uuid->time_hi_and_version, |
| uuid->clock_seq_and_node[0], uuid->clock_seq_and_node[1], |
| uuid->clock_seq_and_node[2], uuid->clock_seq_and_node[3], |
| uuid->clock_seq_and_node[4], uuid->clock_seq_and_node[5], |
| uuid->clock_seq_and_node[6], uuid->clock_seq_and_node[7]); |
| |
| if (rc + file_name_len >= path_out_size) { |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| memcpy(path_out + rc, file_name, file_name_len); |
| path_out[rc + file_name_len] = '\0'; |
| |
| return STORAGE_NO_ERROR; |
| } |
| |
| static enum storage_err file_op_result_to_storage_err( |
| enum file_op_result result) { |
| switch (result) { |
| case FILE_OP_SUCCESS: |
| return STORAGE_NO_ERROR; |
| case FILE_OP_ERR_FAILED: |
| // TODO: Consider returning STORAGE_ERR_TRANSACT consistently |
| return STORAGE_ERR_GENERIC; |
| case FILE_OP_ERR_EXIST: |
| return STORAGE_ERR_EXIST; |
| case FILE_OP_ERR_ALREADY_OPEN: |
| return STORAGE_ERR_NOT_ALLOWED; |
| case FILE_OP_ERR_NOT_FOUND: |
| return STORAGE_ERR_NOT_FOUND; |
| case FILE_OP_ERR_FS_REPAIRED: |
| return STORAGE_ERR_FS_REPAIRED; |
| } |
| |
| SS_ERR("%s: Unknown file_op_result: %d\n", __func__, result); |
| return STORAGE_ERR_GENERIC; |
| } |
| |
| static enum storage_err session_set_files_count( |
| struct storage_client_session* session, |
| size_t files_count) { |
| struct storage_file_handle** files; |
| |
| if (files_count > STORAGE_MAX_OPEN_FILES) { |
| SS_ERR("%s: too many open files\n", __func__); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| if (!files_count) { |
| free(session->files); |
| session->files = NULL; |
| } else { |
| files = realloc(session->files, sizeof(files[0]) * files_count); |
| if (!files) { |
| SS_ERR("%s: out of memory\n", __func__); |
| return STORAGE_ERR_GENERIC; |
| } |
| if (files_count > session->files_count) |
| memset(files + session->files_count, 0, |
| sizeof(files[0]) * (files_count - session->files_count)); |
| |
| session->files = files; |
| } |
| session->files_count = files_count; |
| |
| SS_INFO("%s: new file table size, 0x%zx\n", __func__, files_count); |
| |
| return STORAGE_NO_ERROR; |
| } |
| |
| static void session_shrink_files(struct storage_client_session* session) { |
| uint32_t handle; |
| |
| handle = session->files_count; |
| while (handle > 0 && !session->files[handle - 1]) |
| handle--; |
| |
| if (handle < session->files_count) |
| session_set_files_count(session, handle); |
| } |
| |
| static void session_close_all_files(struct storage_client_session* session) { |
| uint32_t f_handle; |
| struct storage_file_handle* file; |
| |
| for (f_handle = 0; f_handle < session->files_count; f_handle++) { |
| file = session->files[f_handle]; |
| if (file) { |
| file_close(file); |
| free(file); |
| } |
| } |
| if (session->files) { |
| free(session->files); |
| } |
| session->files_count = 0; |
| } |
| |
| static enum storage_err create_file_handle( |
| struct storage_client_session* session, |
| uint32_t* handlep, |
| struct storage_file_handle** file_p) { |
| enum storage_err result; |
| uint32_t handle; |
| struct storage_file_handle* file; |
| |
| for (handle = 0; handle < session->files_count; handle++) |
| if (!session->files[handle]) |
| break; |
| |
| if (handle >= session->files_count) { |
| result = session_set_files_count(session, handle + 1); |
| if (result != STORAGE_NO_ERROR) |
| return result; |
| } |
| |
| file = calloc(1, sizeof(*file)); |
| if (!file) { |
| SS_ERR("%s: out of memory\n", __func__); |
| return STORAGE_ERR_GENERIC; |
| } |
| |
| session->files[handle] = file; |
| |
| SS_INFO("%s: created file handle 0x%" PRIx32 "\n", __func__, handle); |
| *handlep = handle; |
| *file_p = file; |
| return STORAGE_NO_ERROR; |
| } |
| |
| static void free_file_handle(struct storage_client_session* session, |
| uint32_t handle) { |
| if (handle >= session->files_count) { |
| SS_ERR("%s: invalid handle, 0x%" PRIx32 "\n", __func__, handle); |
| return; |
| } |
| if (session->files[handle] == NULL) { |
| SS_ERR("%s: closed handle, 0x%" PRIx32 "\n", __func__, handle); |
| return; |
| } |
| free(session->files[handle]); |
| session->files[handle] = NULL; |
| |
| SS_INFO("%s: deleted file handle 0x%" PRIx32 "\n", __func__, handle); |
| |
| session_shrink_files(session); |
| } |
| |
| static struct storage_file_handle* get_file_handle( |
| struct storage_client_session* session, |
| uint32_t handle) { |
| struct storage_file_handle* file; |
| if (handle >= session->files_count) { |
| SS_ERR("%s: invalid handle, 0x%" PRIx32 "\n", __func__, handle); |
| return NULL; |
| } |
| file = session->files[handle]; |
| if (!file) { |
| SS_ERR("%s: closed handle, 0x%" PRIx32 "\n", __func__, handle); |
| return NULL; |
| } |
| return file; |
| } |
| |
| static enum storage_err assert_checkpoint_flag_valid( |
| struct storage_client_session* session, |
| struct storage_op_flags flags, |
| const char* func_name) { |
| if (flags.update_checkpoint) { |
| if (!(flags.complete_transaction)) { |
| SS_ERR("%s: STORAGE_MSG_FLAG_TRANSACT_CHECKPOINT cannot " |
| "be used without STORAGE_MSG_FLAG_TRANSACT_COMPLETE\n", |
| func_name); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| if (!checkpoint_update_allowed(&session->tr)) { |
| SS_ERR("%s: Checkpoint requested but not currently allowed.\n", |
| func_name); |
| return STORAGE_ERR_NOT_ALLOWED; |
| } |
| } |
| return STORAGE_NO_ERROR; |
| } |
| |
| static enum storage_err ensure_active_transaction( |
| struct storage_client_session* session, |
| struct storage_op_flags flags) { |
| if (session->tr.failed) { |
| if (flags.complete_transaction) { |
| /* last command in current transaction: |
| * reset failed state and return error */ |
| session->tr.failed = false; |
| } |
| return STORAGE_ERR_TRANSACT; |
| } |
| |
| if (!transaction_is_active(&session->tr)) { |
| /* previous transaction complete */ |
| transaction_activate(&session->tr); |
| } |
| return STORAGE_NO_ERROR; |
| } |
| |
| void storage_client_session_init(struct storage_client_session* session, |
| struct fs* fs, |
| const uuid_t* peer) { |
| session->magic = STORAGE_CLIENT_SESSION_MAGIC; |
| session->files = NULL; |
| session->files_count = 0; |
| |
| transaction_init(&session->tr, fs, false); |
| |
| /* cache identity information */ |
| memcpy(&session->uuid, peer, sizeof(*peer)); |
| } |
| |
| void storage_client_session_destroy(struct storage_client_session* session) { |
| if (list_in_list(&session->tr.allocated.node) && !session->tr.failed) { |
| /* discard partial transaction */ |
| transaction_fail(&session->tr); |
| } |
| session_close_all_files(session); |
| transaction_free(&session->tr); |
| } |
| |
| /* abort transaction and clear sticky transaction error */ |
| enum storage_err storage_transaction_end(struct storage_client_session* session, |
| struct storage_op_flags flags) { |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| if (flags.complete_transaction) { |
| /* Allow checkpoint creation without an active transaction */ |
| if (flags.update_checkpoint && !transaction_is_active(&session->tr)) { |
| transaction_activate(&session->tr); |
| } |
| /* try to complete current transaction */ |
| if (transaction_is_active(&session->tr)) { |
| transaction_complete_etc(&session->tr, flags.update_checkpoint); |
| } |
| if (session->tr.failed) { |
| SS_ERR("%s: failed to complete transaction\n", __func__); |
| /* clear transaction failed state */ |
| session->tr.failed = false; |
| return STORAGE_ERR_TRANSACT; |
| } |
| return STORAGE_NO_ERROR; |
| } |
| |
| /* discard current transaction */ |
| if (transaction_is_active(&session->tr)) { |
| transaction_fail(&session->tr); |
| } |
| /* clear transaction failed state */ |
| session->tr.failed = false; |
| return STORAGE_NO_ERROR; |
| } |
| |
| enum storage_err storage_file_delete(struct storage_client_session* session, |
| const char* fname, |
| size_t fname_len, |
| struct storage_op_flags flags) { |
| enum file_op_result delete_res; |
| char path_buf[FS_PATH_MAX]; |
| |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| result = ensure_active_transaction(session, flags); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| /* make sure filename is legal */ |
| if (!is_valid_name(fname, fname_len)) { |
| SS_ERR("%s: invalid filename\n", __func__); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| result = get_path(path_buf, sizeof(path_buf), &session->uuid, fname, |
| fname_len); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| SS_INFO("%s: path %s\n", __func__, path_buf); |
| |
| delete_res = file_delete(&session->tr, path_buf, flags.allow_repaired); |
| |
| if (delete_res != FILE_OP_SUCCESS) { |
| return file_op_result_to_storage_err(delete_res); |
| } |
| |
| if (flags.complete_transaction) { |
| transaction_complete_etc(&session->tr, flags.update_checkpoint); |
| if (session->tr.failed) { |
| SS_ERR("%s: transaction commit failed\n", __func__); |
| return STORAGE_ERR_GENERIC; |
| } |
| } |
| |
| return STORAGE_NO_ERROR; |
| } |
| |
| /** |
| * storage_file_check_name - Check if file handle matches path |
| * @tr: Transaction object. |
| * @file: File handle object. |
| * @path: Path to check. |
| * |
| * Return: %true if @file matches @path, %false otherwise. |
| */ |
| static bool storage_file_check_name(struct transaction* tr, |
| const struct storage_file_handle* file, |
| const char* path) { |
| bool ret; |
| const struct file_info* file_info; |
| struct obj_ref ref = OBJ_REF_INITIAL_VALUE(ref); |
| |
| file_info = file_get_info(tr, &file->block_mac, &ref); |
| if (!file_info) { |
| printf("can't read file entry at %" PRIu64 "\n", |
| block_mac_to_block(tr, &file->block_mac)); |
| return false; |
| } |
| assert(file_info); |
| ret = strcmp(file_info->path, path) == 0; |
| file_info_put(file_info, &ref); |
| |
| return ret; |
| } |
| |
| enum storage_err storage_file_move(struct storage_client_session* session, |
| uint32_t handle, |
| bool src_already_opened, |
| const char* src_name, |
| size_t src_name_len, |
| const char* dst_name, |
| size_t dst_name_len, |
| enum file_create_mode dst_file_create_mode, |
| struct storage_op_flags flags) { |
| enum file_op_result open_result; |
| enum file_op_result move_result; |
| struct storage_file_handle* file = NULL; |
| char path_buf[FS_PATH_MAX]; |
| struct storage_file_handle tmp_file; |
| |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| result = ensure_active_transaction(session, flags); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| /* make sure filenames are legal */ |
| if (src_name && !is_valid_name(src_name, src_name_len)) { |
| SS_ERR("%s: invalid src filename\n", __func__); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| if (!is_valid_name(dst_name, dst_name_len)) { |
| SS_ERR("%s: invalid dst filename\n", __func__); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| if (!src_name && !src_already_opened) { |
| SS_ERR("%s: src needs to be opened, but no src name provided\n", |
| __func__); |
| } |
| |
| if (src_already_opened) { |
| file = get_file_handle(session, handle); |
| if (!file) |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| if (src_name) { |
| result = get_path(path_buf, sizeof(path_buf), &session->uuid, src_name, |
| src_name_len); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| } |
| |
| SS_INFO("%s: src path %s\n", __func__, path_buf); |
| |
| if (file) { |
| if (src_name && |
| !storage_file_check_name(&session->tr, file, path_buf)) { |
| return STORAGE_ERR_NOT_VALID; |
| } |
| } else { |
| open_result = file_open(&session->tr, path_buf, &tmp_file, |
| FILE_OPEN_NO_CREATE, flags.allow_repaired); |
| if (open_result != FILE_OP_SUCCESS) { |
| return file_op_result_to_storage_err(open_result); |
| } |
| file = &tmp_file; |
| } |
| |
| result = get_path(path_buf, sizeof(path_buf), &session->uuid, dst_name, |
| dst_name_len); |
| if (result != STORAGE_NO_ERROR) { |
| if (file == &tmp_file) { |
| file_close(&tmp_file); |
| } |
| return result; |
| } |
| SS_INFO("%s: dst path %s\n", __func__, path_buf); |
| |
| move_result = file_move(&session->tr, file, path_buf, dst_file_create_mode, |
| flags.allow_repaired); |
| if (file == &tmp_file) { |
| file_close(&tmp_file); |
| } |
| |
| if (move_result != FILE_OP_SUCCESS) { |
| return file_op_result_to_storage_err(move_result); |
| } |
| |
| if (flags.complete_transaction) { |
| transaction_complete_etc(&session->tr, flags.update_checkpoint); |
| if (session->tr.failed) { |
| SS_ERR("%s: transaction commit failed\n", __func__); |
| return STORAGE_ERR_GENERIC; |
| } |
| } |
| |
| return STORAGE_NO_ERROR; |
| } |
| |
| enum storage_err storage_file_open(struct storage_client_session* session, |
| const char* fname, |
| size_t fname_len, |
| enum file_create_mode file_create_mode, |
| bool truncate, |
| struct storage_op_flags flags, |
| uint32_t* handle) { |
| enum file_op_result open_result; |
| struct storage_file_handle* file = NULL; |
| uint32_t f_handle; |
| char path_buf[FS_PATH_MAX]; |
| |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| result = ensure_active_transaction(session, flags); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| /* make sure filename is legal */ |
| if (!is_valid_name(fname, fname_len)) { |
| SS_ERR("%s: invalid filename\n", __func__); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| result = get_path(path_buf, sizeof(path_buf), &session->uuid, fname, |
| fname_len); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| SS_INFO("%s: path %s (create_mode: %d, truncate: %d)\n", __func__, path_buf, |
| file_create_mode, truncate); |
| |
| SS_INFO("%s: call create_file_handle\n", __func__); |
| /* alloc file info struct */ |
| result = create_file_handle(session, &f_handle, &file); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| open_result = file_open(&session->tr, path_buf, file, file_create_mode, |
| flags.allow_repaired); |
| if (open_result != FILE_OP_SUCCESS) { |
| result = file_op_result_to_storage_err(open_result); |
| goto err_open_file; |
| } |
| |
| if (truncate && file->size) { |
| file_set_size(&session->tr, file, 0); |
| } |
| |
| if (session->tr.failed) { |
| SS_ERR("%s: transaction failed\n", __func__); |
| result = STORAGE_ERR_GENERIC; |
| goto err_transaction_failed; |
| } |
| |
| if (flags.complete_transaction) { |
| transaction_complete_etc(&session->tr, flags.update_checkpoint); |
| if (session->tr.failed) { |
| SS_ERR("%s: transaction commit failed\n", __func__); |
| result = STORAGE_ERR_GENERIC; |
| goto err_transaction_failed; |
| } |
| } |
| |
| *handle = f_handle; |
| return STORAGE_NO_ERROR; |
| |
| err_transaction_failed: |
| file_close(file); |
| err_open_file: |
| free_file_handle(session, f_handle); |
| return result; |
| } |
| |
| enum storage_err storage_file_close(struct storage_client_session* session, |
| uint32_t handle, |
| struct storage_op_flags flags) { |
| struct storage_file_handle* file; |
| |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| result = ensure_active_transaction(session, flags); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| file = get_file_handle(session, handle); |
| if (!file) { |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| file_close(file); |
| free_file_handle(session, handle); |
| |
| if (flags.complete_transaction) { |
| transaction_complete_etc(&session->tr, flags.update_checkpoint); |
| if (session->tr.failed) { |
| SS_ERR("%s: transaction commit failed\n", __func__); |
| return STORAGE_ERR_GENERIC; |
| } |
| } |
| |
| return STORAGE_NO_ERROR; |
| } |
| |
| enum storage_err storage_file_read(struct storage_client_session* session, |
| uint32_t handle, |
| uint32_t size, |
| uint64_t offset, |
| struct storage_op_flags flags, |
| uint8_t* resp, |
| size_t* resp_len) { |
| void* bufp = resp; |
| size_t buflen; |
| size_t bytes_left, len; |
| struct storage_file_handle* file; |
| size_t block_size = get_file_block_size(session->tr.fs); |
| data_block_t block_num; |
| const uint8_t* block_data; |
| struct obj_ref block_data_ref = OBJ_REF_INITIAL_VALUE(block_data_ref); |
| size_t block_offset; |
| |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| result = ensure_active_transaction(session, flags); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| file = get_file_handle(session, handle); |
| if (!file) { |
| SS_ERR("%s: invalid file handle (%" PRIx32 ")\n", __func__, handle); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| buflen = size; |
| if (buflen > *resp_len) { |
| SS_ERR("can't read more than %zu bytes, requested %zu\n", *resp_len, |
| buflen); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| if (offset > file->size) { |
| SS_ERR("can't read past end of file (%" PRIu64 " > %" PRIu64 ")\n", |
| offset, file->size); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| /* calc number of bytes to read */ |
| if ((offset + buflen) > file->size) { |
| bytes_left = (size_t)(file->size - offset); |
| } else { |
| bytes_left = buflen; |
| } |
| buflen = bytes_left; /* save to return it to caller */ |
| |
| SS_INFO("%s: start 0x%" PRIx64 " cnt %zu\n", __func__, offset, bytes_left); |
| |
| while (bytes_left) { |
| block_num = offset / block_size; |
| block_data = |
| file_get_block(&session->tr, file, block_num, &block_data_ref); |
| block_offset = offset % block_size; |
| len = (block_offset + bytes_left > block_size) |
| ? block_size - block_offset |
| : bytes_left; |
| if (!block_data) { |
| if (session->tr.failed) { |
| SS_ERR("error reading block %" PRIu64 "\n", block_num); |
| return STORAGE_ERR_GENERIC; |
| } |
| memset(bufp, 0, len); |
| } else { |
| memcpy(bufp, block_data + block_offset, len); |
| file_block_put(block_data, &block_data_ref); |
| } |
| |
| bytes_left -= len; |
| offset += len; |
| bufp += len; |
| } |
| |
| *resp_len = buflen; |
| return STORAGE_NO_ERROR; |
| } |
| |
| static enum storage_err storage_create_gap( |
| struct storage_client_session* session, |
| struct storage_file_handle* file) { |
| size_t block_size = get_file_block_size(session->tr.fs); |
| data_block_t block_num; |
| size_t block_offset; |
| uint8_t* block_data; |
| struct obj_ref block_data_ref = OBJ_REF_INITIAL_VALUE(block_data_ref); |
| |
| block_num = file->size / block_size; |
| block_offset = file->size % block_size; |
| |
| if (block_offset) { |
| /* |
| * The file does not currently end on a block boundary. |
| * We don't clear data in partial blocks when truncating |
| * a file, so the last block could contain data that |
| * should not be readable. We unconditionally clear the |
| * exposed data when creating gaps in the file, as we |
| * don't know if that data is already clear. |
| */ |
| block_data = file_get_block_write(&session->tr, file, block_num, true, |
| &block_data_ref); |
| if (!block_data) { |
| SS_ERR("error getting block %" PRIu64 "\n", block_num); |
| return STORAGE_ERR_GENERIC; |
| } |
| |
| memset(block_data + block_offset, 0, block_size - block_offset); |
| file_block_put_dirty(&session->tr, file, block_num, block_data, |
| &block_data_ref); |
| |
| SS_INFO("%s: clear block at old size 0x%" PRIx64 |
| ", block_offset 0x%zx\n", |
| __func__, file->size, block_offset); |
| } |
| return STORAGE_NO_ERROR; |
| } |
| |
| enum storage_err storage_file_write(struct storage_client_session* session, |
| uint32_t handle, |
| uint64_t offset, |
| const uint8_t* data, |
| size_t data_len, |
| struct storage_op_flags flags) { |
| size_t len; |
| struct storage_file_handle* file; |
| size_t block_size = get_file_block_size(session->tr.fs); |
| data_block_t block_num; |
| uint8_t* block_data; |
| struct obj_ref block_data_ref = OBJ_REF_INITIAL_VALUE(block_data_ref); |
| size_t block_offset; |
| |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| result = ensure_active_transaction(session, flags); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| file = get_file_handle(session, handle); |
| if (!file) { |
| SS_ERR("%s: invalid file handle (%" PRIx32 ")\n", __func__, handle); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| if (offset > file->size) { |
| result = storage_create_gap(session, file); |
| if (result != STORAGE_NO_ERROR) { |
| goto err_write; |
| } |
| } |
| |
| /* transfer data one ss block at a time */ |
| while (data_len) { |
| block_num = offset / block_size; |
| block_offset = offset % block_size; |
| len = (block_offset + data_len > block_size) ? block_size - block_offset |
| : data_len; |
| |
| block_data = file_get_block_write(&session->tr, file, block_num, |
| len != block_size, &block_data_ref); |
| if (!block_data) { |
| SS_ERR("error getting block %" PRIu64 "\n", block_num); |
| result = STORAGE_ERR_GENERIC; |
| goto err_write; |
| } |
| |
| memcpy(block_data + block_offset, data, len); |
| file_block_put_dirty(&session->tr, file, block_num, block_data, |
| &block_data_ref); |
| |
| #if TLOG_LVL >= TLOG_LVL_DEBUG |
| SS_INFO("%s: data %p offset 0x%" PRIx64 " len 0x%zx\n", __func__, data, |
| offset, len); |
| #endif |
| offset += len; |
| data += len; |
| data_len -= len; |
| } |
| |
| if (offset > file->size) { |
| file_set_size(&session->tr, file, offset); |
| } |
| |
| if (session->tr.failed) { |
| SS_ERR("%s: transaction failed\n", __func__); |
| return STORAGE_ERR_GENERIC; |
| } |
| |
| if (flags.complete_transaction) { |
| transaction_complete_etc(&session->tr, flags.update_checkpoint); |
| if (session->tr.failed) { |
| SS_ERR("%s: transaction commit failed\n", __func__); |
| return STORAGE_ERR_GENERIC; |
| } |
| } |
| |
| return STORAGE_NO_ERROR; |
| |
| err_write: |
| if (!session->tr.failed) { |
| transaction_fail(&session->tr); |
| } |
| err_transaction_complete: |
| return result; |
| } |
| |
| struct storage_file_list_state { |
| struct file_iterate_state iter; |
| char prefix[34]; |
| size_t prefix_len; |
| uint8_t max_count; |
| uint8_t count; |
| bool (*can_record_path)(void* callback_data, size_t max_path_len); |
| void (*record_path)(void* callback_data, |
| enum storage_file_list_flag flags, |
| const char* path, |
| size_t path_len); |
| void* callback_data; |
| }; |
| |
| static bool storage_file_list_buf_full(struct storage_file_list_state* miter) { |
| if (miter->max_count && miter->count >= miter->max_count) { |
| return true; |
| } |
| |
| return !miter->can_record_path(miter->callback_data, |
| FS_PATH_MAX - miter->prefix_len); |
| } |
| |
| static void storage_file_list_add(struct storage_file_list_state* miter, |
| enum storage_file_list_flag flags, |
| const char* path) { |
| assert(!storage_file_list_buf_full(miter)); |
| size_t path_len = path ? strlen(path) : 0; |
| assert(path_len <= FS_PATH_MAX - miter->prefix_len); |
| miter->record_path(miter->callback_data, flags, path, path_len); |
| miter->count++; |
| } |
| |
| static bool storage_file_list_iter(struct file_iterate_state* iter, |
| struct transaction* tr, |
| const struct block_mac* block_mac, |
| bool added, |
| bool removed) { |
| struct storage_file_list_state* miter = |
| containerof(iter, struct storage_file_list_state, iter); |
| const struct file_info* file_info; |
| struct obj_ref ref = OBJ_REF_INITIAL_VALUE(ref); |
| |
| file_info = file_get_info(tr, block_mac, &ref); |
| if (!file_info) { |
| printf("can't read file entry at %" PRIu64 "\n", |
| block_mac_to_block(tr, block_mac)); |
| return true; |
| } |
| |
| if (strncmp(file_info->path, miter->prefix, miter->prefix_len) == 0) { |
| storage_file_list_add(miter, |
| added ? STORAGE_FILE_LIST_ADDED |
| : removed ? STORAGE_FILE_LIST_REMOVED |
| : STORAGE_FILE_LIST_COMMITTED, |
| file_info->path + miter->prefix_len); |
| } |
| |
| file_info_put(file_info, &ref); |
| |
| return storage_file_list_buf_full(miter); |
| } |
| |
| enum storage_err storage_file_list( |
| struct storage_client_session* session, |
| uint8_t max_count, |
| enum storage_file_list_flag last_state, |
| const char* last_fname, |
| size_t last_fname_len, |
| struct storage_op_flags flags, |
| bool (*can_record_path)(void* callback_data, size_t max_path_len), |
| void (*record_path)(void* callback_data, |
| enum storage_file_list_flag flags, |
| const char* path, |
| size_t path_len), |
| void* callback_data) { |
| enum file_op_result iterate_res; |
| const char* last_name; |
| char path_buf[FS_PATH_MAX]; |
| |
| struct storage_file_list_state state = { |
| .iter.file = storage_file_list_iter, |
| .max_count = max_count, |
| .record_path = record_path, |
| .can_record_path = can_record_path, |
| .callback_data = callback_data, |
| }; |
| |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| result = ensure_active_transaction(session, flags); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| result = |
| get_path(state.prefix, sizeof(state.prefix), &session->uuid, "", 0); |
| if (result != STORAGE_NO_ERROR) { |
| SS_ERR("%s: internal error, get_path failed\n", __func__); |
| return STORAGE_ERR_GENERIC; |
| } |
| state.prefix_len = strlen(state.prefix); |
| |
| if (last_state == STORAGE_FILE_LIST_END) { |
| SS_ERR("%s: invalid request state (%" PRIx8 ")\n", __func__, |
| last_state); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| if (last_state == STORAGE_FILE_LIST_START) { |
| last_name = NULL; |
| } else { |
| /* make sure filename is legal */ |
| if (!is_valid_name(last_fname, last_fname_len)) { |
| SS_ERR("%s: invalid filename\n", __func__); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| result = get_path(path_buf, sizeof(path_buf), &session->uuid, |
| last_fname, last_fname_len); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| last_name = path_buf; |
| } |
| |
| if (last_state != STORAGE_FILE_LIST_ADDED) { |
| iterate_res = file_iterate(&session->tr, last_name, false, &state.iter, |
| flags.allow_repaired); |
| last_name = NULL; |
| } else { |
| iterate_res = FILE_OP_SUCCESS; |
| } |
| if (iterate_res == FILE_OP_SUCCESS && !storage_file_list_buf_full(&state)) { |
| iterate_res = file_iterate(&session->tr, last_name, true, &state.iter, |
| flags.allow_repaired); |
| } |
| if (iterate_res != FILE_OP_SUCCESS) { |
| SS_ERR("%s: file_iterate failed\n", __func__); |
| return file_op_result_to_storage_err(iterate_res); |
| } |
| |
| if (!storage_file_list_buf_full(&state)) { |
| storage_file_list_add(&state, STORAGE_FILE_LIST_END, NULL); |
| } |
| |
| return STORAGE_NO_ERROR; |
| } |
| |
| enum storage_err storage_file_get_size(struct storage_client_session* session, |
| uint32_t handle, |
| struct storage_op_flags flags, |
| uint64_t* size) { |
| bool valid; |
| struct storage_file_handle* file; |
| |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| result = ensure_active_transaction(session, flags); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| file = get_file_handle(session, handle); |
| if (!file) { |
| SS_ERR("%s: invalid file handle (%" PRIx32 ")\n", __func__, handle); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| valid = file_get_size(&session->tr, file, size); |
| if (!valid) { |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| return STORAGE_NO_ERROR; |
| } |
| |
| enum storage_err storage_file_set_size(struct storage_client_session* session, |
| uint32_t handle, |
| uint64_t new_size, |
| struct storage_op_flags flags) { |
| struct storage_file_handle* file; |
| |
| enum storage_err result = |
| assert_checkpoint_flag_valid(session, flags, __func__); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| result = ensure_active_transaction(session, flags); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| |
| file = get_file_handle(session, handle); |
| if (!file) { |
| SS_ERR("%s: invalid file handle (%" PRIx32 ")\n", __func__, handle); |
| return STORAGE_ERR_NOT_VALID; |
| } |
| |
| SS_INFO("%s: new size 0x%" PRIx64 ", old size 0x%" PRIx64 "\n", __func__, |
| new_size, file->size); |
| |
| /* for now we only support shrinking the file */ |
| if (new_size > file->size) { |
| result = storage_create_gap(session, file); |
| if (result != STORAGE_NO_ERROR) { |
| return result; |
| } |
| storage_create_gap(session, file); |
| } |
| |
| /* check for nop */ |
| if (new_size != file->size) { |
| /* update size */ |
| file_set_size(&session->tr, file, new_size); |
| } |
| |
| /* try to commit */ |
| if (flags.complete_transaction) { |
| transaction_complete_etc(&session->tr, flags.update_checkpoint); |
| } |
| |
| if (session->tr.failed) { |
| SS_ERR("%s: transaction failed\n", __func__); |
| return STORAGE_ERR_GENERIC; |
| } |
| |
| return STORAGE_NO_ERROR; |
| } |