123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240 |
- #ifndef _LARGEFILE64_SOURCE
- # define _LARGEFILE64_SOURCE
- #endif
- #include <assert.h> /* assert */
- #include <sys/mount.h>
- #if !defined(BLKSSZGET)
- # define BLKSSZGET _IO(0x12, 104)
- #endif
- #if !defined(BLKGETSIZE64)
- # define BLKGETSIZE64 _IOR(0x12,114,size_t)
- #endif
- #include "libbb.h"
- #include "unicode.h"
- #if BB_LITTLE_ENDIAN
- # define inline_if_little_endian ALWAYS_INLINE
- #else
- # define inline_if_little_endian
- #endif
- #ifndef ENABLE_FEATURE_FDISK_BLKSIZE
- # define ENABLE_FEATURE_FDISK_BLKSIZE 0
- # define IF_FEATURE_FDISK_BLKSIZE(a)
- #endif
- #define DEFAULT_SECTOR_SIZE 512
- #define DEFAULT_SECTOR_SIZE_STR "512"
- #define MAX_SECTOR_SIZE 2048
- #define SECTOR_SIZE 512
- #define MAXIMUM_PARTS 60
- #define ACTIVE_FLAG 0x80
- #define EXTENDED 0x05
- #define WIN98_EXTENDED 0x0f
- #define LINUX_PARTITION 0x81
- #define LINUX_SWAP 0x82
- #define LINUX_NATIVE 0x83
- #define LINUX_EXTENDED 0x85
- #define LINUX_LVM 0x8e
- #define LINUX_RAID 0xfd
- enum {
- OPT_b = 1 << 0,
- OPT_C = 1 << 1,
- OPT_H = 1 << 2,
- OPT_l = 1 << 3,
- OPT_S = 1 << 4,
- OPT_u = 1 << 5,
- OPT_s = (1 << 6) * ENABLE_FEATURE_FDISK_BLKSIZE,
- };
- typedef unsigned long long ullong;
- typedef uint32_t sector_t;
- #if UINT_MAX == 0xffffffff
- # define SECT_FMT ""
- #elif ULONG_MAX == 0xffffffff
- # define SECT_FMT "l"
- #else
- # error Cant detect sizeof(uint32_t)
- #endif
- struct hd_geometry {
- unsigned char heads;
- unsigned char sectors;
- unsigned short cylinders;
- unsigned long start;
- };
- #define HDIO_GETGEO 0x0301
- #if ENABLE_FEATURE_FDISK_WRITABLE \
- || ENABLE_FEATURE_SGI_LABEL \
- || ENABLE_FEATURE_SUN_LABEL
- static const char msg_building_new_label[] ALIGN1 =
- "Building a new %s. Changes will remain in memory only,\n"
- "until you decide to write them. After that the previous content\n"
- "won't be recoverable.\n\n";
- static const char msg_part_already_defined[] ALIGN1 =
- "Partition %u is already defined, delete it before re-adding\n";
- #endif
- struct partition {
- unsigned char boot_ind;
- unsigned char head;
- unsigned char sector;
- unsigned char cyl;
- unsigned char sys_ind;
- unsigned char end_head;
- unsigned char end_sector;
- unsigned char end_cyl;
- unsigned char start4[4];
- unsigned char size4[4];
- } PACKED;
- struct pte {
- struct partition *part_table;
- struct partition *ext_pointer;
- sector_t offset_from_dev_start;
- char *sectorbuffer;
- #if ENABLE_FEATURE_FDISK_WRITABLE
- char changed;
- #endif
- };
- #define unable_to_open "can't open '%s'"
- #define unable_to_read "can't read '%s'"
- #define unable_to_seek "can't seek '%s'"
- enum label_type {
- LABEL_DOS, LABEL_SUN, LABEL_SGI, LABEL_AIX, LABEL_OSF, LABEL_GPT
- };
- #define LABEL_IS_DOS (LABEL_DOS == current_label_type)
- #if ENABLE_FEATURE_SUN_LABEL
- #define LABEL_IS_SUN (LABEL_SUN == current_label_type)
- #define STATIC_SUN static
- #else
- #define LABEL_IS_SUN 0
- #define STATIC_SUN extern
- #endif
- #if ENABLE_FEATURE_SGI_LABEL
- #define LABEL_IS_SGI (LABEL_SGI == current_label_type)
- #define STATIC_SGI static
- #else
- #define LABEL_IS_SGI 0
- #define STATIC_SGI extern
- #endif
- #if ENABLE_FEATURE_AIX_LABEL
- #define LABEL_IS_AIX (LABEL_AIX == current_label_type)
- #define STATIC_AIX static
- #else
- #define LABEL_IS_AIX 0
- #define STATIC_AIX extern
- #endif
- #if ENABLE_FEATURE_OSF_LABEL
- #define LABEL_IS_OSF (LABEL_OSF == current_label_type)
- #define STATIC_OSF static
- #else
- #define LABEL_IS_OSF 0
- #define STATIC_OSF extern
- #endif
- #if ENABLE_FEATURE_GPT_LABEL
- #define LABEL_IS_GPT (LABEL_GPT == current_label_type)
- #define STATIC_GPT static
- #else
- #define LABEL_IS_GPT 0
- #define STATIC_GPT extern
- #endif
- enum action { OPEN_MAIN, TRY_ONLY, CREATE_EMPTY_DOS, CREATE_EMPTY_SUN };
- static void update_units(void);
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static void change_units(void);
- static void reread_partition_table(int leave);
- static void delete_partition(int i);
- static unsigned get_partition(int warn, unsigned max);
- static void list_types(const char *const *sys);
- static sector_t read_int(sector_t low, sector_t dflt, sector_t high, sector_t base, const char *mesg);
- #endif
- static const char *partition_type(unsigned char type);
- static void get_geometry(void);
- static void read_pte(struct pte *pe, sector_t offset);
- #if ENABLE_FEATURE_SUN_LABEL || ENABLE_FEATURE_FDISK_WRITABLE
- static int get_boot(enum action what);
- #else
- static int get_boot(void);
- #endif
- static sector_t get_start_sect(const struct partition *p);
- static sector_t get_nr_sects(const struct partition *p);
- static const char *const i386_sys_types[] ALIGN_PTR = {
- "\x00" "Empty",
- "\x01" "FAT12",
- "\x04" "FAT16 <32M",
- "\x05" "Extended",
- "\x06" "FAT16",
- "\x07" "HPFS/NTFS",
- "\x0a" "OS/2 Boot Manager",
- "\x0b" "Win95 FAT32",
- "\x0c" "Win95 FAT32 (LBA)",
- "\x0e" "Win95 FAT16 (LBA)",
- "\x0f" "Win95 Ext'd (LBA)",
- "\x11" "Hidden FAT12",
- "\x12" "Compaq diagnostics",
- "\x14" "Hidden FAT16 <32M",
- "\x16" "Hidden FAT16",
- "\x17" "Hidden HPFS/NTFS",
- "\x1b" "Hidden Win95 FAT32",
- "\x1c" "Hidden W95 FAT32 (LBA)",
- "\x1e" "Hidden W95 FAT16 (LBA)",
- "\x3c" "Part.Magic recovery",
- "\x41" "PPC PReP Boot",
- "\x42" "SFS",
- "\x63" "GNU HURD or SysV",
- "\x80" "Old Minix",
- "\x81" "Minix / old Linux",
- "\x82" "Linux swap",
- "\x83" "Linux",
- "\x84" "OS/2 hidden C: drive",
- "\x85" "Linux extended",
- "\x86" "NTFS volume set",
- "\x87" "NTFS volume set",
- "\x8e" "Linux LVM",
- "\x9f" "BSD/OS",
- "\xa0" "Thinkpad hibernation",
- "\xa5" "FreeBSD",
- "\xa6" "OpenBSD",
- "\xa8" "Darwin UFS",
- "\xa9" "NetBSD",
- "\xab" "Darwin boot",
- "\xaf" "HFS / HFS+",
- "\xb7" "BSDI fs",
- "\xb8" "BSDI swap",
- "\xbe" "Solaris boot",
- "\xeb" "BeOS fs",
- "\xee" "EFI GPT",
- "\xef" "EFI (FAT-12/16/32)",
- "\xf0" "Linux/PA-RISC boot",
- "\xf2" "DOS secondary",
- "\xf8" "EBBR protective",
- "\xfd" "Linux raid autodetect",
- #if 0
- "\x02" "XENIX root",
- "\x03" "XENIX usr",
- "\x08" "AIX",
- "\x09" "AIX bootable",
- "\x10" "OPUS",
- "\x18" "AST SmartSleep",
- "\x24" "NEC DOS",
- "\x39" "Plan 9",
- "\x40" "Venix 80286",
- "\x4d" "QNX4.x",
- "\x4e" "QNX4.x 2nd part",
- "\x4f" "QNX4.x 3rd part",
- "\x50" "OnTrack DM",
- "\x51" "OnTrack DM6 Aux1",
- "\x52" "CP/M",
- "\x53" "OnTrack DM6 Aux3",
- "\x54" "OnTrackDM6",
- "\x55" "EZ-Drive",
- "\x56" "Golden Bow",
- "\x5c" "Priam Edisk",
- "\x61" "SpeedStor",
- "\x64" "Novell Netware 286",
- "\x65" "Novell Netware 386",
- "\x70" "DiskSecure Multi-Boot",
- "\x75" "PC/IX",
- "\x93" "Amoeba",
- "\x94" "Amoeba BBT",
- "\xa7" "NeXTSTEP",
- "\xbb" "Boot Wizard hidden",
- "\xc1" "DRDOS/sec (FAT-12)",
- "\xc4" "DRDOS/sec (FAT-16 < 32M)",
- "\xc6" "DRDOS/sec (FAT-16)",
- "\xc7" "Syrinx",
- "\xda" "Non-FS data",
- "\xdb" "CP/M / CTOS / ...",
- "\xde" "Dell Utility",
- "\xdf" "BootIt",
- "\xe1" "DOS access",
- "\xe3" "DOS R/O",
- "\xe4" "SpeedStor",
- "\xf1" "SpeedStor",
- "\xf4" "SpeedStor",
- "\xfe" "LANstep",
- "\xff" "BBT",
- #endif
- NULL
- };
- enum {
- dev_fd = 3
- };
- struct globals {
- char *line_ptr;
- const char *disk_device;
- int g_partitions;
- unsigned units_per_sector;
- unsigned sector_size;
- unsigned user_set_sector_size;
- unsigned sector_offset;
- unsigned g_heads, g_sectors, g_cylinders;
- smallint current_label_type;
- smallint display_in_cyl_units;
- #if ENABLE_FEATURE_OSF_LABEL
- smallint possibly_osf_label;
- #endif
- smallint listing;
- smallint dos_compatible_flag;
- #if ENABLE_FEATURE_FDISK_WRITABLE
-
- smallint nowarn;
- #endif
- int ext_index;
- unsigned user_cylinders, user_heads, user_sectors;
- unsigned pt_heads, pt_sectors;
- unsigned kern_heads, kern_sectors;
- sector_t extended_offset;
- sector_t total_number_of_sectors;
- jmp_buf listingbuf;
- char line_buffer[80];
-
- char MBRbuffer[MAX_SECTOR_SIZE];
-
- struct pte ptes[MAXIMUM_PARTS];
- };
- #define G (*ptr_to_globals)
- #define line_ptr (G.line_ptr )
- #define disk_device (G.disk_device )
- #define g_partitions (G.g_partitions )
- #define units_per_sector (G.units_per_sector )
- #define sector_size (G.sector_size )
- #define user_set_sector_size (G.user_set_sector_size)
- #define sector_offset (G.sector_offset )
- #define g_heads (G.g_heads )
- #define g_sectors (G.g_sectors )
- #define g_cylinders (G.g_cylinders )
- #define current_label_type (G.current_label_type )
- #define display_in_cyl_units (G.display_in_cyl_units)
- #define possibly_osf_label (G.possibly_osf_label )
- #define listing (G.listing )
- #define dos_compatible_flag (G.dos_compatible_flag )
- #define nowarn (G.nowarn )
- #define ext_index (G.ext_index )
- #define user_cylinders (G.user_cylinders )
- #define user_heads (G.user_heads )
- #define user_sectors (G.user_sectors )
- #define pt_heads (G.pt_heads )
- #define pt_sectors (G.pt_sectors )
- #define kern_heads (G.kern_heads )
- #define kern_sectors (G.kern_sectors )
- #define extended_offset (G.extended_offset )
- #define total_number_of_sectors (G.total_number_of_sectors)
- #define listingbuf (G.listingbuf )
- #define line_buffer (G.line_buffer )
- #define MBRbuffer (G.MBRbuffer )
- #define ptes (G.ptes )
- #define INIT_G() do { \
- SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
- sector_size = DEFAULT_SECTOR_SIZE; \
- sector_offset = 1; \
- g_partitions = 4; \
- units_per_sector = 1; \
- dos_compatible_flag = 1; \
- } while (0)
- static sector_t bb_BLKGETSIZE_sectors(int fd)
- {
- uint64_t v64;
- unsigned long longsectors;
- if (ioctl(fd, BLKGETSIZE64, &v64) == 0) {
-
- v64 >>= 9;
- if (v64 != (sector_t)v64) {
- ret_trunc:
-
- bb_simple_error_msg("device has more than 2^32 sectors, can't use all of them");
- v64 = (uint32_t)-1L;
- }
- return v64;
- }
-
- if (ioctl(fd, BLKGETSIZE, &longsectors)) {
-
- off_t sz = lseek(fd, 0, SEEK_END);
- longsectors = 0;
- if (sz > 0)
- longsectors = (uoff_t)sz / sector_size;
- lseek(fd, 0, SEEK_SET);
- }
- if (sizeof(long) > sizeof(sector_t)
- && longsectors != (sector_t)longsectors
- ) {
- goto ret_trunc;
- }
- return longsectors;
- }
- #define IS_EXTENDED(i) \
- ((i) == EXTENDED || (i) == WIN98_EXTENDED || (i) == LINUX_EXTENDED)
- #define cround(n) (display_in_cyl_units ? ((n)/units_per_sector)+1 : (n))
- #define scround(x) (((x)+units_per_sector-1)/units_per_sector)
- #define pt_offset(b, n) \
- ((struct partition *)((b) + 0x1be + (n) * sizeof(struct partition)))
- #define sector(s) ((s) & 0x3f)
- #define cylinder(s, c) ((c) | (((s) & 0xc0) << 2))
- static void
- close_dev_fd(void)
- {
-
- xmove_fd(xopen(bb_dev_null, O_RDONLY), dev_fd);
- }
- static const char *
- partname(const char *dev, int pno, int lth)
- {
- const char *p;
- int w, wp;
- int bufsiz;
- char *bufp;
- bufp = auto_string(xzalloc(80));
- bufsiz = 80;
- w = strlen(dev);
- p = "";
- if (isdigit(dev[w-1]))
- p = "p";
-
- if (strcmp(dev + w - 4, "disc") == 0) {
- w -= 4;
- p = "part";
- }
- wp = strlen(p);
- if (lth) {
- snprintf(bufp, bufsiz, "%*.*s%s%-2u",
- lth-wp-2, w, dev, p, pno);
- } else {
- snprintf(bufp, bufsiz, "%.*s%s%-2u", w, dev, p, pno);
- }
- return bufp;
- }
- #if ENABLE_FEATURE_SGI_LABEL || ENABLE_FEATURE_OSF_LABEL
- static ALWAYS_INLINE struct partition *
- get_part_table(int i)
- {
- return ptes[i].part_table;
- }
- #endif
- static ALWAYS_INLINE const char *
- str_units(void)
- {
- return display_in_cyl_units ? "cylinder" : "sector";
- }
- static int
- valid_part_table_flag(const char *mbuffer)
- {
- return (mbuffer[510] == 0x55 && (uint8_t)mbuffer[511] == 0xaa);
- }
- static void fdisk_fatal(const char *why)
- {
- if (listing) {
- close_dev_fd();
- longjmp(listingbuf, 1);
- }
- bb_error_msg_and_die(why, disk_device);
- }
- static void
- seek_sector(sector_t secno)
- {
- #if ENABLE_FDISK_SUPPORT_LARGE_DISKS
- off64_t off = (off64_t)secno * sector_size;
- if (lseek64(dev_fd, off, SEEK_SET) == (off64_t) -1)
- fdisk_fatal(unable_to_seek);
- #else
- uint64_t off = (uint64_t)secno * sector_size;
- if (off > MAXINT(off_t)
- || lseek(dev_fd, (off_t)off, SEEK_SET) == (off_t) -1
- ) {
- fdisk_fatal(unable_to_seek);
- }
- #endif
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static void
- set_all_unchanged(void)
- {
- int i;
- for (i = 0; i < MAXIMUM_PARTS; i++)
- ptes[i].changed = 0;
- }
- static ALWAYS_INLINE void
- set_changed(int i)
- {
- ptes[i].changed = 1;
- }
- static ALWAYS_INLINE void
- write_part_table_flag(char *b)
- {
- b[510] = 0x55;
- b[511] = 0xaa;
- }
- static int
- read_line(const char *prompt)
- {
- int sz;
- sz = read_line_input(NULL, prompt, line_buffer, sizeof(line_buffer));
- if (sz <= 0)
- exit_SUCCESS();
- if (line_buffer[sz-1] == '\n')
- line_buffer[--sz] = '\0';
- line_ptr = line_buffer;
- while (*line_ptr != '\0' && (unsigned char)*line_ptr <= ' ')
- line_ptr++;
- return *line_ptr;
- }
- static char
- read_nonempty(const char *mesg)
- {
- while (!read_line(mesg))
- continue;
- return *line_ptr;
- }
- static char
- read_maybe_empty(const char *mesg)
- {
- if (!read_line(mesg)) {
- line_ptr = line_buffer;
- line_ptr[0] = '\n';
- line_ptr[1] = '\0';
- }
- return line_ptr[0];
- }
- static int
- read_hex(const char *const *sys)
- {
- unsigned long v;
- while (1) {
- read_nonempty("Hex code (type L to list codes): ");
- if ((line_ptr[0] | 0x20) == 'l') {
- list_types(sys);
- continue;
- }
- v = bb_strtoul(line_ptr, NULL, 16);
- if (v <= 0xff)
- return v;
- }
- }
- static void
- write_sector(sector_t secno, const void *buf)
- {
- seek_sector(secno);
- xwrite(dev_fd, buf, sector_size);
- }
- #endif
- #include "fdisk_aix.c"
- struct sun_partition {
- unsigned char info[128];
- unsigned char spare0[14];
- struct sun_info {
- unsigned char spare1;
- unsigned char id;
- unsigned char spare2;
- unsigned char flags;
- } infos[8];
- unsigned char spare1[246];
- unsigned short rspeed;
- unsigned short pcylcount;
- unsigned short sparecyl;
- unsigned char spare2[4];
- unsigned short ilfact;
- unsigned short ncyl;
- unsigned short nacyl;
- unsigned short ntrks;
- unsigned short nsect;
- unsigned char spare3[4];
- struct sun_partinfo {
- uint32_t start_cylinder;
- uint32_t num_sectors;
- } partitions[8];
- unsigned short magic;
- unsigned short csum;
- } FIX_ALIASING;
- typedef struct sun_partition sun_partition;
- #define sunlabel ((sun_partition *)MBRbuffer)
- STATIC_OSF void bsd_select(void);
- STATIC_OSF void xbsd_print_disklabel(int);
- #include "fdisk_osf.c"
- STATIC_GPT void gpt_list_table(int xtra);
- #include "fdisk_gpt.c"
- #if ENABLE_FEATURE_SGI_LABEL || ENABLE_FEATURE_SUN_LABEL
- static uint16_t
- fdisk_swap16(uint16_t x)
- {
- return (x << 8) | (x >> 8);
- }
- static uint32_t
- fdisk_swap32(uint32_t x)
- {
- return (x << 24) |
- ((x & 0xFF00) << 8) |
- ((x & 0xFF0000) >> 8) |
- (x >> 24);
- }
- #endif
- STATIC_SGI const char *const sgi_sys_types[];
- STATIC_SGI unsigned sgi_get_num_sectors(int i);
- STATIC_SGI int sgi_get_sysid(int i);
- STATIC_SGI void sgi_delete_partition(int i);
- STATIC_SGI void sgi_change_sysid(int i, int sys);
- STATIC_SGI void sgi_list_table(int xtra);
- #if ENABLE_FEATURE_FDISK_ADVANCED
- STATIC_SGI void sgi_set_xcyl(void);
- #endif
- STATIC_SGI int verify_sgi(int verbose);
- STATIC_SGI void sgi_add_partition(int n, int sys);
- STATIC_SGI void sgi_set_swappartition(int i);
- STATIC_SGI const char *sgi_get_bootfile(void);
- STATIC_SGI void sgi_set_bootfile(const char* aFile);
- STATIC_SGI void create_sgiinfo(void);
- STATIC_SGI void sgi_write_table(void);
- STATIC_SGI void sgi_set_bootpartition(int i);
- #include "fdisk_sgi.c"
- STATIC_SUN const char *const sun_sys_types[];
- STATIC_SUN void sun_delete_partition(int i);
- STATIC_SUN void sun_change_sysid(int i, int sys);
- STATIC_SUN void sun_list_table(int xtra);
- STATIC_SUN void add_sun_partition(int n, int sys);
- #if ENABLE_FEATURE_FDISK_ADVANCED
- STATIC_SUN void sun_set_alt_cyl(void);
- STATIC_SUN void sun_set_ncyl(int cyl);
- STATIC_SUN void sun_set_xcyl(void);
- STATIC_SUN void sun_set_ilfact(void);
- STATIC_SUN void sun_set_rspeed(void);
- STATIC_SUN void sun_set_pcylcount(void);
- #endif
- STATIC_SUN void toggle_sunflags(int i, unsigned char mask);
- STATIC_SUN void verify_sun(void);
- STATIC_SUN void sun_write_table(void);
- #include "fdisk_sun.c"
- static inline_if_little_endian unsigned
- read4_little_endian(const unsigned char *cp)
- {
- uint32_t v;
- move_from_unaligned32(v, cp);
- return SWAP_LE32(v);
- }
- static sector_t
- get_start_sect(const struct partition *p)
- {
- return read4_little_endian(p->start4);
- }
- static sector_t
- get_nr_sects(const struct partition *p)
- {
- return read4_little_endian(p->size4);
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static inline_if_little_endian void
- store4_little_endian(unsigned char *cp, unsigned val)
- {
- uint32_t v = SWAP_LE32(val);
- move_to_unaligned32(cp, v);
- }
- static void
- set_start_sect(struct partition *p, unsigned start_sect)
- {
- store4_little_endian(p->start4, start_sect);
- }
- static void
- set_nr_sects(struct partition *p, unsigned nr_sects)
- {
- store4_little_endian(p->size4, nr_sects);
- }
- #endif
- static void
- read_pte(struct pte *pe, sector_t offset)
- {
- pe->offset_from_dev_start = offset;
- pe->sectorbuffer = xzalloc(sector_size);
- seek_sector(offset);
-
- if (full_read(dev_fd, pe->sectorbuffer, sector_size) != sector_size)
- fdisk_fatal(unable_to_read);
- #if ENABLE_FEATURE_FDISK_WRITABLE
- pe->changed = 0;
- #endif
- pe->part_table = pe->ext_pointer = NULL;
- }
- static sector_t
- get_partition_start_from_dev_start(const struct pte *pe)
- {
- return pe->offset_from_dev_start + get_start_sect(pe->part_table);
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
- #ifdef UNUSED
- static int
- is_dos_partition(int t)
- {
- return (t == 1 || t == 4 || t == 6 ||
- t == 0x0b || t == 0x0c || t == 0x0e ||
- t == 0x11 || t == 0x12 || t == 0x14 || t == 0x16 ||
- t == 0x1b || t == 0x1c || t == 0x1e || t == 0x24 ||
- t == 0xc1 || t == 0xc4 || t == 0xc6);
- }
- #endif
- static void
- menu(void)
- {
- puts("Command Action");
- if (LABEL_IS_SUN) {
- puts("a\ttoggle a read only flag");
- puts("b\tedit bsd disklabel");
- puts("c\ttoggle the mountable flag");
- puts("d\tdelete a partition");
- puts("l\tlist known partition types");
- puts("n\tadd a new partition");
- puts("o\tcreate a new empty DOS partition table");
- puts("p\tprint the partition table");
- puts("q\tquit without saving changes");
- puts("s\tcreate a new empty Sun disklabel");
- puts("t\tchange a partition's system id");
- puts("u\tchange display/entry units");
- puts("v\tverify the partition table");
- puts("w\twrite table to disk and exit");
- #if ENABLE_FEATURE_FDISK_ADVANCED
- puts("x\textra functionality (experts only)");
- #endif
- } else if (LABEL_IS_SGI) {
- puts("a\tselect bootable partition");
- puts("b\tedit bootfile entry");
- puts("c\tselect sgi swap partition");
- puts("d\tdelete a partition");
- puts("l\tlist known partition types");
- puts("n\tadd a new partition");
- puts("o\tcreate a new empty DOS partition table");
- puts("p\tprint the partition table");
- puts("q\tquit without saving changes");
- puts("s\tcreate a new empty Sun disklabel");
- puts("t\tchange a partition's system id");
- puts("u\tchange display/entry units");
- puts("v\tverify the partition table");
- puts("w\twrite table to disk and exit");
- } else if (LABEL_IS_AIX) {
- puts("o\tcreate a new empty DOS partition table");
- puts("q\tquit without saving changes");
- puts("s\tcreate a new empty Sun disklabel");
- } else if (LABEL_IS_GPT) {
- puts("o\tcreate a new empty DOS partition table");
- puts("p\tprint the partition table");
- puts("q\tquit without saving changes");
- puts("s\tcreate a new empty Sun disklabel");
- } else {
- puts("a\ttoggle a bootable flag");
- puts("b\tedit bsd disklabel");
- puts("c\ttoggle the dos compatibility flag");
- puts("d\tdelete a partition");
- puts("l\tlist known partition types");
- puts("n\tadd a new partition");
- puts("o\tcreate a new empty DOS partition table");
- puts("p\tprint the partition table");
- puts("q\tquit without saving changes");
- puts("s\tcreate a new empty Sun disklabel");
- puts("t\tchange a partition's system id");
- puts("u\tchange display/entry units");
- puts("v\tverify the partition table");
- puts("w\twrite table to disk and exit");
- #if ENABLE_FEATURE_FDISK_ADVANCED
- puts("x\textra functionality (experts only)");
- #endif
- }
- }
- #endif
- #if ENABLE_FEATURE_FDISK_ADVANCED
- static void
- xmenu(void)
- {
- puts("Command Action");
- if (LABEL_IS_SUN) {
- puts("a\tchange number of alternate cylinders");
- puts("c\tchange number of cylinders");
- puts("d\tprint the raw data in the partition table");
- puts("e\tchange number of extra sectors per cylinder");
- puts("h\tchange number of heads");
- puts("i\tchange interleave factor");
- puts("o\tchange rotation speed (rpm)");
- puts("p\tprint the partition table");
- puts("q\tquit without saving changes");
- puts("r\treturn to main menu");
- puts("s\tchange number of sectors/track");
- puts("v\tverify the partition table");
- puts("w\twrite table to disk and exit");
- puts("y\tchange number of physical cylinders");
- } else if (LABEL_IS_SGI) {
- puts("b\tmove beginning of data in a partition");
- puts("c\tchange number of cylinders");
- puts("d\tprint the raw data in the partition table");
- puts("e\tlist extended partitions");
- puts("g\tcreate an IRIX (SGI) partition table");
- puts("h\tchange number of heads");
- puts("p\tprint the partition table");
- puts("q\tquit without saving changes");
- puts("r\treturn to main menu");
- puts("s\tchange number of sectors/track");
- puts("v\tverify the partition table");
- puts("w\twrite table to disk and exit");
- } else if (LABEL_IS_AIX) {
- puts("b\tmove beginning of data in a partition");
- puts("c\tchange number of cylinders");
- puts("d\tprint the raw data in the partition table");
- puts("e\tlist extended partitions");
- puts("g\tcreate an IRIX (SGI) partition table");
- puts("h\tchange number of heads");
- puts("p\tprint the partition table");
- puts("q\tquit without saving changes");
- puts("r\treturn to main menu");
- puts("s\tchange number of sectors/track");
- puts("v\tverify the partition table");
- puts("w\twrite table to disk and exit");
- } else {
- puts("b\tmove beginning of data in a partition");
- puts("c\tchange number of cylinders");
- puts("d\tprint the raw data in the partition table");
- puts("e\tlist extended partitions");
- puts("f\tfix partition order");
- #if ENABLE_FEATURE_SGI_LABEL
- puts("g\tcreate an IRIX (SGI) partition table");
- #endif
- puts("h\tchange number of heads");
- puts("p\tprint the partition table");
- puts("q\tquit without saving changes");
- puts("r\treturn to main menu");
- puts("s\tchange number of sectors/track");
- puts("v\tverify the partition table");
- puts("w\twrite table to disk and exit");
- }
- }
- #endif
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static const char *const *
- get_sys_types(void)
- {
- return (
- LABEL_IS_SUN ? sun_sys_types :
- LABEL_IS_SGI ? sgi_sys_types :
- i386_sys_types);
- }
- #else
- #define get_sys_types() i386_sys_types
- #endif
- static const char *
- partition_type(unsigned char type)
- {
- int i;
- const char *const *types = get_sys_types();
- for (i = 0; types[i]; i++)
- if ((unsigned char)types[i][0] == type)
- return types[i] + 1;
- return "Unknown";
- }
- static int
- is_cleared_partition(const struct partition *p)
- {
-
- const char *cp = (const char *)p;
- int cnt = sizeof(*p);
- char bits = 0;
- while (--cnt >= 0)
- bits |= *cp++;
- return (bits == 0);
- }
- static void
- clear_partition(struct partition *p)
- {
- if (p)
- memset(p, 0, sizeof(*p));
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static int
- get_sysid(int i)
- {
- return LABEL_IS_SUN ? sunlabel->infos[i].id :
- (LABEL_IS_SGI ? sgi_get_sysid(i) :
- ptes[i].part_table->sys_ind);
- }
- static void
- list_types(const char *const *sys)
- {
- enum { COLS = 3 };
- unsigned last[COLS];
- unsigned done, next, size;
- int i;
- for (size = 0; sys[size]; size++)
- continue;
- done = 0;
- for (i = COLS-1; i >= 0; i--) {
- done += (size + i - done) / (i + 1);
- last[COLS-1 - i] = done;
- }
- i = done = next = 0;
- do {
- printf("%c%2x %-22.22s", i ? ' ' : '\n',
- (unsigned char)sys[next][0],
- sys[next] + 1);
- next = last[i++] + done;
- if (i >= COLS || next >= last[i]) {
- i = 0;
- next = ++done;
- }
- } while (done < last[0]);
- bb_putchar('\n');
- }
- #define set_hsc(h, s, c, sector) do \
- { \
- s = sector % g_sectors + 1; \
- sector /= g_sectors; \
- h = sector % g_heads; \
- sector /= g_heads; \
- c = sector & 0xff; \
- s |= (sector >> 2) & 0xc0; \
- } while (0)
- static void set_hsc_start_end(struct partition *p, sector_t start, sector_t stop)
- {
- if (dos_compatible_flag && (start / (g_sectors * g_heads) > 1023))
- start = g_heads * g_sectors * 1024 - 1;
- set_hsc(p->head, p->sector, p->cyl, start);
- if (dos_compatible_flag && (stop / (g_sectors * g_heads) > 1023))
- stop = g_heads * g_sectors * 1024 - 1;
- set_hsc(p->end_head, p->end_sector, p->end_cyl, stop);
- }
- static void
- set_partition(int i, int doext, sector_t start, sector_t stop, int sysid)
- {
- struct partition *p;
- sector_t offset;
- if (doext) {
- p = ptes[i].ext_pointer;
- offset = extended_offset;
- } else {
- p = ptes[i].part_table;
- offset = ptes[i].offset_from_dev_start;
- }
- p->boot_ind = 0;
- p->sys_ind = sysid;
- set_start_sect(p, start - offset);
- set_nr_sects(p, stop - start + 1);
- set_hsc_start_end(p, start, stop);
- ptes[i].changed = 1;
- }
- #endif
- static int
- warn_geometry(void)
- {
- if (g_heads && g_sectors && g_cylinders)
- return 0;
- printf("Unknown value(s) for:");
- if (!g_heads)
- printf(" heads");
- if (!g_sectors)
- printf(" sectors");
- if (!g_cylinders)
- printf(" cylinders");
- #if ENABLE_FEATURE_FDISK_WRITABLE
- puts(" (settable in the extra functions menu)");
- #else
- bb_putchar('\n');
- #endif
- return 1;
- }
- static void
- update_units(void)
- {
- int cyl_units = g_heads * g_sectors;
- if (display_in_cyl_units && cyl_units)
- units_per_sector = cyl_units;
- else
- units_per_sector = 1;
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static void
- warn_cylinders(void)
- {
- if (LABEL_IS_DOS && g_cylinders > 1024 && !nowarn)
- printf("\n"
- "The number of cylinders for this disk is set to %u.\n"
- "There is nothing wrong with that, but this is larger than 1024,\n"
- "and could in certain setups cause problems with:\n"
- "1) software that runs at boot time (e.g., old versions of LILO)\n"
- "2) booting and partitioning software from other OSs\n"
- " (e.g., DOS FDISK, OS/2 FDISK)\n",
- g_cylinders);
- }
- #endif
- static void
- read_extended(int ext)
- {
- int i;
- struct pte *pex;
- struct partition *p, *q;
- ext_index = ext;
- pex = &ptes[ext];
- pex->ext_pointer = pex->part_table;
- p = pex->part_table;
- if (!get_start_sect(p)) {
- puts("Bad offset in primary extended partition");
- return;
- }
- while (IS_EXTENDED(p->sys_ind)) {
- struct pte *pe = &ptes[g_partitions];
- if (g_partitions >= MAXIMUM_PARTS) {
-
- struct pte *pre = &ptes[g_partitions - 1];
- #if ENABLE_FEATURE_FDISK_WRITABLE
- printf("Warning: deleting partitions after %u\n",
- g_partitions);
- pre->changed = 1;
- #endif
- clear_partition(pre->ext_pointer);
- return;
- }
- read_pte(pe, extended_offset + get_start_sect(p));
- if (!extended_offset)
- extended_offset = get_start_sect(p);
- q = p = pt_offset(pe->sectorbuffer, 0);
- for (i = 0; i < 4; i++, p++) if (get_nr_sects(p)) {
- if (IS_EXTENDED(p->sys_ind)) {
- if (pe->ext_pointer)
- printf("Warning: extra link "
- "pointer in partition table"
- " %u\n", g_partitions + 1);
- else
- pe->ext_pointer = p;
- } else if (p->sys_ind) {
- if (pe->part_table)
- printf("Warning: ignoring extra "
- "data in partition table"
- " %u\n", g_partitions + 1);
- else
- pe->part_table = p;
- }
- }
-
- if (!pe->part_table) {
- if (q != pe->ext_pointer)
- pe->part_table = q;
- else
- pe->part_table = q + 1;
- }
- if (!pe->ext_pointer) {
- if (q != pe->part_table)
- pe->ext_pointer = q;
- else
- pe->ext_pointer = q + 1;
- }
- p = pe->ext_pointer;
- g_partitions++;
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
-
- remove:
- for (i = 4; i < g_partitions; i++) {
- struct pte *pe = &ptes[i];
- if (!get_nr_sects(pe->part_table)
- && (g_partitions > 5 || ptes[4].part_table->sys_ind)
- ) {
- printf("Omitting empty partition (%u)\n", i+1);
- delete_partition(i);
- goto remove;
- }
- }
- #endif
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static void
- create_doslabel(void)
- {
- printf(msg_building_new_label, "DOS disklabel");
- current_label_type = LABEL_DOS;
- #if ENABLE_FEATURE_OSF_LABEL
- possibly_osf_label = 0;
- #endif
- g_partitions = 4;
- memset(&MBRbuffer[510 - 4*16], 0, 4*16);
- write_part_table_flag(MBRbuffer);
- extended_offset = 0;
- set_all_unchanged();
- set_changed(0);
- get_boot(CREATE_EMPTY_DOS);
- }
- #endif
- static void
- get_sectorsize(void)
- {
- if (!user_set_sector_size) {
- int arg;
- if (ioctl(dev_fd, BLKSSZGET, &arg) == 0)
- sector_size = arg;
- if (sector_size != DEFAULT_SECTOR_SIZE)
- printf("Note: sector size is %u "
- "(not " DEFAULT_SECTOR_SIZE_STR ")\n",
- sector_size);
- }
- }
- static void
- get_kernel_geometry(void)
- {
- struct hd_geometry geometry;
- if (!ioctl(dev_fd, HDIO_GETGEO, &geometry)) {
- kern_heads = geometry.heads;
- kern_sectors = geometry.sectors;
-
- }
- }
- static void
- get_partition_table_geometry(void)
- {
- const unsigned char *bufp = (const unsigned char *)MBRbuffer;
- struct partition *p;
- int i, h, s, hh, ss;
- int first = 1;
- int bad = 0;
- if (!(valid_part_table_flag((char*)bufp)))
- return;
- hh = ss = 0;
- for (i = 0; i < 4; i++) {
- p = pt_offset(bufp, i);
- if (p->sys_ind != 0) {
- h = p->end_head + 1;
- s = (p->end_sector & 077);
- if (first) {
- hh = h;
- ss = s;
- first = 0;
- } else if (hh != h || ss != s)
- bad = 1;
- }
- }
- if (!first && !bad) {
- pt_heads = hh;
- pt_sectors = ss;
- }
- }
- static void
- get_geometry(void)
- {
- int sec_fac;
- get_sectorsize();
- sec_fac = sector_size / 512;
- #if ENABLE_FEATURE_SUN_LABEL
- guess_device_type();
- #endif
- g_heads = g_cylinders = g_sectors = 0;
- kern_heads = kern_sectors = 0;
- pt_heads = pt_sectors = 0;
- get_kernel_geometry();
- get_partition_table_geometry();
- g_heads = user_heads ? user_heads :
- pt_heads ? pt_heads :
- kern_heads ? kern_heads : 255;
- g_sectors = user_sectors ? user_sectors :
- pt_sectors ? pt_sectors :
- kern_sectors ? kern_sectors : 63;
- total_number_of_sectors = bb_BLKGETSIZE_sectors(dev_fd);
- sector_offset = 1;
- if (dos_compatible_flag)
- sector_offset = g_sectors;
- g_cylinders = total_number_of_sectors / (g_heads * g_sectors * sec_fac);
- if (!g_cylinders)
- g_cylinders = user_cylinders;
- }
- #if ENABLE_FEATURE_SUN_LABEL || ENABLE_FEATURE_FDISK_WRITABLE
- static int get_boot(enum action what)
- #else
- static int get_boot(void)
- #define get_boot(what) get_boot()
- #endif
- {
- int i, fd;
- g_partitions = 4;
- for (i = 0; i < 4; i++) {
- struct pte *pe = &ptes[i];
- pe->part_table = pt_offset(MBRbuffer, i);
- pe->ext_pointer = NULL;
- pe->offset_from_dev_start = 0;
- pe->sectorbuffer = MBRbuffer;
- #if ENABLE_FEATURE_FDISK_WRITABLE
- pe->changed = (what == CREATE_EMPTY_DOS);
- #endif
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
- if (what == CREATE_EMPTY_DOS IF_FEATURE_SUN_LABEL(|| what == CREATE_EMPTY_SUN))
- goto created_table;
- fd = open(disk_device, (option_mask32 & OPT_l) ? O_RDONLY : O_RDWR);
- if (fd < 0) {
- fd = open(disk_device, O_RDONLY);
- if (fd < 0) {
- if (what == TRY_ONLY)
- return 1;
- fdisk_fatal(unable_to_open);
- }
- printf("'%s' is opened for read only\n", disk_device);
- }
- xmove_fd(fd, dev_fd);
- if (512 != full_read(dev_fd, MBRbuffer, 512)) {
- if (what == TRY_ONLY) {
- close_dev_fd();
- return 1;
- }
- fdisk_fatal(unable_to_read);
- }
- #else
- fd = open(disk_device, O_RDONLY);
- if (fd < 0)
- return 1;
- if (512 != full_read(fd, MBRbuffer, 512)) {
- close(fd);
- return 1;
- }
- xmove_fd(fd, dev_fd);
- #endif
- get_geometry();
- update_units();
- #if ENABLE_FEATURE_SUN_LABEL
- if (check_sun_label())
- return 0;
- #endif
- #if ENABLE_FEATURE_SGI_LABEL
- if (check_sgi_label())
- return 0;
- #endif
- #if ENABLE_FEATURE_AIX_LABEL
- if (check_aix_label())
- return 0;
- #endif
- #if ENABLE_FEATURE_GPT_LABEL
- if (check_gpt_label())
- return 0;
- #endif
- #if ENABLE_FEATURE_OSF_LABEL
- if (check_osf_label()) {
- possibly_osf_label = 1;
- if (!valid_part_table_flag(MBRbuffer)) {
- current_label_type = LABEL_OSF;
- return 0;
- }
- puts("This disk has both DOS and BSD magic.\n"
- "Give the 'b' command to go to BSD mode.");
- }
- #endif
- #if !ENABLE_FEATURE_FDISK_WRITABLE
- if (!valid_part_table_flag(MBRbuffer))
- return -1;
- #else
- if (!valid_part_table_flag(MBRbuffer)) {
- if (what == OPEN_MAIN) {
- puts("Device contains neither a valid DOS "
- "partition table, nor Sun, SGI, OSF or GPT "
- "disklabel");
- #ifdef __sparc__
- IF_FEATURE_SUN_LABEL(create_sunlabel();)
- #else
- create_doslabel();
- #endif
- return 0;
- }
-
- return -1;
- }
- created_table:
- #endif
- IF_FEATURE_FDISK_WRITABLE(warn_cylinders();)
- warn_geometry();
- for (i = 0; i < 4; i++) {
- if (IS_EXTENDED(ptes[i].part_table->sys_ind)) {
- if (g_partitions != 4)
- printf("Ignoring extra extended "
- "partition %u\n", i + 1);
- else
- read_extended(i);
- }
- }
- for (i = 3; i < g_partitions; i++) {
- struct pte *pe = &ptes[i];
- if (!valid_part_table_flag(pe->sectorbuffer)) {
- printf("Warning: invalid flag 0x%02x,0x%02x of partition "
- "table %u will be corrected by w(rite)\n",
- pe->sectorbuffer[510],
- pe->sectorbuffer[511],
- i + 1);
- IF_FEATURE_FDISK_WRITABLE(pe->changed = 1;)
- }
- }
- return 0;
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static sector_t
- read_int(sector_t low, sector_t dflt, sector_t high, sector_t base, const char *mesg)
- {
- sector_t value;
- int default_ok = 1;
- const char *fmt = "%s (%u-%u, default %u): ";
- if (dflt < low || dflt > high) {
- fmt = "%s (%u-%u): ";
- default_ok = 0;
- }
- while (1) {
- int use_default = default_ok;
-
- do {
- printf(fmt, mesg, low, high, dflt);
- read_maybe_empty("");
- } while (*line_ptr != '\n' && !isdigit(*line_ptr)
- && *line_ptr != '-' && *line_ptr != '+');
- if (*line_ptr == '+' || *line_ptr == '-') {
- int minus = (*line_ptr == '-');
- unsigned scale_shift;
- if (sizeof(value) <= sizeof(long))
- value = strtoul(line_ptr + 1, NULL, 10);
- else
- value = strtoull(line_ptr + 1, NULL, 10);
-
- while (isdigit(*++line_ptr))
- use_default = 0;
- scale_shift = 0;
- switch (*line_ptr | 0x20) {
- case 'k':
- scale_shift = 10;
- break;
- case 'm':
- scale_shift = 20;
- break;
- case 'g':
- scale_shift = 30;
- break;
- case 't':
- scale_shift = 40;
- break;
- default:
- break;
- }
- if (scale_shift) {
- ullong bytes;
- unsigned long unit;
- bytes = (ullong) value << scale_shift;
- unit = sector_size * units_per_sector;
- bytes += unit/2;
- bytes /= unit;
- value = (bytes != 0 ? bytes - 1 : 0);
- }
- if (minus)
- value = -value;
- value += base;
- } else {
- if (sizeof(value) <= sizeof(long))
- value = strtoul(line_ptr, NULL, 10);
- else
- value = strtoull(line_ptr, NULL, 10);
- while (isdigit(*line_ptr)) {
- line_ptr++;
- use_default = 0;
- }
- }
- if (use_default) {
- value = dflt;
- printf("Using default value %u\n", value);
- }
- if (value >= low && value <= high)
- break;
- puts("Value is out of range");
- }
- return value;
- }
- static unsigned
- get_partition(int warn, unsigned max)
- {
- struct pte *pe;
- unsigned i;
- i = read_int(1, 0, max, 0, "Partition number") - 1;
- pe = &ptes[i];
- if (warn) {
- if ((!LABEL_IS_SUN && !LABEL_IS_SGI && !pe->part_table->sys_ind)
- || (LABEL_IS_SUN && (!sunlabel->partitions[i].num_sectors || !sunlabel->infos[i].id))
- || (LABEL_IS_SGI && !sgi_get_num_sectors(i))
- ) {
- printf("Warning: partition %u has empty type\n", i+1);
- }
- }
- return i;
- }
- static int
- get_existing_partition(int warn, unsigned max)
- {
- int pno = -1;
- unsigned i;
- for (i = 0; i < max; i++) {
- struct pte *pe = &ptes[i];
- struct partition *p = pe->part_table;
- if (p && !is_cleared_partition(p)) {
- if (pno >= 0)
- goto not_unique;
- pno = i;
- }
- }
- if (pno >= 0) {
- printf("Selected partition %u\n", pno+1);
- return pno;
- }
- puts("No partition is defined yet!");
- return -1;
- not_unique:
- return get_partition(warn, max);
- }
- static int
- get_nonexisting_partition(void)
- {
- const int max = 4;
- int pno = -1;
- unsigned i;
- for (i = 0; i < max; i++) {
- struct pte *pe = &ptes[i];
- struct partition *p = pe->part_table;
- if (p && is_cleared_partition(p)) {
- if (pno >= 0)
- goto not_unique;
- pno = i;
- }
- }
- if (pno >= 0) {
- printf("Selected partition %u\n", pno+1);
- return pno;
- }
- puts("All primary partitions have been defined already!");
- return -1;
- not_unique:
- return get_partition( 0, max);
- }
- static void
- change_units(void)
- {
- display_in_cyl_units = !display_in_cyl_units;
- update_units();
- printf("Changing display/entry units to %ss\n",
- str_units());
- }
- static void
- toggle_active(int i)
- {
- struct pte *pe = &ptes[i];
- struct partition *p = pe->part_table;
- if (IS_EXTENDED(p->sys_ind) && !p->boot_ind)
- printf("WARNING: Partition %u is an extended partition\n", i + 1);
- p->boot_ind = (p->boot_ind ? 0 : ACTIVE_FLAG);
- pe->changed = 1;
- }
- static void
- toggle_dos_compatibility_flag(void)
- {
- dos_compatible_flag = 1 - dos_compatible_flag;
- if (dos_compatible_flag) {
- sector_offset = g_sectors;
- printf("DOS Compatibility flag is %sset\n", "");
- } else {
- sector_offset = 1;
- printf("DOS Compatibility flag is %sset\n", "not ");
- }
- }
- static void
- delete_partition(int i)
- {
- struct pte *pe = &ptes[i];
- struct partition *p = pe->part_table;
- struct partition *q = pe->ext_pointer;
- if (warn_geometry())
- return;
- pe->changed = 1;
- if (LABEL_IS_SUN) {
- sun_delete_partition(i);
- return;
- }
- if (LABEL_IS_SGI) {
- sgi_delete_partition(i);
- return;
- }
- if (i < 4) {
- if (IS_EXTENDED(p->sys_ind) && i == ext_index) {
- g_partitions = 4;
- ptes[ext_index].ext_pointer = NULL;
- extended_offset = 0;
- }
- clear_partition(p);
- return;
- }
- if (!q->sys_ind && i > 4) {
-
- --g_partitions;
- --i;
- clear_partition(ptes[i].ext_pointer);
- ptes[i].changed = 1;
- } else {
-
- if (i > 4) {
-
- p = ptes[i-1].ext_pointer;
- *p = *q;
- set_start_sect(p, get_start_sect(q));
- set_nr_sects(p, get_nr_sects(q));
- ptes[i-1].changed = 1;
- } else if (g_partitions > 5) {
-
- pe = &ptes[5];
- if (pe->part_table)
- set_start_sect(pe->part_table,
- get_partition_start_from_dev_start(pe) -
- extended_offset);
- pe->offset_from_dev_start = extended_offset;
- pe->changed = 1;
- }
- if (g_partitions > 5) {
- g_partitions--;
- while (i < g_partitions) {
- ptes[i] = ptes[i+1];
- i++;
- }
- } else {
-
- clear_partition(ptes[i].part_table);
- }
- }
- }
- static void
- change_sysid(void)
- {
- int i, sys, origsys;
- struct partition *p;
-
- if (!LABEL_IS_SGI) {
- i = get_existing_partition(0, g_partitions);
- } else {
- i = get_partition(0, g_partitions);
- }
- if (i == -1)
- return;
- p = ptes[i].part_table;
- origsys = sys = get_sysid(i);
-
- if (!sys && !LABEL_IS_SGI && !LABEL_IS_SUN && !get_nr_sects(p)) {
- printf("Partition %u does not exist yet!\n", i + 1);
- return;
- }
- while (1) {
- sys = read_hex(get_sys_types());
- if (!sys && !LABEL_IS_SGI && !LABEL_IS_SUN) {
- puts("Type 0 means free space to many systems\n"
- "(but not to Linux). Having partitions of\n"
- "type 0 is probably unwise.");
-
- }
- if (!LABEL_IS_SUN && !LABEL_IS_SGI) {
- if (IS_EXTENDED(sys) != IS_EXTENDED(p->sys_ind)) {
- puts("You cannot change a partition into"
- " an extended one or vice versa");
- break;
- }
- }
- if (sys < 256) {
- #if ENABLE_FEATURE_SUN_LABEL
- if (LABEL_IS_SUN && i == 2 && sys != SUN_WHOLE_DISK)
- puts("Consider leaving partition 3 "
- "as Whole disk (5),\n"
- "as SunOS/Solaris expects it and "
- "even Linux likes it\n");
- #endif
- #if ENABLE_FEATURE_SGI_LABEL
- if (LABEL_IS_SGI &&
- (
- (i == 10 && sys != SGI_ENTIRE_DISK) ||
- (i == 8 && sys != 0)
- )
- ) {
- puts("Consider leaving partition 9 "
- "as volume header (0),\nand "
- "partition 11 as entire volume (6)"
- "as IRIX expects it\n");
- }
- #endif
- if (sys == origsys)
- break;
- if (LABEL_IS_SUN) {
- sun_change_sysid(i, sys);
- } else if (LABEL_IS_SGI) {
- sgi_change_sysid(i, sys);
- } else
- p->sys_ind = sys;
- printf("Changed system type of partition %u "
- "to %x (%s)\n", i + 1, sys,
- partition_type(sys));
- ptes[i].changed = 1;
-
-
- break;
- }
- }
- }
- #endif
- static void
- linear2chs(unsigned ls, unsigned *c, unsigned *h, unsigned *s)
- {
- int spc = g_heads * g_sectors;
- *c = ls / spc;
- ls = ls % spc;
- *h = ls / g_sectors;
- *s = ls % g_sectors + 1;
- }
- static void
- check_consistency(const struct partition *p, int partition)
- {
- unsigned pbc, pbh, pbs;
- unsigned pec, peh, pes;
- unsigned lbc, lbh, lbs;
- unsigned lec, leh, les;
- if (!g_heads || !g_sectors || (partition >= 4))
- return;
- pbc = cylinder(p->sector, p->cyl);
- pbh = p->head;
- pbs = sector(p->sector);
- pec = cylinder(p->end_sector, p->end_cyl);
- peh = p->end_head;
- pes = sector(p->end_sector);
- linear2chs(get_start_sect(p), &lbc, &lbh, &lbs);
- linear2chs(get_start_sect(p) + get_nr_sects(p) - 1, &lec, &leh, &les);
- if (g_cylinders <= 1024 && (pbc != lbc || pbh != lbh || pbs != lbs)) {
- printf("Partition %u has different physical/logical "
- "start (non-Linux?):\n", partition + 1);
- printf(" phys=(%u,%u,%u) ", pbc, pbh, pbs);
- printf("logical=(%u,%u,%u)\n", lbc, lbh, lbs);
- }
- if (g_cylinders <= 1024 && (pec != lec || peh != leh || pes != les)) {
- printf("Partition %u has different physical/logical "
- "end:\n", partition + 1);
- printf(" phys=(%u,%u,%u) ", pec, peh, pes);
- printf("logical=(%u,%u,%u)\n", lec, leh, les);
- }
- }
- static void
- list_disk_geometry(void)
- {
- ullong xbytes = total_number_of_sectors / (1024*1024 / 512);
- char x = 'M';
- if (xbytes >= 10000) {
- xbytes += 512;
- xbytes /= 1024;
- x = 'G';
- }
- printf("Disk %s: %llu %cB, %llu bytes, %"SECT_FMT"u sectors\n"
- "%u cylinders, %u heads, %u sectors/track\n"
- "Units: %ss of %u * %u = %u bytes\n"
- "\n",
- disk_device, xbytes, x,
- ((ullong)total_number_of_sectors * 512), total_number_of_sectors,
- g_cylinders, g_heads, g_sectors,
- str_units(),
- units_per_sector, sector_size, units_per_sector * sector_size
- );
- }
- static int
- wrong_p_order(int *prev)
- {
- const struct pte *pe;
- const struct partition *p;
- sector_t last_p_start_pos = 0, p_start_pos;
- unsigned i, last_i = 0;
- for (i = 0; i < g_partitions; i++) {
- if (i == 4) {
- last_i = 4;
- last_p_start_pos = 0;
- }
- pe = &ptes[i];
- p = pe->part_table;
- if (p->sys_ind) {
- p_start_pos = get_partition_start_from_dev_start(pe);
- if (last_p_start_pos > p_start_pos) {
- if (prev)
- *prev = last_i;
- return i;
- }
- last_p_start_pos = p_start_pos;
- last_i = i;
- }
- }
- return 0;
- }
- #if ENABLE_FEATURE_FDISK_ADVANCED
- static void
- fix_chain_of_logicals(void)
- {
- int j, oj, ojj, sj, sjj;
- struct partition *pj,*pjj,tmp;
-
-
- stage1:
- for (j = 5; j < g_partitions - 1; j++) {
- oj = ptes[j].offset_from_dev_start;
- ojj = ptes[j+1].offset_from_dev_start;
- if (oj > ojj) {
- ptes[j].offset_from_dev_start = ojj;
- ptes[j+1].offset_from_dev_start = oj;
- pj = ptes[j].part_table;
- set_start_sect(pj, get_start_sect(pj)+oj-ojj);
- pjj = ptes[j+1].part_table;
- set_start_sect(pjj, get_start_sect(pjj)+ojj-oj);
- set_start_sect(ptes[j-1].ext_pointer,
- ojj-extended_offset);
- set_start_sect(ptes[j].ext_pointer,
- oj-extended_offset);
- goto stage1;
- }
- }
-
- stage2:
- for (j = 4; j < g_partitions - 1; j++) {
- pj = ptes[j].part_table;
- pjj = ptes[j+1].part_table;
- sj = get_start_sect(pj);
- sjj = get_start_sect(pjj);
- oj = ptes[j].offset_from_dev_start;
- ojj = ptes[j+1].offset_from_dev_start;
- if (oj+sj > ojj+sjj) {
- tmp = *pj;
- *pj = *pjj;
- *pjj = tmp;
- set_start_sect(pj, ojj+sjj-oj);
- set_start_sect(pjj, oj+sj-ojj);
- goto stage2;
- }
- }
-
- for (j = 4; j < g_partitions; j++)
- ptes[j].changed = 1;
- }
- static void
- fix_partition_table_order(void)
- {
- struct pte *pei, *pek;
- int i,k;
- if (!wrong_p_order(NULL)) {
- puts("Ordering is already correct\n");
- return;
- }
- while ((i = wrong_p_order(&k)) != 0 && i < 4) {
-
-
- struct partition *pi, *pk, *pe, pbuf;
- pei = &ptes[i];
- pek = &ptes[k];
- pe = pei->ext_pointer;
- pei->ext_pointer = pek->ext_pointer;
- pek->ext_pointer = pe;
- pi = pei->part_table;
- pk = pek->part_table;
- memmove(&pbuf, pi, sizeof(struct partition));
- memmove(pi, pk, sizeof(struct partition));
- memmove(pk, &pbuf, sizeof(struct partition));
- pei->changed = pek->changed = 1;
- }
- if (i)
- fix_chain_of_logicals();
- puts("Done");
- }
- #endif
- static const char *
- chs_string11(unsigned cyl, unsigned head, unsigned sect)
- {
- char *buf = auto_string(xzalloc(sizeof(int)*3 * 3));
- sprintf(buf, "%u,%u,%u", cylinder(sect,cyl), head, sector(sect));
- return buf;
- }
- static void
- list_table(int xtra)
- {
- int i, w;
- if (LABEL_IS_SUN) {
- sun_list_table(xtra);
- return;
- }
- if (LABEL_IS_SGI) {
- sgi_list_table(xtra);
- return;
- }
- if (LABEL_IS_GPT) {
- gpt_list_table(xtra);
- return;
- }
- list_disk_geometry();
- if (LABEL_IS_OSF) {
- xbsd_print_disklabel(xtra);
- return;
- }
-
- w = strlen(disk_device);
- if (w && isdigit(disk_device[w-1]))
- w++;
- if (w < 7)
- w = 7;
- printf("%-*s Boot StartCHS EndCHS StartLBA EndLBA Sectors Size Id Type\n",
- w-1, "Device");
- for (i = 0; i < g_partitions; i++) {
- const struct partition *p;
- const struct pte *pe = &ptes[i];
- char boot4[4];
- char numstr6[6];
- sector_t start_sect;
- sector_t end_sect;
- sector_t nr_sects;
- p = pe->part_table;
- if (!p || is_cleared_partition(p))
- continue;
- sprintf(boot4, "%02x", p->boot_ind);
- if ((p->boot_ind & 0x7f) == 0) {
-
- boot4[0] = p->boot_ind ? '*' : ' ';
- boot4[1] = ' ';
- }
- start_sect = get_partition_start_from_dev_start(pe);
- end_sect = start_sect;
- nr_sects = get_nr_sects(p);
- if (nr_sects != 0)
- end_sect += nr_sects - 1;
- smart_ulltoa5((ullong)nr_sects * sector_size,
- numstr6, " KMGTPEZY")[0] = '\0';
- #define SFMT SECT_FMT
-
- printf("%s%s %-11s"" %-11s"" %10"SFMT"u %10"SFMT"u %10"SFMT"u %s %2x %s\n",
- partname(disk_device, i+1, w+2),
- boot4,
- chs_string11(p->cyl, p->head, p->sector),
- chs_string11(p->end_cyl, p->end_head, p->end_sector),
- start_sect,
- end_sect,
- nr_sects,
- numstr6,
- p->sys_ind,
- partition_type(p->sys_ind)
- );
- #undef SFMT
- check_consistency(p, i);
- }
-
-
- if (LABEL_IS_DOS && wrong_p_order(NULL)) {
-
- puts("\nPartition table entries are not in disk order");
- }
- }
- #if ENABLE_FEATURE_FDISK_ADVANCED
- static void
- x_list_table(int extend)
- {
- const struct pte *pe;
- const struct partition *p;
- int i;
- printf("\nDisk %s: %u heads, %u sectors, %u cylinders\n\n",
- disk_device, g_heads, g_sectors, g_cylinders);
- puts("Nr AF Hd Sec Cyl Hd Sec Cyl Start Size ID");
- for (i = 0; i < g_partitions; i++) {
- pe = &ptes[i];
- p = (extend ? pe->ext_pointer : pe->part_table);
- if (p != NULL) {
- printf("%2u %02x%4u%4u%5u%4u%4u%5u%11"SECT_FMT"u%11"SECT_FMT"u %02x\n",
- i + 1, p->boot_ind,
- p->head,
- sector(p->sector),
- cylinder(p->sector, p->cyl),
- p->end_head,
- sector(p->end_sector),
- cylinder(p->end_sector, p->end_cyl),
- get_start_sect(p),
- get_nr_sects(p),
- p->sys_ind
- );
- if (p->sys_ind)
- check_consistency(p, i);
- }
- }
- }
- #endif
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static void
- fill_bounds(sector_t *first, sector_t *last)
- {
- unsigned i;
- const struct pte *pe = &ptes[0];
- const struct partition *p;
- for (i = 0; i < g_partitions; pe++,i++) {
- p = pe->part_table;
- if (!p->sys_ind || IS_EXTENDED(p->sys_ind)) {
- first[i] = 0xffffffff;
- last[i] = 0;
- } else {
- first[i] = get_partition_start_from_dev_start(pe);
- last[i] = first[i] + get_nr_sects(p) - 1;
- }
- }
- }
- static void
- check(int n, unsigned h, unsigned s, unsigned c, sector_t start)
- {
- sector_t total, real_s, real_c;
- real_s = sector(s) - 1;
- real_c = cylinder(s, c);
- total = (real_c * g_sectors + real_s) * g_heads + h;
- if (!total)
- printf("Partition %u contains sector 0\n", n);
- if (h >= g_heads)
- printf("Partition %u: head %u greater than maximum %u\n",
- n, h + 1, g_heads);
- if (real_s >= g_sectors)
- printf("Partition %u: sector %u greater than "
- "maximum %u\n", n, s, g_sectors);
- if (real_c >= g_cylinders)
- printf("Partition %u: cylinder %"SECT_FMT"u greater than "
- "maximum %u\n", n, real_c + 1, g_cylinders);
- if (g_cylinders <= 1024 && start != total)
- printf("Partition %u: previous sectors %"SECT_FMT"u disagrees with "
- "total %"SECT_FMT"u\n", n, start, total);
- }
- static void
- verify(void)
- {
- int i, j;
- sector_t total = 1;
- sector_t chs_size;
- sector_t first[g_partitions], last[g_partitions];
- struct partition *p;
- if (warn_geometry())
- return;
- if (LABEL_IS_SUN) {
- verify_sun();
- return;
- }
- if (LABEL_IS_SGI) {
- verify_sgi(1);
- return;
- }
- fill_bounds(first, last);
- for (i = 0; i < g_partitions; i++) {
- struct pte *pe = &ptes[i];
- p = pe->part_table;
- if (p->sys_ind && !IS_EXTENDED(p->sys_ind)) {
- check_consistency(p, i);
- if (get_partition_start_from_dev_start(pe) < first[i])
- printf("Warning: bad start-of-data in "
- "partition %u\n", i + 1);
- check(i + 1, p->end_head, p->end_sector, p->end_cyl,
- last[i]);
- total += last[i] + 1 - first[i];
- for (j = 0; j < i; j++) {
- if ((first[i] >= first[j] && first[i] <= last[j])
- || ((last[i] <= last[j] && last[i] >= first[j]))) {
- printf("Warning: partition %u overlaps "
- "partition %u\n", j + 1, i + 1);
- total += first[i] >= first[j] ?
- first[i] : first[j];
- total -= last[i] <= last[j] ?
- last[i] : last[j];
- }
- }
- }
- }
- if (extended_offset) {
- struct pte *pex = &ptes[ext_index];
- sector_t e_last = get_start_sect(pex->part_table) +
- get_nr_sects(pex->part_table) - 1;
- for (i = 4; i < g_partitions; i++) {
- total++;
- p = ptes[i].part_table;
- if (!p->sys_ind) {
- if (i != 4 || i + 1 < g_partitions)
- printf("Warning: partition %u "
- "is empty\n", i + 1);
- } else if (first[i] < extended_offset || last[i] > e_last) {
- printf("Logical partition %u not entirely in "
- "partition %u\n", i + 1, ext_index + 1);
- }
- }
- }
- chs_size = (sector_t)g_heads * g_sectors * g_cylinders;
- if (total > chs_size)
- printf("Total allocated sectors %u"
- " greater than CHS size %"SECT_FMT"u\n",
- total, chs_size
- );
- else {
- total = chs_size - total;
- if (total != 0)
- printf("%"SECT_FMT"u unallocated sectors\n", total);
- }
- }
- static void
- add_partition(int n, int sys)
- {
- char mesg[256];
- int i, num_read = 0;
- struct partition *p = ptes[n].part_table;
- struct partition *q = ptes[ext_index].part_table;
- sector_t limit, temp;
- sector_t start, stop = 0;
- sector_t first[g_partitions], last[g_partitions];
- if (p && p->sys_ind) {
- printf(msg_part_already_defined, n + 1);
- return;
- }
- fill_bounds(first, last);
- if (n < 4) {
- start = sector_offset;
- if (display_in_cyl_units || !total_number_of_sectors)
- limit = (sector_t) g_heads * g_sectors * g_cylinders - 1;
- else
- limit = total_number_of_sectors - 1;
- if (extended_offset) {
- first[ext_index] = extended_offset;
- last[ext_index] = get_start_sect(q) +
- get_nr_sects(q) - 1;
- }
- } else {
- start = extended_offset + sector_offset;
- limit = get_start_sect(q) + get_nr_sects(q) - 1;
- }
- if (display_in_cyl_units)
- for (i = 0; i < g_partitions; i++)
- first[i] = (cround(first[i]) - 1) * units_per_sector;
- snprintf(mesg, sizeof(mesg), "First %s", str_units());
- do {
- temp = start;
- for (i = 0; i < g_partitions; i++) {
- int lastplusoff;
- if (start == ptes[i].offset_from_dev_start)
- start += sector_offset;
- lastplusoff = last[i] + ((n < 4) ? 0 : sector_offset);
- if (start >= first[i] && start <= lastplusoff)
- start = lastplusoff + 1;
- }
- if (start > limit)
- break;
- if (start >= temp+units_per_sector && num_read) {
- printf("Sector %"SECT_FMT"u is already allocated\n", temp);
- temp = start;
- num_read = 0;
- }
- if (!num_read && start == temp) {
- sector_t saved_start;
- saved_start = start;
- start = read_int(cround(saved_start), cround(saved_start), cround(limit), 0, mesg);
- if (display_in_cyl_units) {
- start = (start - 1) * units_per_sector;
- if (start < saved_start)
- start = saved_start;
- }
- num_read = 1;
- }
- } while (start != temp || !num_read);
- if (n > 4) {
- struct pte *pe = &ptes[n];
- pe->offset_from_dev_start = start - sector_offset;
- if (pe->offset_from_dev_start == extended_offset) {
- pe->offset_from_dev_start++;
- if (sector_offset == 1)
- start++;
- }
- }
- for (i = 0; i < g_partitions; i++) {
- struct pte *pe = &ptes[i];
- if (start < pe->offset_from_dev_start && limit >= pe->offset_from_dev_start)
- limit = pe->offset_from_dev_start - 1;
- if (start < first[i] && limit >= first[i])
- limit = first[i] - 1;
- }
- if (start > limit) {
- puts("No free sectors available");
- if (n > 4)
- g_partitions--;
- return;
- }
- if (cround(start) == cround(limit)) {
- stop = limit;
- } else {
- snprintf(mesg, sizeof(mesg),
- "Last %s or +size{,K,M,G,T}",
- str_units()
- );
- stop = read_int(cround(start), cround(limit), cround(limit), cround(start), mesg);
- if (display_in_cyl_units) {
- stop = stop * units_per_sector - 1;
- if (stop >limit)
- stop = limit;
- }
- }
- set_partition(n, 0, start, stop, sys);
- if (n > 4)
- set_partition(n - 1, 1, ptes[n].offset_from_dev_start, stop, EXTENDED);
- if (IS_EXTENDED(sys)) {
- struct pte *pe4 = &ptes[4];
- struct pte *pen = &ptes[n];
- ext_index = n;
- pen->ext_pointer = p;
- pe4->offset_from_dev_start = extended_offset = start;
- pe4->sectorbuffer = xzalloc(sector_size);
- pe4->part_table = pt_offset(pe4->sectorbuffer, 0);
- pe4->ext_pointer = pe4->part_table + 1;
- pe4->changed = 1;
- g_partitions = 5;
- }
- }
- static void
- add_logical(void)
- {
- if (g_partitions > 5 || ptes[4].part_table->sys_ind) {
- struct pte *pe = &ptes[g_partitions];
- pe->sectorbuffer = xzalloc(sector_size);
- pe->part_table = pt_offset(pe->sectorbuffer, 0);
- pe->ext_pointer = pe->part_table + 1;
- pe->offset_from_dev_start = 0;
- pe->changed = 1;
- g_partitions++;
- }
- add_partition(g_partitions - 1, LINUX_NATIVE);
- }
- static void
- new_partition(void)
- {
- int i, free_primary = 0;
- if (warn_geometry())
- return;
- if (LABEL_IS_SUN) {
- add_sun_partition(get_partition(0, g_partitions), LINUX_NATIVE);
- return;
- }
- if (LABEL_IS_SGI) {
- sgi_add_partition(get_partition(0, g_partitions), LINUX_NATIVE);
- return;
- }
- if (LABEL_IS_AIX) {
- puts("Sorry - this fdisk cannot handle AIX disk labels.\n"
- "If you want to add DOS-type partitions, create a new empty DOS partition\n"
- "table first (use 'o'). This will destroy the present disk contents.");
- return;
- }
- for (i = 0; i < 4; i++)
- free_primary += !ptes[i].part_table->sys_ind;
- if (!free_primary && g_partitions >= MAXIMUM_PARTS) {
- puts("The maximum number of partitions has been created");
- return;
- }
- if (!free_primary) {
- if (extended_offset)
- add_logical();
- else
- puts("You must delete some partition and add "
- "an extended partition first");
- } else {
- char c, line[80];
- snprintf(line, sizeof(line),
- "Partition type\n"
- " p primary partition (1-4)\n"
- " %s\n",
- (extended_offset ?
- "l logical (5 or over)" : "e extended"));
- while (1) {
- c = read_nonempty(line);
- c |= 0x20;
- if (c == 'p') {
- i = get_nonexisting_partition();
- if (i >= 0)
- add_partition(i, LINUX_NATIVE);
- return;
- }
- if (c == 'l' && extended_offset) {
- add_logical();
- return;
- }
- if (c == 'e' && !extended_offset) {
- i = get_nonexisting_partition();
- if (i >= 0)
- add_partition(i, EXTENDED);
- return;
- }
- printf("Invalid partition number "
- "for type '%c'\n", c);
- }
- }
- }
- static void
- reread_partition_table(int leave)
- {
- int i;
- puts("Calling ioctl() to re-read partition table");
- sync();
-
- sleep1();
- i = ioctl_or_perror(dev_fd, BLKRRPART, NULL,
- "WARNING: rereading partition table "
- "failed, kernel still uses old table");
- #if 0
- if (dos_changed)
- puts(
- "\nWARNING: If you have created or modified any DOS 6.x\n"
- "partitions, please see the fdisk manual page for additional\n"
- "information");
- #endif
- if (leave) {
- if (ENABLE_FEATURE_CLEAN_UP)
- close_dev_fd();
- exit(i != 0);
- }
- }
- static void
- write_table(void)
- {
- int i;
- if (LABEL_IS_DOS) {
- for (i = 0; i < 3; i++)
- if (ptes[i].changed)
- ptes[3].changed = 1;
- for (i = 3; i < g_partitions; i++) {
- struct pte *pe = &ptes[i];
- if (pe->changed) {
- write_part_table_flag(pe->sectorbuffer);
- write_sector(pe->offset_from_dev_start, pe->sectorbuffer);
- }
- }
- }
- else if (LABEL_IS_SGI) {
-
- sgi_write_table();
- }
- else if (LABEL_IS_SUN) {
- for (i = 0; i < 8; i++) {
- if (ptes[i].changed) {
- sun_write_table();
- break;
- }
- }
- }
- puts("The partition table has been altered.");
- reread_partition_table(1);
- }
- #endif
- #if ENABLE_FEATURE_FDISK_ADVANCED
- #define MAX_PER_LINE 16
- static void
- print_buffer(char *pbuffer)
- {
- int i,l;
- for (i = 0, l = 0; i < sector_size; i++, l++) {
- if (l == 0)
- printf("0x%03X:", i);
- printf(" %02X", (unsigned char) pbuffer[i]);
- if (l == MAX_PER_LINE - 1) {
- bb_putchar('\n');
- l = -1;
- }
- }
- if (l > 0)
- bb_putchar('\n');
- bb_putchar('\n');
- }
- static void
- print_raw(void)
- {
- int i;
- printf("Device: %s\n", disk_device);
- if (LABEL_IS_SGI || LABEL_IS_SUN)
- print_buffer(MBRbuffer);
- else {
- for (i = 3; i < g_partitions; i++)
- print_buffer(ptes[i].sectorbuffer);
- }
- }
- static void
- move_begin(unsigned i)
- {
- struct pte *pe = &ptes[i];
- struct partition *p = pe->part_table;
- sector_t new, first, nr_sects;
- if (warn_geometry())
- return;
- nr_sects = get_nr_sects(p);
- if (!p->sys_ind || !nr_sects || IS_EXTENDED(p->sys_ind)) {
- printf("Partition %u has no data area\n", i + 1);
- return;
- }
- first = get_partition_start_from_dev_start(pe);
- new = read_int(0 , first, first + nr_sects - 1, first, "New beginning of data");
- if (new != first) {
- sector_t new_relative = new - pe->offset_from_dev_start;
- nr_sects += (get_start_sect(p) - new_relative);
- set_start_sect(p, new_relative);
- set_nr_sects(p, nr_sects);
- read_nonempty("Recalculate C/H/S values? (Y/N): ");
- if ((line_ptr[0] | 0x20) == 'y')
- set_hsc_start_end(p, new, new + nr_sects - 1);
- pe->changed = 1;
- }
- }
- static void
- xselect(void)
- {
- char c;
- while (1) {
- bb_putchar('\n');
- c = 0x20 | read_nonempty("Expert command (m for help): ");
- switch (c) {
- case 'a':
- if (LABEL_IS_SUN)
- sun_set_alt_cyl();
- break;
- case 'b':
- if (LABEL_IS_DOS)
- move_begin(get_partition(0, g_partitions));
- break;
- case 'c':
- user_cylinders = g_cylinders =
- read_int(1, g_cylinders, 1048576, 0,
- "Number of cylinders");
- if (LABEL_IS_SUN)
- sun_set_ncyl(g_cylinders);
- if (LABEL_IS_DOS)
- warn_cylinders();
- break;
- case 'd':
- print_raw();
- break;
- case 'e':
- if (LABEL_IS_SGI)
- sgi_set_xcyl();
- else if (LABEL_IS_SUN)
- sun_set_xcyl();
- else if (LABEL_IS_DOS)
- x_list_table(1);
- break;
- case 'f':
- if (LABEL_IS_DOS)
- fix_partition_table_order();
- break;
- case 'g':
- #if ENABLE_FEATURE_SGI_LABEL
- create_sgilabel();
- #endif
- break;
- case 'h':
- user_heads = g_heads = read_int(1, g_heads, 256, 0, "Number of heads");
- update_units();
- break;
- case 'i':
- if (LABEL_IS_SUN)
- sun_set_ilfact();
- break;
- case 'o':
- if (LABEL_IS_SUN)
- sun_set_rspeed();
- break;
- case 'p':
- if (LABEL_IS_SUN)
- list_table(1);
- else
- x_list_table(0);
- break;
- case 'q':
- if (ENABLE_FEATURE_CLEAN_UP)
- close_dev_fd();
- bb_putchar('\n');
- exit_SUCCESS();
- case 'r':
- return;
- case 's':
- user_sectors = g_sectors = read_int(1, g_sectors, 63, 0, "Number of sectors");
- if (dos_compatible_flag) {
- sector_offset = g_sectors;
- puts("Warning: setting sector offset for DOS "
- "compatibility");
- }
- update_units();
- break;
- case 'v':
- verify();
- break;
- case 'w':
- write_table();
- break;
- case 'y':
- if (LABEL_IS_SUN)
- sun_set_pcylcount();
- break;
- default:
- xmenu();
- }
- }
- }
- #endif
- static int
- is_ide_cdrom_or_tape(const char *device)
- {
- FILE *procf;
- char buf[100];
- struct stat statbuf;
- int is_ide = 0;
-
- if (!is_prefixed_with(device, "/dev/hd"))
- return 0;
- snprintf(buf, sizeof(buf), "/proc/ide/%s/media", device+5);
- procf = fopen_for_read(buf);
- if (procf != NULL && fgets(buf, sizeof(buf), procf))
- is_ide = (is_prefixed_with(buf, "cdrom") ||
- is_prefixed_with(buf, "tape"));
- else
-
- if (stat(device, &statbuf) == 0)
- is_ide = ((statbuf.st_mode & 0222) == 0);
- if (procf)
- fclose(procf);
- return is_ide;
- }
- static void
- open_list_and_close(const char *device, int user_specified)
- {
- int gb;
- disk_device = device;
- if (setjmp(listingbuf))
- return;
- if (!user_specified)
- if (is_ide_cdrom_or_tape(device))
- return;
-
- errno = 0;
- gb = get_boot(TRY_ONLY);
- if (gb > 0) {
-
- if (user_specified || errno == EACCES)
- bb_perror_msg("can't open '%s'", device);
- return;
- }
- if (gb < 0) {
- list_disk_geometry();
- if (LABEL_IS_AIX)
- goto ret;
- #if ENABLE_FEATURE_OSF_LABEL
- if (bsd_trydev(device) < 0)
- #endif
- printf("Disk %s doesn't contain a valid "
- "partition table\n", device);
- } else {
- list_table(0);
- #if ENABLE_FEATURE_FDISK_WRITABLE
- if (!LABEL_IS_SUN && g_partitions > 4) {
- delete_partition(ext_index);
- }
- #endif
- }
- ret:
- close_dev_fd();
- }
- static int is_whole_disk(const char *disk)
- {
- unsigned len;
- int fd = open(disk, O_RDONLY);
- if (fd != -1) {
- struct hd_geometry geometry;
- int err = ioctl(fd, HDIO_GETGEO, &geometry);
- close(fd);
- if (!err)
- return (geometry.start == 0);
- }
-
-
- len = strlen(disk);
- if (len != 0 && isdigit(disk[len - 1]))
- return 0;
- return 1;
- }
- static void
- list_devs_in_proc_partititons(void)
- {
- FILE *procpt;
- char line[100], ptname[100], devname[120];
- int ma, mi, sz;
- procpt = fopen_or_warn("/proc/partitions", "r");
- while (fgets(line, sizeof(line), procpt)) {
- if (sscanf(line, " %u %u %u %[^\n ]",
- &ma, &mi, &sz, ptname) != 4)
- continue;
- sprintf(devname, "/dev/%s", ptname);
- if (is_whole_disk(devname))
- open_list_and_close(devname, 0);
- }
- #if ENABLE_FEATURE_CLEAN_UP
- fclose(procpt);
- #endif
- }
- #if ENABLE_FEATURE_FDISK_WRITABLE
- static void
- unknown_command(int c)
- {
- printf("%c: unknown command\n", c);
- }
- #endif
- int fdisk_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
- int fdisk_main(int argc UNUSED_PARAM, char **argv)
- {
- unsigned opt;
-
- INIT_G();
- close_dev_fd();
- opt = getopt32(argv, "b:+C:+H:+lS:+u" IF_FEATURE_FDISK_BLKSIZE("s"),
- §or_size, &user_cylinders, &user_heads, &user_sectors);
- argv += optind;
- if (opt & OPT_b) {
-
- if (sector_size < 512
- || sector_size > 0x10000
- || (sector_size & (sector_size-1))
- ) {
- bb_show_usage();
- }
- sector_offset = 2;
- user_set_sector_size = 1;
- }
- if (user_heads <= 0 || user_heads >= 256)
- user_heads = 0;
- if (user_sectors <= 0 || user_sectors >= 64)
- user_sectors = 0;
- if (opt & OPT_u)
- display_in_cyl_units = 0;
- #if ENABLE_FEATURE_FDISK_WRITABLE
- if (opt & OPT_l) {
- nowarn = 1;
- #endif
- if (*argv) {
- listing = 1;
- do {
- open_list_and_close(*argv, 1);
- } while (*++argv);
- } else {
-
-
- list_devs_in_proc_partititons();
- }
- return 0;
- #if ENABLE_FEATURE_FDISK_WRITABLE
- }
- #endif
- #if ENABLE_FEATURE_FDISK_BLKSIZE
- if (opt & OPT_s) {
- int j;
- nowarn = 1;
- if (!argv[0])
- bb_show_usage();
- for (j = 0; argv[j]; j++) {
- unsigned long long size;
- fd = xopen(argv[j], O_RDONLY);
- size = bb_BLKGETSIZE_sectors(fd) / 2;
- close(fd);
- if (argv[1])
- printf("%llu\n", size);
- else
- printf("%s: %llu\n", argv[j], size);
- }
- return 0;
- }
- #endif
- #if ENABLE_FEATURE_FDISK_WRITABLE
- if (!argv[0] || argv[1])
- bb_show_usage();
- disk_device = argv[0];
- get_boot(OPEN_MAIN);
- if (LABEL_IS_OSF) {
-
- printf("Detected an OSF/1 disklabel on %s, entering "
- "disklabel mode\n", disk_device);
- bsd_select();
-
- current_label_type = LABEL_DOS;
-
- }
- while (1) {
- int c;
- bb_putchar('\n');
- c = 0x20 | read_nonempty("Command (m for help): ");
- switch (c) {
- case 'a':
- if (LABEL_IS_DOS)
- toggle_active(get_partition(1, g_partitions));
- else if (LABEL_IS_SUN)
- toggle_sunflags(get_partition(1, g_partitions),
- 0x01);
- else if (LABEL_IS_SGI)
- sgi_set_bootpartition(
- get_partition(1, g_partitions));
- else
- unknown_command(c);
- break;
- case 'b':
- if (LABEL_IS_SGI) {
- printf("\nThe current boot file is: %s\n",
- sgi_get_bootfile());
- if (read_maybe_empty("Please enter the name of the "
- "new boot file: ") == '\n')
- puts("Boot file unchanged");
- else
- sgi_set_bootfile(line_ptr);
- }
- #if ENABLE_FEATURE_OSF_LABEL
- else
- bsd_select();
- #endif
- break;
- case 'c':
- if (LABEL_IS_DOS)
- toggle_dos_compatibility_flag();
- else if (LABEL_IS_SUN)
- toggle_sunflags(get_partition(1, g_partitions),
- 0x10);
- else if (LABEL_IS_SGI)
- sgi_set_swappartition(
- get_partition(1, g_partitions));
- else
- unknown_command(c);
- break;
- case 'd':
- {
- int j;
-
- if (!LABEL_IS_SGI) {
- j = get_existing_partition(1, g_partitions);
- } else {
- j = get_partition(1, g_partitions);
- }
- if (j >= 0)
- delete_partition(j);
- }
- break;
- case 'i':
- if (LABEL_IS_SGI)
- create_sgiinfo();
- else
- unknown_command(c);
- case 'l':
- list_types(get_sys_types());
- break;
- case 'm':
- menu();
- break;
- case 'n':
- new_partition();
- break;
- case 'o':
- create_doslabel();
- break;
- case 'p':
- list_table(0);
- break;
- case 'q':
- if (ENABLE_FEATURE_CLEAN_UP)
- close_dev_fd();
- bb_putchar('\n');
- return 0;
- case 's':
- #if ENABLE_FEATURE_SUN_LABEL
- create_sunlabel();
- #endif
- break;
- case 't':
- change_sysid();
- break;
- case 'u':
- change_units();
- break;
- case 'v':
- verify();
- break;
- case 'w':
- write_table();
- break;
- #if ENABLE_FEATURE_FDISK_ADVANCED
- case 'x':
- if (LABEL_IS_SGI) {
- puts("\n\tSorry, no experts menu for SGI "
- "partition tables available\n");
- } else
- xselect();
- break;
- #endif
- default:
- unknown_command(c);
- menu();
- }
- }
- return 0;
- #endif
- }
|