Update twin ui to use discover server

Signed-off-by: Geoff Levand <geoff@infradead.org>

1 files changed, 494 insertions, 0 deletions

diff --git a/ui/twin/main-ps3.c b/ui/twin/main-ps3.c
new file mode 100644
index 0000000..db0539a
--- /dev/null
+++ b/ui/twin/main-ps3.c
@@ -0,0 +1,494 @@
+/*
+ * Petitboot twin bootloader for the PS3 game console
+ *
+ *  Copyright Geoff Levand <geoff@infradead.org>
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; version 2 of the License.
+ *
+ *  This program is distributed in the hope that 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, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#if defined(HAVE_CONFIG_H)
+#include "config.h"
+#endif
+
+#define _GNU_SOURCE
+#include <assert.h>
+#include <errno.h>
+#include <getopt.h>
+#include <signal.h>
+#include <stdlib.h>
+#include <string.h>
+#include <linux/input.h>
+#include <sys/time.h>
+
+#include "log/log.h"
+#include "talloc/talloc.h"
+#include "waiter/waiter.h"
+#include "ui/common/timer.h"
+#include "ui/common/ps3.h"
+
+#include "pbt-client.h"
+#include "pbt-main.h"
+
+/* control to keyboard mappings for the sixaxis controller */
+
+static const uint8_t ps3_sixaxis_map[] = {
+	0,		/*   0  Select		*/
+	0,		/*   1  L3              */
+	0,		/*   2  R3              */
+	0,		/*   3  Start           */
+	KEY_UP,		/*   4  Dpad Up         */
+	KEY_RIGHT,	/*   5  Dpad Right      */
+	KEY_DOWN,	/*   6  Dpad Down       */
+	KEY_LEFT,	/*   7  Dpad Left       */
+	0,		/*   8  L2              */
+	0,		/*   9  R2              */
+	0,		/*  10  L1              */
+	0,		/*  11  R1              */
+	0,		/*  12  Triangle        */
+	KEY_ENTER,	/*  13  Circle          */
+	0,		/*  14  Cross           */
+	KEY_DELETE,	/*  15  Square          */
+	0,		/*  16  PS Button       */
+	0,		/*  17  nothing      */
+	0,		/*  18  nothing      */
+};
+
+static struct pbt_item *ps3_setup_system_item(struct pbt_menu *menu,
+	const struct pbt_menu_layout *layout)
+{
+#if 0
+	struct _twin_rect r;
+	struct pbt_item *main_item;
+	struct pbt_item *sub_item;
+	twin_pixmap_t *icon;
+
+	icon = pbt_icon_load(NULL);
+
+	/* Main item */
+
+	pbt_view_get_item_rect(menu->main, layout, icon, 0, &r);
+	main_item = pbt_item_create(menu->main, menu->main, &r);
+
+	if (!main_item)
+		goto fail;
+
+	main_item->image = icon;
+
+	list_add_tail(menu->main->items_list, &main_item->list);
+
+	/* Sub items */
+
+	main_item->sub_items_list = talloc(main_item, struct list);
+	list_init(main_item->sub_items_list);
+
+	pbt_view_get_item_rect(menu->sub, layout, icon, 0, &r);
+	sub_item = pbt_item_create(main_item, menu->sub, &r);
+
+	if (!sub_item)
+		goto fail;
+
+	sub_item->image = pbt_item_create_text_image(layout, pbt_rect_width(&r),
+		pbt_rect_height(&r), icon, "Boot GameOS",
+		"Reboot the PS3 into the GameOS");
+
+	list_add_tail(main_item->sub_items_list, &sub_item->list);
+
+	pbt_view_get_item_rect(menu->sub, layout, icon, 1, &r);
+	sub_item = pbt_item_create(main_item, menu->sub, &r);
+
+	if (!sub_item)
+		goto fail;
+
+	sub_item->image = pbt_item_create_text_image(layout, pbt_rect_width(&r),
+		pbt_rect_height(&r), icon, "Set Video Mode",
+		"Set the current video mode");
+
+	list_add_tail(main_item->sub_items_list, &sub_item->list);
+
+	pbt_view_get_item_rect(menu->sub, layout, icon, 2, &r);
+	sub_item = pbt_item_create(main_item, menu->sub, &r);
+
+	if (!sub_item)
+		goto fail;
+
+	sub_item->image = pbt_item_create_text_image(layout, pbt_rect_width(&r),
+		pbt_rect_height(&r), icon, "Exit to Shell",
+		"Exit to a system shell prompt");
+
+	list_add_tail(main_item->sub_items_list, &sub_item->list);
+
+	menu->sub->selected = sub_item;
+	menu->sub->items_list = main_item->sub_items_list;
+
+	return main_item;
+fail:
+#endif
+	// FIXME: need cleanup
+	assert(0);
+	return NULL;
+}
+
+static int ps3_setup_test_item(struct pbt_menu *menu,
+	const struct pbt_menu_layout *layout)
+{
+#if 0
+	struct _twin_rect r;
+	struct pbt_item *main_item;
+	struct pbt_item *sub_item;
+	twin_pixmap_t *icon;
+
+	icon = pbt_icon_load(PB_ARTWORK_PATH "/hdd.png");
+
+	/* Main item */
+
+	pbt_view_get_item_rect(menu->main, layout, icon, 1, &r);
+	main_item = pbt_item_create(menu->main, menu->main, &r);
+
+	if (!main_item)
+		goto fail;
+
+	main_item->image = icon;
+
+	list_add_tail(menu->main->items_list, &main_item->list);
+
+	/* Sub items */
+
+	main_item->sub_items_list = talloc(main_item, struct list);
+	list_init(main_item->sub_items_list);
+
+	pbt_view_get_item_rect(menu->sub, layout, icon, 0, &r);
+	sub_item = pbt_item_create(main_item, menu->sub, &r);
+
+	if (!sub_item)
+		goto fail;
+
+	sub_item->active = 0;
+	sub_item->image = pbt_item_create_text_image(layout, pbt_rect_width(&r),
+		pbt_rect_height(&r), icon, "test 1",
+		"text for test 1");
+
+	list_add_tail(main_item->sub_items_list, &sub_item->list);
+
+	pbt_view_get_item_rect(menu->sub, layout, icon, 1, &r);
+	sub_item = pbt_item_create(main_item, menu->sub, &r);
+
+	if (!sub_item)
+		goto fail;
+
+	sub_item->active = 0;
+	sub_item->image = pbt_item_create_text_image(layout, pbt_rect_width(&r),
+		pbt_rect_height(&r), icon, "test 2",
+		"text for test 2");
+
+	list_add_tail(main_item->sub_items_list, &sub_item->list);
+
+	menu->sub->selected = sub_item;
+	menu->sub->items_list = main_item->sub_items_list;
+
+	return 0;
+
+fail:
+#endif
+	// FIXME: need cleanup
+	assert(0);
+	return -1;
+}
+
+static struct pbt_menu *ps3_menu_create(void *ctx, struct pbt_scr *scr)
+{
+#if 0
+	struct pbt_menu *menu;
+	struct _twin_rect r;
+	twin_pixmap_t *icon;
+	const struct pbt_border *border;
+	static const struct pbt_menu_layout layout = {
+		.item_space = 10,
+		.icon_space = 6,
+		.title_font_size = 30,
+		.title_font_color = 0xff000000,
+		.text_font_size = 18,
+		.text_font_color = 0xff400000,
+	};
+
+	assert(scr && scr->sig == pbt_scr_sig);
+
+	menu = talloc_zero(ctx, struct pbt_menu);
+
+	if (!menu)
+		return NULL;
+
+	menu->sig = pbt_menu_sig;
+	menu->scr = scr;
+
+	icon = pbt_icon_load(NULL);
+
+	if (!icon)
+		return NULL;
+
+	/* Setup main view */
+
+	border = &pbt_right_border;
+	r.left = 0;
+	r.top = 0;
+	r.right = icon->width + 2 * layout.item_space + 2 * layout.icon_space
+		+ border->left + border->right;
+	r.bottom = menu->scr->tscreen->height;
+
+	menu->main = pbt_view_create(menu, &r);
+
+	if (!menu->main)
+		goto fail_main;
+
+	menu->main->background_color = 0x80000000;
+	menu->main->border = *border;
+	menu->main->items_list = talloc(menu->main, struct list);
+	list_init(menu->main->items_list);
+
+	/* Setup sub view */
+
+	r.left = r.right;
+	r.top = 0;
+	r.right = menu->scr->tscreen->width;
+	r.bottom = menu->scr->tscreen->height;
+
+	menu->sub = pbt_view_create(menu, &r);
+
+	if (!menu->sub)
+		goto fail_sub;
+
+	menu->sub->background_color = 0x40000000;
+
+	/* Setup system item */
+
+	menu->main->selected = ps3_setup_system_item(menu, &layout);
+
+	if (!menu->main->selected)
+		goto fail_main_item;
+
+	//ps3_setup_test_item(menu, &layout);
+
+	return menu;
+
+fail_main_item:
+	// FIXME: need cleanup
+fail_sub:
+	// FIXME: need cleanup
+fail_main:
+	talloc_free(menu);
+#endif
+	assert(0);
+	return NULL;
+}
+
+/**
+ * struct ps3_gui - Main gui program instance.
+ */
+
+
+struct ps3_gui {
+	struct ui_timer timer;
+	struct ps3_flash_values values;
+	int dirty_values;
+
+	struct pbt_scr scr;
+	struct pbt_frame frame;
+	struct pbt_menu *menu;
+};
+
+static struct ps3_gui ps3;
+
+static struct pbt_scr *ps3_scr_from_tscreen(twin_screen_t *tscreen)
+{
+	assert(ps3.scr.tscreen == tscreen);
+	return &ps3.scr;
+}
+
+static twin_bool_t ps3_scr_event_cb(twin_screen_t *tscreen, twin_event_t *event)
+{
+	struct pbt_scr *scr = ps3_scr_from_tscreen(tscreen);
+
+	pbt_dump_event("scr", NULL, event);
+
+	switch(event->kind) {
+	case TwinEventJoyButton:
+		/* convert joystick events to key events */
+		// FIXME: need to test
+		if (event->u.js.control < sizeof(ps3_sixaxis_map)) {
+			event->u.key.key = ps3_sixaxis_map[event->u.js.control];
+			event->kind = event->u.js.value ? TwinEventKeyUp
+				: TwinEventKeyDown;
+		}
+		break;
+	default:
+		break;
+	}
+	return TWIN_FALSE;
+}
+
+static void sig_handler(int signum)
+{
+	DBGS("%d\n", signum);
+
+	switch (signum) {
+	case SIGALRM:
+		ui_timer_sigalrm(&ps3.timer);
+		break;
+	case SIGWINCH:
+//		if (ps3.gui)
+//			gui_resize(ps3.gui);
+		break;
+	default:
+		assert(0 && "unknown sig");
+		/* fall through */
+	case SIGINT:
+	case SIGHUP:
+	case SIGTERM:
+		exit(EXIT_FAILURE);
+//		if (ps3.gui)
+//			gui_abort(ps3.gui);
+		break;
+	}
+}
+
+/**
+ * main - twin bootloader main routine.
+ */
+
+int main(int argc, char *argv[])
+{
+	static struct sigaction sa;
+	static struct pbt_opts opts;
+	int result;
+	int ui_result = -1;
+	unsigned int mode;
+	FILE *log;
+
+	result = pbt_opts_parse(&opts, argc, argv);
+
+	if (result) {
+		pbt_print_usage();
+		return EXIT_FAILURE;
+	}
+
+	if (opts.show_help == pbt_opt_yes) {
+		pbt_print_usage();
+		return EXIT_SUCCESS;
+	}
+
+	if (opts.show_version == pbt_opt_yes) {
+		pbt_print_version();
+		return EXIT_SUCCESS;
+	}
+
+	log = fopen(opts.log_file, "a");
+	assert(log);
+	pb_log_set_stream(log);
+
+#if defined(DEBUG)
+	pb_log_always_flush(1);
+#endif
+
+	pb_log("--- pb-twin ---\n");
+
+	sa.sa_handler = sig_handler;
+	result = sigaction(SIGALRM, &sa, NULL);
+	result += sigaction(SIGHUP, &sa, NULL);
+	result += sigaction(SIGINT, &sa, NULL);
+	result += sigaction(SIGTERM, &sa, NULL);
+	result += sigaction(SIGWINCH, &sa, NULL);
+
+	if (result) {
+		pb_log("%s sigaction failed.\n", __func__);
+		return EXIT_FAILURE;
+	}
+
+	ps3.values = ps3_flash_defaults;
+
+	if (opts.reset_defaults != pbt_opt_yes)
+		ps3.dirty_values = ps3_flash_get_values(&ps3.values);
+
+#if 0
+	twin_feature_init(); // need it???
+
+	/* Setup screen. */
+
+#if defined(HAVE_LIBTWIN_TWIN_X11_H)
+# if defined(USE_X11)
+	if (1) {
+# else
+	if (0) {
+# endif
+		//ps3.scr.x11 = twin_x11_create(XOpenDisplay(0), 1024, 768);
+		ps3.scr.x11 = twin_x11_create(XOpenDisplay(0), 512, 384);
+
+		if (!ps3.scr.x11) {
+			perror("failed to create x11 screen !\n");
+			return EXIT_FAILURE;
+		}
+
+		ps3.scr.tscreen = ps3.scr.x11->screen;
+	} else {
+#else
+	if (1) {
+#endif
+		result = ps3_get_video_mode(&mode);
+
+		/* Current becomes default if ps3_flash_get_values() failed. */
+
+		if (ps3.dirty_values && !result)
+			ps3.values.video_mode = mode;
+
+		/* Set mode if not at default. */
+
+		if (!result && (ps3.values.video_mode != (uint16_t)mode))
+			ps3_set_video_mode(ps3.values.video_mode);
+
+		ps3.scr.fbdev = twin_fbdev_create(-1, SIGUSR1);
+
+		if (!ps3.scr.fbdev) {
+			perror("failed to create fbdev screen !\n");
+			return EXIT_FAILURE;
+		}
+
+		ps3.scr.tscreen = ps3.scr.fbdev->screen;
+	}
+
+	ps3.scr.tscreen->event_filter = pbt_scr_event;
+
+	twin_screen_set_background(ps3.scr.tscreen,
+		pbt_background_load(ps3.scr.tscreen, NULL));
+
+	/* setup menu */
+
+	ps3.menu = ps3_menu_create(NULL, &ps3.scr);
+
+	if (!ps3.menu)
+		return EXIT_FAILURE;
+
+	/* Console switch */
+
+	if (ps3.scr.fbdev)
+		twin_fbdev_activate(ps3.scr.fbdev);
+
+	/* run twin */
+
+	// need to hookup pb waiters to twin...
+	twin_dispatch();
+
+#endif
+
+	pb_log("--- end ---\n");
+
+	return ui_result ? EXIT_FAILURE : EXIT_SUCCESS;
+}