1235537Sgber/*- 2330449Seadler * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 3330449Seadler * 4235537Sgber * Copyright (c) 2010-2012 Semihalf. 5235537Sgber * All rights reserved. 6235537Sgber * 7235537Sgber * Redistribution and use in source and binary forms, with or without 8235537Sgber * modification, are permitted provided that the following conditions 9235537Sgber * are met: 10235537Sgber * 1. Redistributions of source code must retain the above copyright 11235537Sgber * notice, this list of conditions and the following disclaimer. 12235537Sgber * 2. Redistributions in binary form must reproduce the above copyright 13235537Sgber * notice, this list of conditions and the following disclaimer in the 14235537Sgber * documentation and/or other materials provided with the distribution. 15235537Sgber * 16235537Sgber * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17235537Sgber * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18235537Sgber * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19235537Sgber * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20235537Sgber * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21235537Sgber * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22235537Sgber * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23235537Sgber * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24235537Sgber * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25235537Sgber * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26235537Sgber * SUCH DAMAGE. 27235537Sgber */ 28235537Sgber 29235537Sgber#include <sys/cdefs.h> 30235537Sgber__FBSDID("$FreeBSD: stable/11/usr.sbin/nandtool/nand_write.c 330449 2018-03-05 07:26:05Z eadler $"); 31235537Sgber 32235537Sgber#include <stdio.h> 33235537Sgber#include <stdlib.h> 34235537Sgber#include <unistd.h> 35235537Sgber#include <fcntl.h> 36235537Sgber#include <libgeom.h> 37235537Sgber#include <sys/disk.h> 38235537Sgber#include <dev/nand/nand_dev.h> 39235537Sgber#include "nandtool.h" 40235537Sgber 41235537Sgberint nand_write(struct cmd_param *params) 42235537Sgber{ 43235537Sgber struct chip_param_io chip_params; 44235537Sgber char *dev, *file; 45235537Sgber int in_fd = -1, ret = 0, done = 0; 46235537Sgber int fd, block_size, mult, pos, count; 47235537Sgber uint8_t *buf = NULL; 48235537Sgber 49235537Sgber if (!(dev = param_get_string(params, "dev"))) { 50235537Sgber fprintf(stderr, "Please supply 'dev' argument.\n"); 51235537Sgber return (1); 52235537Sgber } 53235537Sgber 54235537Sgber if (!(file = param_get_string(params, "in"))) { 55235537Sgber fprintf(stderr, "Please supply 'in' argument.\n"); 56235537Sgber return (1); 57235537Sgber } 58235537Sgber 59235537Sgber if ((fd = g_open(dev, 1)) == -1) { 60235537Sgber perrorf("Cannot open %s", dev); 61235537Sgber return (1); 62235537Sgber } 63235537Sgber 64235537Sgber if ((in_fd = open(file, O_RDONLY)) == -1) { 65235537Sgber perrorf("Cannot open file %s", file); 66235537Sgber ret = 1; 67235537Sgber goto out; 68235537Sgber } 69235537Sgber 70235537Sgber if (ioctl(fd, NAND_IO_GET_CHIP_PARAM, &chip_params) == -1) { 71235537Sgber perrorf("Cannot ioctl(NAND_IO_GET_CHIP_PARAM)"); 72235537Sgber ret = 1; 73235537Sgber goto out; 74235537Sgber } 75235537Sgber 76235537Sgber block_size = chip_params.page_size * chip_params.pages_per_block; 77235537Sgber 78235537Sgber if (param_has_value(params, "page")) { 79235537Sgber pos = chip_params.page_size * param_get_int(params, "page"); 80235537Sgber mult = chip_params.page_size; 81235537Sgber } else if (param_has_value(params, "block")) { 82235537Sgber pos = block_size * param_get_int(params, "block"); 83235537Sgber mult = block_size; 84235537Sgber } else if (param_has_value(params, "pos")) { 85235537Sgber pos = param_get_int(params, "pos"); 86235537Sgber mult = 1; 87235537Sgber if (pos % chip_params.page_size) { 88235537Sgber fprintf(stderr, "Position must be page-size " 89235537Sgber "aligned!\n"); 90235537Sgber ret = 1; 91235537Sgber goto out; 92235537Sgber } 93235537Sgber } else { 94235537Sgber fprintf(stderr, "You must specify one of: 'block', 'page'," 95235537Sgber "'pos' arguments\n"); 96235537Sgber ret = 1; 97235537Sgber goto out; 98235537Sgber } 99235537Sgber 100235537Sgber if (!(param_has_value(params, "count"))) 101235537Sgber count = mult; 102235537Sgber else 103235537Sgber count = param_get_int(params, "count") * mult; 104235537Sgber 105235537Sgber if (!(buf = malloc(chip_params.page_size))) { 106235537Sgber perrorf("Cannot allocate buffer [size %x]", 107235537Sgber chip_params.page_size); 108235537Sgber ret = 1; 109235537Sgber goto out; 110235537Sgber } 111235537Sgber 112235537Sgber lseek(fd, pos, SEEK_SET); 113235537Sgber 114235537Sgber while (done < count) { 115235537Sgber if ((ret = read(in_fd, buf, chip_params.page_size)) != 116235537Sgber (int32_t)chip_params.page_size) { 117235537Sgber if (ret > 0) { 118235537Sgber /* End of file ahead, truncate here */ 119235537Sgber break; 120235537Sgber } else { 121235537Sgber perrorf("Cannot read from %s", file); 122235537Sgber ret = 1; 123235537Sgber goto out; 124235537Sgber } 125235537Sgber } 126235537Sgber 127235537Sgber if ((ret = write(fd, buf, chip_params.page_size)) != 128235537Sgber (int32_t)chip_params.page_size) { 129235537Sgber ret = 1; 130235537Sgber goto out; 131235537Sgber } 132235537Sgber 133235537Sgber done += ret; 134235537Sgber } 135235537Sgber 136235537Sgberout: 137235537Sgber g_close(fd); 138235537Sgber if (in_fd != -1) 139235537Sgber close(in_fd); 140235537Sgber if (buf) 141235537Sgber free(buf); 142235537Sgber 143235537Sgber return (ret); 144235537Sgber} 145235537Sgber 146