ddftw.c

/*
 * Copyright (C) 2011 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>

#include "ddftw.h"
#include "common.h"
#include "extra-functions.h"
#include "bootloader.h"
#include "backstore.h"

// get locations from our device.info
void getLocations()
{
	FILE *fp;
	int tmpInt;
	char tmpText[50];
	if (strcmp(get_fstype(),"mtd") == 0)
	{
		fp = __popen("cat /proc/mtd", "r");
	} else if (strcmp(get_fstype(),"emmc") == 0) {
		fp = __popen("cat /proc/emmc", "r");
	}
	if (fp == NULL)
	{
		LOGI("\n=> Could not determine flash type!\n");
	} else {
		while (fscanf(fp,"%s %0x %*s %*c%s",tmp.dev,&tmp.sze,tmp.mnt) != EOF) // populate structs from proc
		{
			if (strcmp(tmp.dev,"dev:") != 0)
			{
				tmp.dev[strlen(tmp.dev)-1] = '\0';
				tmp.mnt[strlen(tmp.mnt)-1] = '\0';
				if (sscanf(tmp.dev,"mtd%d",&tmpInt) == 1)
				{
					sprintf(tmpText,"%smtdblock%d",tw_block,tmpInt);
					strcpy(tmp.blk,tmpText);
					sprintf(tmpText,"%s%s",tw_mtd,tmp.dev);
					strcpy(tmp.dev,tmpText);
					strcpy(tmp.fst,"mtd");
				} else {
					sprintf(tmpText,"%s%s",tw_block,tmp.dev);
					strcpy(tmp.dev,tmpText);
					strcpy(tmp.blk,tmp.dev);
					strcpy(tmp.fst,"emmc");
				}
			}
			if (strcmp(tmp.mnt,"system") == 0) { // read in system line
				strcpy(sys.mnt,tmp.mnt);
				strcpy(sys.dev,tmp.dev);
				strcpy(sys.blk,tmp.blk);
			}
			if (strcmp(tmp.mnt,"userdata") == 0) {
				strcpy(dat.mnt,"data");
				strcpy(dat.dev,tmp.dev);
				strcpy(dat.blk,tmp.blk);
			}
			if (strcmp(tmp.mnt,"boot") == 0) {
				strcpy(boo.mnt,tmp.mnt);
				strcpy(boo.dev,tmp.dev);
				strcpy(boo.blk,tmp.blk);
				strcpy(boo.fst,tmp.fst);
				boo.sze = tmp.sze;
			}
			if (strcmp(tmp.mnt,"recovery") == 0) {
				strcpy(rec.mnt,tmp.mnt);
				strcpy(rec.dev,tmp.dev);
				strcpy(rec.blk,tmp.blk);
				strcpy(rec.fst,tmp.fst);
				rec.sze = tmp.sze;
			}
			if (strcmp(tmp.mnt,"cache") == 0) {
				strcpy(cac.mnt,tmp.mnt);
				strcpy(cac.dev,tmp.dev);
				strcpy(cac.blk,tmp.blk);
			}
			if (strcmp(tmp.mnt,"wimax") == 0) {
				strcpy(wim.mnt,tmp.mnt);
				strcpy(wim.dev,tmp.dev);
				strcpy(wim.blk,tmp.blk);
				strcpy(wim.fst,tmp.fst);
				wim.sze = tmp.sze;
			}
			if (strcmp(tmp.mnt,"efs") == 0) {
				strcpy(wim.mnt,tmp.mnt);
				strcpy(wim.dev,tmp.dev);
				strcpy(wim.blk,tmp.blk);
				strcpy(wim.fst,"yaffs2");
				wim.sze = tmp.sze;
			}
		}
		pclose(fp);
		readRecFstab();
	}
	get_device_id();
}

void readRecFstab()
{
	FILE *fp;
	char tmpText[255];
	fp = fopen("/etc/recovery.fstab", "r");
	if (fp == NULL) {
		LOGI("=> Can not open /etc/recovery.fstab.\n");
	} else {
		fgets(tmpText, 255, fp);
		fgets(tmpText, 255, fp);
		while (fgets(tmpText,255,fp) != NULL)
		{
			sscanf(tmpText,"%*c%s %s %s %s",tmp.mnt,tmp.fst,tmp.blk,tmp.dev); // populate structs from recovery.fstab
			if (strcmp(tmp.mnt,"system") == 0)
			{
				strcpy(sys.fst,tmp.fst);
				if (strcmp(sys.mnt,"system") != 0)
				{
					strcpy(sys.mnt,tmp.mnt);
					strcpy(sys.blk,tmp.blk);
					strcpy(sys.dev,tmp.blk);
				}
			}
			if (strcmp(tmp.mnt,"data") == 0)
			{
				strcpy(dat.fst,tmp.fst);
				if (strcmp(dat.mnt,"data") != 0)
				{
					strcpy(dat.mnt,tmp.mnt);
					strcpy(dat.blk,tmp.blk);
					strcpy(dat.dev,tmp.blk);
				}
			}
			if (strcmp(tmp.mnt,"cache") == 0)
			{
				strcpy(cac.fst,tmp.fst);
				if (strcmp(cac.mnt,"cache") != 0)
				{
					strcpy(cac.mnt,tmp.mnt);
					strcpy(cac.blk,tmp.blk);
					strcpy(cac.dev,tmp.blk);
				}
			}
			if (strcmp(tmp.mnt,"sdcard") == 0)
			{
				strcpy(sdc.fst,tmp.fst);
				if (strcmp(sdc.mnt,"sdcard") != 0)
				{
					strcpy(sdc.mnt,tmp.mnt);
					strcpy(sdc.blk,tmp.blk);
					strcpy(sdc.dev,tmp.dev);
				}
			}
			if (strcmp(tmp.mnt,"sd-ext") == 0)
			{
				strcpy(sde.mnt,tmp.mnt);
				strcpy(sde.fst,tmp.fst);
				strcpy(sde.blk,tmp.blk);
				strcpy(sde.dev,tmp.dev);
			}
		}
	}
	fclose(fp);
	strcpy(ase.dev,"/sdcard/.android_secure"); // android secure stuff
	strcpy(ase.blk,ase.dev);
	strcpy(ase.mnt,".android_secure");
	strcpy(ase.fst,"vfat");
	if (strcmp(sde.mnt,"sd-ext") != 0) // sd-ext stuff
	{
		int tmpInt;
		char tmpBase[50];
		char tmpWildCard[50];
		strcpy(tmpBase,sdc.blk);
		tmpBase[strlen(tmpBase)-1] = '\0';
		sprintf(tmpWildCard,"%s%%d",tmpBase);
		sscanf(sdc.blk,tmpWildCard,&tmpInt); // sdcard block used as sd-ext base
		sprintf(sde.blk,"%s%d",tmpBase,tmpInt+1);
	}
	LOGI("=> Let's update filesystem types.\n");
	verifyFst(); // use blkid to REALLY REALLY determine partition filesystem type, blkid doesn't like mtd
	LOGI("=> And update our fstab also.\n\n");
	createFstab(); // used for our busybox mount command
}

// write fstab so we can mount in adb shell
void createFstab()
{
	FILE *fp;
	struct stat st;
	fp = fopen("/etc/fstab", "w");
	if (fp == NULL) {
		LOGI("=> Can not open /etc/fstab.\n");
	} else 
	{
		char tmpString[255];
		sprintf(tmpString,"%s /%s %s rw\n",sys.blk,sys.mnt,sys.fst);
		fputs(tmpString, fp);
		sprintf(tmpString,"%s /%s %s rw\n",dat.blk,dat.mnt,dat.fst);
		fputs(tmpString, fp);
		sprintf(tmpString,"%s /%s %s rw\n",cac.blk,cac.mnt,cac.fst);
		fputs(tmpString, fp);
		sprintf(tmpString,"%s /%s %s rw\n",sdc.blk,sdc.mnt,sdc.fst);
		fputs(tmpString, fp);
		if (stat(sde.blk,&st) == 0) {
			strcpy(sde.mnt,"sd-ext");
			strcpy(sde.dev,sde.blk);
			sprintf(tmpString,"%s /%s %s rw\n",sde.blk,sde.mnt,sde.fst);
			fputs(tmpString, fp);
			if (stat("/sd-ext",&st) != 0) {
				if(mkdir("/sd-ext",0777) == -1) {
					LOGI("=> Can not create /sd-ext folder.\n");
				} else {
					LOGI("=> Created /sd-ext folder.\n");
				}
			}
		}
		if (strcmp(wim.mnt,"efs") == 0) {
			sprintf(tmpString,"%s /%s %s rw\n",wim.blk,wim.mnt,wim.fst);
			fputs(tmpString, fp);
			if (stat("/efs",&st) != 0) {
				if(mkdir("/efs",0777) == -1) {
					LOGI("=> Can not create /efs folder.\n");
				} else {
					LOGI("=> Created /efs folder.\n");
				}
			}
		}
	}
	fclose(fp);
}

void verifyFst()
{
	FILE *fp;
	char blkOutput[100];
	char blk[50];
	char arg2[50];
	char arg3[50];
	char arg4[50];
	char fst[10];

    // Do not run this routine on MTD devices. The blkid command hangs.
    if (strcmp(get_fstype(),"mtd") == 0)
        return;

	fp = __popen("blkid","r");
	while (fgets(blkOutput,sizeof(blkOutput),fp) != NULL) {
		if (sscanf(blkOutput,"%s %s %s TYPE=\"%s",blk,arg2,arg3,arg4) == 4) {
			arg4[strlen(arg4)-1] = '\0';
			strcpy(fst,arg4);
		} else if (sscanf(blkOutput,"%s %s TYPE=\"%s",blk,arg2,arg3) == 3) {
			arg3[strlen(arg3)-1] = '\0';
			strcpy(fst,arg3);
		}
		blk[strlen(blk)-1] = '\0';
		if (strcmp(blk,sys.blk) == 0) {
			strcpy(sys.fst,fst);
		} else if (strcmp(blk,dat.blk) == 0) {
			strcpy(dat.fst,fst);
		} else if (strcmp(blk,cac.blk) == 0) {
			strcpy(cac.fst,fst);
		} else if (strcmp(blk,sde.blk) == 0) {
			strcpy(sde.fst,fst);
		}
	}
	__pclose(fp);
}