svn_rev_643

structure cleanup

1 files changed, 486 insertions, 0 deletions

diff --git a/commands/edit.c b/commands/edit.c
new file mode 100644
index 0000000000..902bce28fe
--- /dev/null
+++ b/commands/edit.c
@@ -0,0 +1,486 @@
+#include <common.h>
+#include <command.h>
+#include <malloc.h>
+#include <fs.h>
+#include <linux/ctype.h>
+#include <fcntl.h>
+#include <readkey.h>
+#include <errno.h>
+#include <xfuncs.h>
+#include <linux/stat.h>
+
+#define TABSPACE 8
+
+struct line {
+	int length;
+	struct line *next;
+	struct line *prev;
+	char *data;
+};
+
+static struct line *buffer;
+
+static struct line *lastscrline;
+
+static int screenwidth  = 80;
+static int screenheight = 25;
+
+static int cursx  = 0;		/* position on screen */
+static int cursy  = 0;
+
+static int textx  = 0;		/* position in text */
+
+static struct line *curline;	/* line where the cursor is */
+
+static struct line *scrline;	/* the first line on screen */
+int scrcol = 0;			/* the first column on screen */
+
+static void pos(int x, int y)
+{
+	printf("%c[%d;%dH", 27, y + 1, x + 1);
+}
+
+static char *screenline(char *line, int *pos)
+{
+	int i, outpos = 0;
+	static char lbuf[1024];
+
+	memset(lbuf, 0, 1024);
+
+	if (!line) {
+		lbuf[0] = '~';
+		return lbuf;
+	}
+
+	for (i = 0; outpos < 1024; i++) {
+		if (i == textx && pos)
+			*pos = outpos;
+		if (!line[i])
+			break;
+		if (line[i] == '\t') {
+			lbuf[outpos++] = ' ';
+			while (outpos % TABSPACE)
+				lbuf[outpos++] = ' ';
+			continue;
+		}
+		lbuf[outpos++] = line[i];
+	}
+
+	return lbuf;
+}
+
+static int setpos(char *line, int position)
+{
+	int i = 0;
+	int linepos = 0;
+
+	while(line[linepos]) {
+		if (line[linepos] == '\t')
+			while ((i + 1) % TABSPACE)
+				i++;
+		if (i >= position)
+			return linepos;
+		linepos++;
+		i++;
+	}
+
+	return linepos;
+}
+
+static void refresh_line(struct line *line, int ypos)
+{
+	char *str = screenline(line->data, NULL) + scrcol;
+	pos(0, ypos);
+	str[screenwidth] = 0;
+	printf("%s%c[K", str, 27);
+	pos(cursx, cursy);
+}
+
+static void refresh(int full)
+{
+	int i;
+	struct line *l = scrline;
+
+	if (!full) {
+		if (scrline->next == lastscrline) {
+			printf("%c[1T", 27);
+			refresh_line(scrline, 0);
+			pos(0, screenheight);
+			printf("%*s", screenwidth, "");
+			return;
+		}
+
+		if (scrline->prev == lastscrline) {
+			printf("%c[1S", 27);
+			for (i = 0; i < screenheight - 1; i++) {
+				l = l->next;
+				if (!l)
+					return;
+			}
+			refresh_line(l, screenheight - 1);
+			return;
+		}
+	}
+
+	for (i = 0; i < screenheight; i++) {
+		refresh_line(l, i);
+		l = l->next;
+		if (!l)
+			break;
+	}
+
+	i++;
+	while (i < screenheight) {
+		pos(0, i++);
+		printf("~");
+	}
+}
+
+static void line_free(struct line *line)
+{
+	free(line->data);
+	free(line);
+}
+
+static struct line *line_realloc(int len, struct line *line)
+{
+	int size = 32;
+
+	if (!line) {
+		line = xzalloc(sizeof(struct line));
+		line->data = malloc(32);
+	}
+
+	while (size < len)
+		size <<= 1;
+
+	line->data = realloc(line->data, size);
+	return line;
+}
+
+static int edit_read_file(const char *path)
+{
+	struct line *line;
+	struct line *lastline = NULL;
+	char *filebuffer;
+	char *linestr;
+	struct stat s;
+
+	if (!stat(path, &s)) {
+		filebuffer = read_file(path);
+		if (!filebuffer) {
+			printf("could not read %s: %s\n", path, errno_str());
+			return -1;
+		}
+
+		linestr = strtok(filebuffer, "\r\n");
+		while (linestr) {
+			line = line_realloc(strlen(linestr) + 1, NULL);
+			if (!buffer)
+				buffer = line;
+			memcpy(line->data, linestr, strlen(linestr) + 1);
+			line->prev = lastline;
+			if (lastline)
+				lastline->next = line;
+			line->next = 0;
+			lastline = line;
+			linestr = strtok(NULL, "\r\n");
+		}
+	}
+
+	if (!buffer) {
+		buffer = line_realloc(0, NULL);
+		buffer->data[0] = 0;
+	}
+
+	return 0;
+}
+
+static int save_file(const char *path)
+{
+	struct line *line, *tmp;
+	int fd;
+
+	fd = open(path, O_WRONLY | O_TRUNC | O_CREAT);
+	if (fd < 0) {
+		printf("could not open file for writing: %s\n", errno_str());
+		return -1;
+	}
+
+	line = buffer;
+
+	while(line) {
+		tmp = line->next;
+		write(fd, line->data, strlen(line->data));
+		write(fd, "\n", 1);
+		line_free(line);
+		line = tmp;
+	}
+	close(fd);
+	return 0;
+}
+
+static void insert_char(char c)
+{
+	int pos = textx;
+	char *line = curline->data;
+	int end = strlen(line);
+
+	line_realloc(strlen(curline->data) + 2, curline);
+
+	while (end >= pos) {
+		line[end + 1] = line[end];
+		end--;
+	}
+	line[pos] = c;
+	textx++;
+	refresh_line(curline, cursy);
+}
+
+static void delete_char(int pos)
+{
+	char *line = curline->data;
+	int end = strlen(line);
+
+	while (pos < end) {
+		line[pos] = line[pos + 1];
+		pos++;
+	}
+	refresh_line(curline, cursy);
+}
+
+static void split_line(void)
+{
+	int length = strlen(curline->data + textx);
+	struct line *newline = line_realloc(length + 1, NULL);
+	struct line *tmp;
+
+	memcpy(newline->data, curline->data + textx, length + 1);
+
+	curline->data[textx] = 0;
+
+	tmp = curline->next;
+	curline->next = newline;
+	newline->prev = curline;
+	newline->next = tmp;
+	if (tmp)
+		tmp->prev = newline;
+
+	textx = 0;
+	cursy++;
+	curline = curline->next;
+	refresh(1);
+}
+
+static void merge_line(struct line *line)
+{
+	struct line *tmp;
+
+	line_realloc(strlen(line->data) + strlen(line->next->data) + 1, line);
+
+	tmp = line->next;
+
+	line->next = line->next->next;
+	if (line->next)
+		line->next->prev = line;
+	strcat(line->data, tmp->data);
+
+	line_free(tmp);
+
+	refresh(1);
+}
+
+/* not a good idea on slow serial lines */
+/* #define GETWINSIZE */
+
+#ifdef GETWINSIZE
+static void getwinsize(void) {
+	int y, yy = 25, xx = 80, i, n, r;
+	char buf[100];
+	char *endp;
+
+	for (y = 25; y < 320; y++) {
+		pos(y, y);
+		printf("%c[6n", 27);
+		i = 0;
+		while ((r = getc()) != 'R') {
+			buf[i] = r;
+			i++;
+		}
+		n = simple_strtoul(buf + 2, &endp, 10);
+		if (n == y + 1)
+			yy = y + 1;
+		n = simple_strtoul(endp + 1, NULL, 10);
+		if (n == y + 1)
+			xx = y + 1;
+	}
+	pos(0,0);
+	screenheight = yy;
+	screenwidth = xx;
+	printf("%d %d\n", xx, yy);
+	mdelay(1000);
+}
+#endif
+
+int do_edit(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
+{
+	int lastscrcol;
+	int i;
+	int linepos;
+	char c;
+
+	buffer = NULL;
+	if(edit_read_file(argv[1]))
+		return 1;
+
+#ifdef GETWINSIZE
+	getwinsize();
+#endif
+
+	cursx  = 0;
+	cursy  = 0;
+	textx  = 0;
+	scrcol = 0;
+	curline = buffer;
+	scrline = curline;
+	lastscrline = scrline;
+	lastscrcol = 0;
+
+	printf("%c[2J", 27);
+	refresh(1);
+
+	while (1) {
+		int curlen = strlen(curline->data);
+
+		if (textx > curlen)
+			textx = curlen;
+		if (textx < 0)
+			textx = 0;
+
+		screenline(curline->data, &linepos);
+
+		if (linepos > scrcol + screenwidth)
+			scrcol = linepos - screenwidth;
+
+		if (scrcol > linepos)
+			scrcol = linepos;
+
+		cursx = linepos - scrcol;
+
+		while (cursy >= screenheight) {
+			cursy--;
+			scrline = scrline->next;
+		}
+
+		while (cursy < 0) {
+			cursy++;
+			scrline = scrline->prev;
+		}
+
+		if (scrline != lastscrline || scrcol != lastscrcol)
+			refresh(0);
+
+		lastscrcol  = scrcol;
+		lastscrline = scrline;
+		pos(cursx, cursy);
+
+		c = read_key();
+		switch (c) {
+		case KEY_UP:
+			if (!curline->prev)
+				continue;
+
+			curline = curline->prev;
+			cursy--;
+			textx = setpos(curline->data, linepos);
+			break;
+		case KEY_DOWN:
+			if (!curline->next)
+				continue;
+
+			curline = curline->next;
+			cursy++;
+			textx = setpos(curline->data, linepos);
+			break;
+		case KEY_RIGHT:
+			textx++;
+			break;
+		case KEY_LEFT:
+			textx--;
+			break;
+		case KEY_HOME:
+			textx = 0;
+			break;
+		case KEY_END:
+			textx = curlen;
+			break;
+		case KEY_PAGEUP:
+			for (i = 0; i < screenheight - 1; i++) {
+				if (!curline->prev)
+					break;
+				cursy--;
+				curline = curline->prev;
+			}
+			textx = setpos(curline->data, linepos);
+			break;
+		case KEY_PAGEDOWN:
+			for (i = 0; i < screenheight - 1; i++) {
+				if (!curline->next)
+					break;
+				cursy++;
+				curline = curline->next;
+			}
+			textx = setpos(curline->data, linepos);
+			break;
+		case KEY_DEL:
+			if (textx == curlen) {
+				if (curline->next)
+					merge_line(curline);
+			} else
+				delete_char(textx);
+			break;
+		case 13:
+		case 10:
+			split_line();
+			break;
+		case 127:
+			if (textx > 0) {
+				textx--;
+				delete_char(textx);
+			} else {
+				if (!curline->prev)
+					break;
+				curline = curline->prev;
+				cursy--;
+				textx = strlen(curline->data);
+				merge_line(curline);
+			}
+			break;
+		case 4:
+			save_file(argv[1]);
+			goto out;
+		case 3:
+			goto out;
+		default:
+			if ((signed char)c != -1)
+				insert_char(c);
+		}
+	}
+out:
+	printf("%c[2J", 27);
+	return 0;
+}
+
+static __maybe_unused char cmd_edit_help[] =
+"Usage: edit <file>\n"
+"This is a very small editor. Its only features are moving the cursor with\n"
+"the usual keys and typing characters.\n"
+"<ctrl-c> quits the editor without saving,\n"
+"<ctrl-d> quits the editor with saving the current file.\n";
+
+U_BOOT_CMD_START(edit)
+	.maxargs	= 2,
+	.cmd		= do_edit,
+	.usage		= "edit a file",
+	U_BOOT_CMD_HELP(cmd_edit_help)
+U_BOOT_CMD_END