summaryrefslogtreecommitdiffstats
path: root/src/geometryn.c
diff options
context:
space:
mode:
authorChristophe Grenier <grenier@cgsecurity.org>2013-05-15 13:41:20 +0200
committerChristophe Grenier <grenier@cgsecurity.org>2013-05-15 13:41:20 +0200
commitfd9c697a03a5501e326479192dad98e2d6fdb64e (patch)
tree7c5862a7d2408640a11425922f32b66e313cc48f /src/geometryn.c
parent38084f56e0fea23249b010f0fc37c20c33871840 (diff)
Split change_geometry() into change_geometry_cli() and change_geometry_ncurses()
Diffstat (limited to 'src/geometryn.c')
-rw-r--r--src/geometryn.c188
1 files changed, 188 insertions, 0 deletions
diff --git a/src/geometryn.c b/src/geometryn.c
new file mode 100644
index 0000000..c80c6b7
--- /dev/null
+++ b/src/geometryn.c
@@ -0,0 +1,188 @@
+/*
+
+ File: geometryn.c
+
+ Copyright (C) 1998-2013 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
+
+#include <stdio.h>
+#ifdef HAVE_STDLIB_H
+#include <stdlib.h>
+#endif
+#ifdef HAVE_STRING_H
+#include <string.h>
+#endif
+#include "types.h"
+#include "common.h"
+#include "intrf.h"
+#ifdef HAVE_NCURSES
+#include "intrfn.h"
+#endif
+#include "log.h"
+#include "hdaccess.h"
+#include "autoset.h"
+#include "geometry.h"
+#include "geometryn.h"
+
+#define MAX_HEADS 255u
+
+int change_geometry_ncurses(disk_t *disk_car)
+{
+ int done = 0;
+ char def[128];
+ char response[128];
+ long int tmp_val=0;
+ int command;
+ int default_option=4;
+ int cyl_modified=0;
+ int geo_modified=0;
+ while (done==0)
+ {
+ static struct MenuItem menuGeometry[]=
+ {
+ { 'c', "Cylinders", "Change cylinder geometry" },
+ { 'h', "Heads", "Change head geometry" },
+ { 's', "Sectors", "Change sector geometry" },
+ { 'n', "Sector Size", "Change sector size (WARNING: VERY DANGEROUS!)" },
+ { 'q', "Ok", "Done with changing geometry" },
+ { 0, NULL, NULL }
+ };
+ aff_copy(stdscr);
+ wmove(stdscr,5,0);
+ wprintw(stdscr,"%s, sector size=%u\n",disk_car->description(disk_car),disk_car->sector_size);
+ wmove(stdscr,7,0);
+ wprintw(stdscr,"Because these numbers change the way that TestDisk looks for partitions");
+ wmove(stdscr,8,0);
+ wprintw(stdscr,"and calculates their sizes, it's important to have the correct disk geometry.");
+ wmove(stdscr,9,0);
+ wprintw(stdscr,"PC partitioning programs often make partitions end on cylinder boundaries.");
+ wmove(stdscr,11,0);
+ wprintw(stdscr,"A partition's CHS values are based on disk translations which make them");
+ wmove(stdscr,12,0);
+ wprintw(stdscr,"different than its physical geometry. The most common CHS head values");
+ wmove(stdscr,13,0);
+ wprintw(stdscr,"are: 255, 240 and sometimes 16.");
+ wmove(stdscr,INTER_GEOM_Y, INTER_GEOM_X);
+ wclrtoeol(stdscr);
+ wrefresh(stdscr);
+ command=wmenuSimple(stdscr,menuGeometry, default_option);
+ switch (command) {
+ case 'c':
+ case 'C':
+ {
+ sprintf(def, "%lu", disk_car->geom.cylinders);
+ mvwaddstr(stdscr,INTER_GEOM_Y, INTER_GEOM_X, "Enter the number of cylinders: ");
+ wclrtoeol(stdscr);
+ if (get_string(stdscr, response, sizeof(response), def) > 0) {
+ tmp_val = atol(response);
+ if (tmp_val > 0) {
+ disk_car->geom.cylinders = tmp_val;
+ cyl_modified=1;
+ if(geo_modified==0)
+ geo_modified=1;
+ } else
+ wprintw(stdscr,"Illegal cylinders value");
+ }
+ }
+ default_option=1;
+ break;
+ case 'h':
+ case 'H':
+ {
+ sprintf(def, "%u", disk_car->geom.heads_per_cylinder);
+ mvwaddstr(stdscr,INTER_GEOM_Y, INTER_GEOM_X, "Enter the number of heads (1-255): ");
+ wclrtoeol(stdscr);
+ if (get_string(stdscr, response, sizeof(response), def) > 0) {
+ tmp_val = atoi(response);
+ if (tmp_val > 0 && tmp_val <= MAX_HEADS) {
+ disk_car->geom.heads_per_cylinder = tmp_val;
+ if(geo_modified==0)
+ geo_modified=1;
+ if(cyl_modified==0)
+ set_cylinders_from_size_up(disk_car);
+ } else
+ wprintw(stdscr,"Illegal heads value");
+ }
+ }
+ default_option=2;
+ break;
+ case 's':
+ case 'S':
+ {
+ sprintf(def, "%u", disk_car->geom.sectors_per_head);
+ /* FIXME SUN partition can have more than 63 sectors */
+ mvwaddstr(stdscr,INTER_GEOM_Y, INTER_GEOM_X, "Enter the number of sectors per track (1-63): ");
+ wclrtoeol(stdscr);
+ if (get_string(stdscr, response, sizeof(response), def) > 0)
+ {
+ tmp_val = atoi(response);
+ /* TODO Check for the maximum value */
+ if (tmp_val > 0) {
+ disk_car->geom.sectors_per_head = tmp_val;
+ if(geo_modified==0)
+ geo_modified=1;
+ if(cyl_modified==0)
+ set_cylinders_from_size_up(disk_car);
+ } else
+ wprintw(stdscr,"Illegal sectors value");
+ }
+ }
+ default_option=3;
+ break;
+ case 'n':
+ case 'N':
+ {
+ sprintf(def, "%u", disk_car->sector_size);
+ mvwaddstr(stdscr,INTER_GEOM_Y, INTER_GEOM_X, "Enter the sector size (512, 1024, 2048, 4096, 8192): ");
+ wclrtoeol(stdscr);
+ if (get_string(stdscr, response, sizeof(response), def) > 0) {
+ tmp_val = atoi(response);
+ if(change_sector_size(disk_car, cyl_modified, tmp_val))
+ wprintw(stdscr,"Illegal sector size");
+ else
+ geo_modified=2;
+ }
+ }
+ default_option=4;
+ break;
+ case key_ESC:
+ case 'q':
+ case 'Q':
+ done = 1;
+ break;
+ }
+ if(cyl_modified!=0)
+ disk_car->disk_size=(uint64_t)disk_car->geom.cylinders*disk_car->geom.heads_per_cylinder*disk_car->geom.sectors_per_head*disk_car->sector_size;
+ }
+ if(geo_modified!=0)
+ {
+ disk_car->disk_size=(uint64_t)disk_car->geom.cylinders*disk_car->geom.heads_per_cylinder*disk_car->geom.sectors_per_head*disk_car->sector_size;
+#ifdef __APPLE__
+ /* On MacOSX if HD contains some bad sectors, the disk size may not be correctly detected */
+ disk_car->disk_real_size=disk_car->disk_size;
+#endif
+ log_info("New geometry\n%s sector_size=%u\n", disk_car->description(disk_car), disk_car->sector_size);
+ autoset_unit(disk_car);
+ if(geo_modified==2)
+ return 1;
+ }
+ return 0;
+}