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