android/cLK
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
cLK/lk/lib/vptable/vptable.c
cmhtcleo 9a76f8e4e9 added Rick_1995 code in
2011-11-06 23:17:49 +00:00

405 lines
8.0 KiB
C
Raw Permalink Blame History

/*
Simple routines for reading and writing partition table
Copyright (c) 2011 - Danijel Posilovic - dan1j3l
*/
#include <debug.h>
#include <arch/arm.h>
#include <dev/udc.h>
#include <string.h>
#include <kernel/thread.h>
#include <arch/ops.h>
#include <dev/flash.h>
#include <lib/ptable.h>
#include <dev/keys.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <lib/vptable.h>
static const int MISC_PAGES = 3;
static const int MISC_COMMAND_PAGE = 0;
static char buf[4096];
extern struct ptable *flash_get_vptable(void);
int read_vptable(struct vpartitions *out){
struct ptentry *ptn;
struct ptable *ptable;
unsigned offset = 0;
unsigned pagesize = flash_page_size();
ptable = flash_get_vptable();
if (ptable == NULL) {
dprintf(CRITICAL, "ERROR: VPTABLE not found!!!\n");
return -1;
}
ptn = ptable_find(ptable, "vptable");
if (ptn == NULL) {
dprintf(CRITICAL, "ERROR: No vptable partition!!!\n");
return -1;
}
offset += (pagesize * MISC_COMMAND_PAGE);
if (flash_read(ptn, offset, buf, pagesize)) {
//dprintf(CRITICAL, "ERROR: Cannot read vpt header\n");
//return -1;
}
memcpy(out, buf, sizeof(*out));
return 0;
}
int write_vptable(const struct vpartitions *in){
struct ptentry *ptn;
struct ptable *ptable;
unsigned offset = 0;
unsigned pagesize = flash_page_size();
unsigned n = 0;
ptable = flash_get_vptable();
if (ptable == NULL) {
printf(CRITICAL, "ERROR: VPTABLE table not found!!!\n");
return -1;
}
ptn = ptable_find(ptable, "vptable");
if (ptn == NULL) {
printf(CRITICAL, "ERROR: No vptable partition!!!\n");
return -1;
}
n = pagesize * (MISC_COMMAND_PAGE + 1);
//if (flash_read(ptn, offset, SCRATCH_ADDR, n)) {
// printf(CRITICAL, "ERROR: Cannot read vpt header\n");
// return -1;
//}
offset += (pagesize * MISC_COMMAND_PAGE);
offset += SCRATCH_ADDR;
memcpy((void *)offset, in, sizeof(*in));
if (flash_write(ptn, 0, (void *)SCRATCH_ADDR, n)) {
printf(CRITICAL, "ERROR: flash write fail!\n");
return -1;
}
return 0;
}
unsigned vpart_available_size(){
unsigned used =0;
for (unsigned i=0;;i++){
if ((strlen(vparts.pdef[i].name) != 0) && (vparts.pdef[i].asize == 0)){
used += vparts.pdef[i].size;
} else {
break;
}
}
return (flash_size_blk - used);
}
bool vpart_variable_exist(){
for (unsigned i=0;i < (unsigned)MAX_NUM_PART;i++){
if ((strlen(vparts.pdef[i].name) != 0) && (vparts.pdef[i].asize == 1)) return 1;
}
return 0;
}
bool vpart_partition_exist(const char* pName){
for (unsigned i=0;i < (unsigned)MAX_NUM_PART;i++){
if (!memcmp(vparts.pdef[i].name,pName,strlen(pName)))return 1;
}
return 0;
}
void vpart_resize_asize(){
for (unsigned i=0; i < (unsigned)MAX_NUM_PART;i++){
if ((strlen(vparts.pdef[i].name) != 0) && (vparts.pdef[i].asize == 1)){
vparts.pdef[i].size = vpart_available_size();
return;
}
}
}
extern int atoi ( const char * str );
void vpart_add(const char *pData){
char buff[64];
char name[32];
char* tmp_buff;
unsigned size;
strcpy(buff,pData);
tmp_buff = strtok(buff,":");
strcpy(name,tmp_buff);
tmp_buff=strtok(NULL,":");
size = atoi(tmp_buff)*blk_pmb;
if (strlen(name)==0){
//fastboot_fail("Undefined partition name ! usage: fastboot oem part-add name:size");
return;
}
if (size==0 && vpart_available_size()<8){
//fastboot_fail("No available space for variable partition!");
return;
}
if (size==0 && vpart_variable_exist()){
//fastboot_fail("Autosize partition already exist!");
return;
}
if (vpart_partition_exist(name)){
//fastboot_fail("Named partition already exist!");
return;
}
for (unsigned i=0; i< (unsigned)MAX_NUM_PART; i++){
if (strlen(vparts.pdef[i].name) == 0){
strcpy(vparts.pdef[i].name,name);
if (size==0){
vparts.pdef[i].size=vpart_available_size();
vparts.pdef[i].asize=1;
}else{
vparts.pdef[i].size = size;
vparts.pdef[i].asize=0;
}
vpart_resize_asize();
break;
}
}
}
extern int atoi ( const char * str );
void vpart_resize(const char *pData){
char buff[64];
char name[32];
char* tmp_buff;
unsigned size;
unsigned old_size=0;
unsigned tmp_available_size=0;
strcpy(buff,pData);
tmp_buff = strtok(buff,":");
strcpy(name,tmp_buff);
tmp_buff=strtok(NULL,":");
size = atoi(tmp_buff)*blk_pmb;
if (strlen(name)==0){
//fastboot_fail("Undefined partition size!");
return;
}
for (unsigned i=0; i < (unsigned)MAX_NUM_PART;i++){
if ((strlen(vparts.pdef[i].name) != 0) && (!memcmp(vparts.pdef[i].name,name,strlen(name))) && (vparts.pdef[i].asize == 0) ){
old_size = vparts.pdef[i].size;
}
}
tmp_available_size = vpart_available_size() + old_size;
if (size>tmp_available_size)
size=tmp_available_size;
if (size==0 && tmp_available_size<8){
//fastboot_fail("No more space for variable partition!");
return;
}
if (size==0 && vpart_variable_exist()){
//fastboot_fail("Unable to set to autosize, autosize partition already exist!");
return;
}
for (unsigned i=0;i<(unsigned)MAX_NUM_PART;i++){
if (!memcmp(vparts.pdef[i].name,name,strlen(name))){
if (size==0){
vparts.pdef[i].size=tmp_available_size;
vparts.pdef[i].asize=1;
}else{
vparts.pdef[i].size = size;
vparts.pdef[i].asize=0;
}
vpart_resize_asize();
break;
}
}
}
// Restruct ptable and remove empty space between partitions
void vpart_restruct(){
for (unsigned i=0;i < (unsigned)MAX_NUM_PART;i++){
if (strlen(vparts.pdef[i].name)==0){
for (unsigned j=i; j < (unsigned)MAX_NUM_PART;j++){
if (strlen(vparts.pdef[j].name) != 0){
// Copy values to empty space
strcpy(vparts.pdef[i].name,vparts.pdef[j].name);
vparts.pdef[i].size=vparts.pdef[j].size;
vparts.pdef[i].asize=vparts.pdef[j].asize;
// Mark old values dirty
strcpy(vparts.pdef[j].name,"");
vparts.pdef[j].size=0;
vparts.pdef[j].asize=0;
break;
}
}
}
}
}
// Remove partition from ptable
void vpart_del(const char *pName){
for (unsigned i=0;i < (unsigned)MAX_NUM_PART;i++){
if (!memcmp(vparts.pdef[i].name,pName,strlen(pName))){
strcpy(vparts.pdef[i].name,"");
vparts.pdef[i].size=0;
vparts.pdef[i].asize=0;
vpart_restruct();
vpart_resize_asize();
}
}
}
// clear curent layout
void vpart_clear(){
strcpy(vparts.tag,"");
for (unsigned i=0;i < (unsigned)MAX_NUM_PART;i++){
strcpy(vparts.pdef[i].name,"");
vparts.pdef[i].size=0;
vparts.pdef[i].asize=0;
}
}
// Print current ptable
void vpart_list(){
printf("\n\n============================== PARTITION TABLE ==============================\n\n");
for (unsigned i=0;;i++){
if (strlen(vparts.pdef[i].name) != 0){
printf("%i.",i+1);
//fbcon_move_right(4);
printf("%s", vparts.pdef[i].name);
//fbcon_move_right(25);
printf( "| blocks: %i",vparts.pdef[i].size);
//fbcon_move_right(50);
printf( "| size: %i Mb\n",(vparts.pdef[i].size/blk_pmb));
} else {
break;
}
}
printf("\n\n============================== PARTITION TABLE ==============================\n\n");
}
// Create default partition layout
void vpart_create_default(){
vpart_clear();
vpart_add("misc:1");
vpart_add("recovery:5");
vpart_add("boot:5");
vpart_add("system:150");
vpart_add("cache:5");
vpart_add("userdata:0");
}
void vpart_commit(){
strcpy(vparts.tag,"VPTABLE");
write_vptable(&vparts);
}
void vpart_read(){
vpart_clear();
read_vptable(&vparts);
}
void vpart_enable_extrom(){
vparts.extrom_enabled = 1;
vpart_commit();
//cprint(cRed,cWhite,"ExtROM Enabled ! \n Reboot your phone and repartition nand to use ExtROM !!!\n\n");
}
void vpart_disable_extrom(){
vparts.extrom_enabled = 0;
vpart_commit();
//cprint(cRed,cWhite,"ExtROM Disabled ! \n Reboot your phone and repartition nand to remove ExtROM !!!\n\n");
}
// Init and load ptable from nand
void init_vpart(){
vpart_read();
if (memcmp(vparts.tag,"VPTABLE",7)){ // Bad mojo ! ptable does not exist, lets create default one !!!
vparts.extrom_enabled = 0;
flash_size_blk = flash_size_blk - 191; // extrom size
vpart_create_default();
vpart_commit();
}
if (vparts.extrom_enabled==0)
flash_size_blk = flash_size_blk - 191; // extrom size
}
Reference in New Issue View Git Blame Copy Permalink