Atmosphere/fusee/fusee-secondary/src/device_partition.c

/*
 * Copyright (c) 2018-2019 Atmosphère-NX
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <stdio.h> 
#include <string.h>
#include "device_partition.h"

int device_partition_read_data(device_partition_t *devpart, void *dst, uint64_t sector, uint64_t num_sectors)
{
    int rc;
    if (!devpart->initialized) {
        rc = devpart->initializer(devpart);
        if (rc != 0) {
            return rc;
        }
    }
    if (devpart->read_cipher != NULL && devpart->crypto_mode != DevicePartitionCryptoMode_None) {
        for (uint64_t i = 0; i < num_sectors; i += devpart->crypto_work_buffer_num_sectors) {
            uint64_t n = (i + devpart->crypto_work_buffer_num_sectors > num_sectors) ? (num_sectors - i) : devpart->crypto_work_buffer_num_sectors;
            rc = devpart->reader(devpart, devpart->crypto_work_buffer, sector + i, n);
            if (rc != 0) {
                return rc;
            }
            rc = devpart->read_cipher(devpart, sector + i, n);
            if (rc != 0) {
                return rc;
            }
            memcpy(dst + (size_t)(devpart->sector_size * i), devpart->crypto_work_buffer, (size_t)(devpart->sector_size * n)); 
        }
        return 0;
    } else {
        return devpart->reader(devpart, dst, sector, num_sectors);
    }
}

int device_partition_write_data(device_partition_t *devpart, const void *src, uint64_t sector, uint64_t num_sectors)
{
    int rc;
    if (!devpart->initialized) {
        rc = devpart->initializer(devpart);
        if (rc != 0) {
            return rc;
        }
    }
    if (devpart->read_cipher != NULL && devpart->crypto_mode != DevicePartitionCryptoMode_None) {
        for (uint64_t i = 0; i < num_sectors; i += devpart->crypto_work_buffer_num_sectors) {
            uint64_t n = (i + devpart->crypto_work_buffer_num_sectors > num_sectors) ? (num_sectors - i) : devpart->crypto_work_buffer_num_sectors;
            memcpy(devpart->crypto_work_buffer, src + (size_t)(devpart->sector_size * i), (size_t)(devpart->sector_size * n)); 
            rc = devpart->write_cipher(devpart, sector + i, n);
            if (rc != 0) {
                return rc;
            }
            rc = devpart->writer(devpart, devpart->crypto_work_buffer, sector + i, n);
            if (rc != 0) {
                return rc;
            }
        }
        return 0;
    } else {
        return devpart->writer(devpart, src, sector, num_sectors);
    }
}

int emu_device_partition_read_data(device_partition_t *devpart, void *dst, uint64_t sector, uint64_t num_sectors, const char *origin_path)
{
    int rc = 0;
    
    /* Read partition data using our backing file. */
    FILE *origin = fopen(origin_path, "rb");
    fseek(origin, sector * devpart->sector_size, SEEK_CUR);
    rc = (fread(dst, devpart->sector_size, num_sectors, origin) > 0) ? 0 : -1;
    fclose(origin);
    
    return rc;
}

int emu_device_partition_write_data(device_partition_t *devpart, const void *src, uint64_t sector, uint64_t num_sectors, const char *origin_path)
{
    int rc = 0;
    
    /* Write partition data using our backing file. */
    FILE *origin = fopen(origin_path, "wb");
    fseek(origin, sector * devpart->sector_size, SEEK_CUR);
    rc = (fwrite(src, devpart->sector_size, num_sectors, origin) > 0) ? 0 : -1;
    fclose(origin);
    
    return rc;
}