From f02fecd19a658c54eb2ef2d5af35c0b46380461b Mon Sep 17 00:00:00 2001
From: Karel Zak
Date: Tue, 25 Jun 2013 15:55:02 +0200
Subject: fdisk: (dos) add dos menu callback

Signed-off-by: Karel Zak <kzak@redhat.com>
---
 fdisks/fdisk-menu.c | 67 ++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 66 insertions(+), 1 deletion(-)

(limited to 'fdisks/fdisk-menu.c')

diff --git a/fdisks/fdisk-menu.c b/fdisks/fdisk-menu.c
index e854f0bc7..ae6dda8bd 100644
--- a/fdisks/fdisk-menu.c
+++ b/fdisks/fdisk-menu.c
@@ -8,6 +8,10 @@
 #include "c.h"
 #include "fdisk.h"
 #include "pt-sun.h"
+#include "pt-mbr.h"
+
+#include "fdiskdoslabel.h"
+#include "fdiskbsdlabel.h"
 
 struct menu_entry {
 	const char	key;
@@ -47,6 +51,7 @@ struct menu_context {
 DECLARE_MENU_CB(gpt_menu_cb);
 DECLARE_MENU_CB(sun_menu_cb);
 DECLARE_MENU_CB(geo_menu_cb);
+DECLARE_MENU_CB(dos_menu_cb);
 
 /*
  * Menu entry macros:
@@ -172,7 +177,7 @@ struct menu menu_sgi = {
 };
 
 struct menu menu_dos = {
-/*	.callback = dos_menu_cb, */
+	.callback = dos_menu_cb,
 	.label = FDISK_DISKLABEL_DOS,
 	.entries = {
 		MENU_BSEP(N_("DOS (MBR)")),
@@ -409,6 +414,66 @@ static int gpt_menu_cb(struct fdisk_context *cxt,
 	return rc;
 }
 
+
+/*
+ * This is fdisk frontend for MBR specific libfdisk functions that
+ * are not expported by generic libfdisk API.
+ */
+static int dos_menu_cb(struct fdisk_context *cxt,
+		       const struct menu *menu __attribute__((__unused__)),
+		       const struct menu_entry *ent)
+{
+	int rc = 0;
+
+	if (!ent->expert) {
+		switch (ent->key) {
+		case 'a':
+		{
+			size_t n;
+			rc = fdisk_ask_partnum(cxt, &n, FALSE);
+			if (!rc)
+				rc = fdisk_partition_toggle_flag(cxt, n, DOS_FLAG_ACTIVE);
+			break;
+		}
+		case 'b':
+		{
+			struct fdisk_context *bsd
+					= fdisk_new_nested_context(cxt, "bsd");
+			if (bsd)
+				bsd_command_prompt(bsd);
+			fdisk_free_context(bsd);
+			break;
+		}
+		case 'c':
+			toggle_dos_compatibility_flag(cxt);
+			break;
+		}
+		return rc;
+	}
+
+	/* expert mode */
+	switch (ent->key) {
+	case 'b':
+	{
+		size_t n;
+		rc = fdisk_ask_partnum(cxt, &n, FALSE);
+		if (!rc)
+			dos_move_begin(cxt, n);
+		break;
+	}
+	case 'e':
+		rc = fdisk_dos_list_extended(cxt);
+		break;
+	case 'f':
+		dos_fix_partition_table_order(cxt);
+		break;
+	case 'i':
+		rc = fdisk_set_disklabel_id(cxt);
+		break;
+	}
+	return rc;
+}
+
 static int sun_menu_cb(struct fdisk_context *cxt,
 		       const struct menu *menu __attribute__((__unused__)),
 		       const struct menu_entry *ent)
-- 
cgit v1.2.3-55-g7522