1/*- 2 * Copyright (c) 2010-2012 Semihalf. 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 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 */ 26 27#include <sys/cdefs.h> 28__FBSDID("$FreeBSD: releng/11.0/usr.sbin/nandtool/nand_write.c 235537 2012-05-17 10:11:18Z gber $"); 29 30#include <stdio.h> 31#include <stdlib.h> 32#include <unistd.h> 33#include <fcntl.h> 34#include <libgeom.h> 35#include <sys/disk.h> 36#include <dev/nand/nand_dev.h> 37#include "nandtool.h" 38 39int nand_write(struct cmd_param *params) 40{ 41 struct chip_param_io chip_params; 42 char *dev, *file; 43 int in_fd = -1, ret = 0, done = 0; 44 int fd, block_size, mult, pos, count; 45 uint8_t *buf = NULL; 46 47 if (!(dev = param_get_string(params, "dev"))) { 48 fprintf(stderr, "Please supply 'dev' argument.\n"); 49 return (1); 50 } 51 52 if (!(file = param_get_string(params, "in"))) { 53 fprintf(stderr, "Please supply 'in' argument.\n"); 54 return (1); 55 } 56 57 if ((fd = g_open(dev, 1)) == -1) { 58 perrorf("Cannot open %s", dev); 59 return (1); 60 } 61 62 if ((in_fd = open(file, O_RDONLY)) == -1) { 63 perrorf("Cannot open file %s", file); 64 ret = 1; 65 goto out; 66 } 67 68 if (ioctl(fd, NAND_IO_GET_CHIP_PARAM, &chip_params) == -1) { 69 perrorf("Cannot ioctl(NAND_IO_GET_CHIP_PARAM)"); 70 ret = 1; 71 goto out; 72 } 73 74 block_size = chip_params.page_size * chip_params.pages_per_block; 75 76 if (param_has_value(params, "page")) { 77 pos = chip_params.page_size * param_get_int(params, "page"); 78 mult = chip_params.page_size; 79 } else if (param_has_value(params, "block")) { 80 pos = block_size * param_get_int(params, "block"); 81 mult = block_size; 82 } else if (param_has_value(params, "pos")) { 83 pos = param_get_int(params, "pos"); 84 mult = 1; 85 if (pos % chip_params.page_size) { 86 fprintf(stderr, "Position must be page-size " 87 "aligned!\n"); 88 ret = 1; 89 goto out; 90 } 91 } else { 92 fprintf(stderr, "You must specify one of: 'block', 'page'," 93 "'pos' arguments\n"); 94 ret = 1; 95 goto out; 96 } 97 98 if (!(param_has_value(params, "count"))) 99 count = mult; 100 else 101 count = param_get_int(params, "count") * mult; 102 103 if (!(buf = malloc(chip_params.page_size))) { 104 perrorf("Cannot allocate buffer [size %x]", 105 chip_params.page_size); 106 ret = 1; 107 goto out; 108 } 109 110 lseek(fd, pos, SEEK_SET); 111 112 while (done < count) { 113 if ((ret = read(in_fd, buf, chip_params.page_size)) != 114 (int32_t)chip_params.page_size) { 115 if (ret > 0) { 116 /* End of file ahead, truncate here */ 117 break; 118 } else { 119 perrorf("Cannot read from %s", file); 120 ret = 1; 121 goto out; 122 } 123 } 124 125 if ((ret = write(fd, buf, chip_params.page_size)) != 126 (int32_t)chip_params.page_size) { 127 ret = 1; 128 goto out; 129 } 130 131 done += ret; 132 } 133 134out: 135 g_close(fd); 136 if (in_fd != -1) 137 close(in_fd); 138 if (buf) 139 free(buf); 140 141 return (ret); 142} 143 144