summaryrefslogtreecommitdiffstats
path: root/src/chgtype.c
diff options
context:
space:
mode:
authorChristophe Grenier <grenier@cgsecurity.org>2007-10-29 22:38:52 +0100
committerChristophe Grenier <grenier@cgsecurity.org>2007-10-29 22:38:52 +0100
commit9928d99936105b4653d2d1b8ca74dc3ffba5c71e (patch)
tree06aa4f5e9f0055027c6fb54dd47a8414cf2fba32 /src/chgtype.c
First version in git
Diffstat (limited to 'src/chgtype.c')
-rw-r--r--src/chgtype.c156
1 files changed, 156 insertions, 0 deletions
diff --git a/src/chgtype.c b/src/chgtype.c
new file mode 100644
index 0000000..f8bba22
--- /dev/null
+++ b/src/chgtype.c
@@ -0,0 +1,156 @@
+/*
+
+ File: chgtype.c
+
+ Copyright (C) 1998-2005 Christophe GRENIER <grenier@cgsecurity.org>
+
+ This software 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; either version 2 of the License, or
+ (at your option) any later version.
+
+ 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 the Free Software Foundation, Inc., 51
+ Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+ */
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#ifdef HAVE_STDLIB_H
+#include <stdlib.h>
+#endif
+#ifdef HAVE_STRING_H
+#include <string.h>
+#endif
+#include "types.h"
+#include "common.h"
+#include "lang.h"
+#include "intrf.h"
+#include "intrfn.h"
+#include "fnctdsk.h"
+#include "chgtype.h"
+#include "log.h"
+#include "guid_cmp.h"
+
+extern const arch_fnct_t arch_gpt;
+extern const arch_fnct_t arch_none;
+
+struct part_name_struct
+{
+ unsigned int index;
+ const char *name;
+};
+
+static void change_part_type_cli(const disk_t *disk_car,partition_t *partition, char **current_cmd)
+{
+ while(*current_cmd[0]==',')
+ (*current_cmd)++;
+ {
+ int tmp_val= strtol(*current_cmd, NULL, 16);
+ while(*current_cmd[0]!=',' && *current_cmd[0]!='\0')
+ (*current_cmd)++;
+ partition->arch->set_part_type(partition,tmp_val);
+ }
+}
+
+#ifdef HAVE_NCURSES
+static void change_part_type_ncurses(const disk_t *disk_car,partition_t *partition)
+{
+ partition_t *new_partition;
+ char response[100];
+ int size=0;
+ int i;
+ unsigned int last[3], done = 0, next = 0;
+ struct part_name_struct part_name[0x100];
+ struct MenuItem menuType[]=
+ {
+ { 'P', "Previous",""},
+ { 'N', "Next","" },
+ { 'Q', "Proceed","Go set the partition type"},
+ { 0, NULL, NULL }
+ };
+ /* Create an index of all partition type except Intel extended */
+ new_partition=partition_new(NULL);
+ dup_partition_t(new_partition,partition);
+ for(i=0;i<=0xFF;i++)
+ {
+ if(partition->arch->set_part_type(new_partition,i)==0)
+ {
+ part_name[size].name=new_partition->arch->get_partition_typename(new_partition);
+ if(part_name[size].name!=NULL)
+ part_name[size++].index=i;
+ }
+ }
+ free(new_partition);
+
+ /* Display the list of partition type in 3 columns */
+ aff_buffer(BUFFER_RESET,"Q");
+ aff_buffer(BUFFER_ADD,"List of partition type\n");
+ for (i = 2; i >= 0; i--)
+ last[2 - i] = done += (size + i - done) / (i + 1);
+ i = done = 0;
+ while (done < last[0])
+ {
+ aff_buffer(BUFFER_ADD, "%02x %-20s%c", part_name[next].index, part_name[next].name,(i==2 ? '\n' : ' '));
+ next = last[i++] + done;
+ if (i > 2 || next >= last[i]) {
+ i = 0;
+ next = ++done;
+ }
+ }
+
+ /* Ask for the new partition type*/
+ aff_copy(stdscr);
+ wmove(stdscr,4,0);
+ aff_part(stdscr,AFF_PART_ORDER,disk_car,partition);
+ screen_buffer_display(stdscr,"",menuType);
+ wmove(stdscr,23,0);
+ wdoprintf(stdscr,"New partition type [current %02x] ? ",partition->arch->get_part_type(partition));
+ if (get_string(response, sizeof(response), NULL) > 0) {
+ int tmp_val = strtol(response, NULL, 16);
+ partition->arch->set_part_type(partition,tmp_val);
+ }
+}
+#endif
+
+void change_part_type(const disk_t *disk_car,partition_t *partition, char **current_cmd)
+{
+ const arch_fnct_t *arch=partition->arch;
+ if(partition->arch==NULL)
+ {
+ log_error("change_part_type arch==NULL\n");
+ return;
+ }
+ if(partition->arch==&arch_gpt)
+ partition->arch=&arch_none;
+ if(partition->arch->set_part_type==NULL)
+ {
+ log_error("change_part_type set_part_type==NULL\n");
+ partition->arch=arch;
+ return;
+ }
+ if(*current_cmd!=NULL)
+ change_part_type_cli(disk_car, partition, current_cmd);
+ else
+ {
+#ifdef HAVE_NCURSES
+ change_part_type_ncurses(disk_car, partition);
+#endif
+ }
+ if(arch==&arch_gpt)
+ {
+ partition->arch=arch;
+ /* TODO: use upart_type to set part_type_gpt */
+ if(guid_cmp(partition->part_type_gpt, GPT_ENT_TYPE_UNUSED)==0)
+ partition->part_type_gpt=GPT_ENT_TYPE_MS_BASIC_DATA;
+ }
+ log_info("Change partition type:\n");
+ log_partition(disk_car,partition);
+}