/*
 * overclock_routerstation.c
 *
 * Copyright (C) 2008 Sebastian Gottschall <gottschall@dd-wrt.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *
 * usage:
 * nvram set cpuclk=800
 * startservice overclock
 * 
 * valid cpuclk values are 200, 300 ,333, 400, 600, 680, 720, 800
 */
#include <unistd.h>
#include <sys/types.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <fcntl.h>



int overclock(FILE *out,char *freq,int value)
{
fseek(out,0xd3,SEEK_SET);
int val = getc(out);
if (val==value)
	{
	fprintf(stderr,"board already clocked to %sMhz\n",freq);
	return -1;
	}
fseek(out,0xd3,SEEK_SET);
putc(value,out);
return 0;
}

int main(int argc, char *argv[])
{
    long len;
    long i;

    FILE *in = fopen( "/dev/mtdblock0", "rb" );
    fseek(in,0,SEEK_END);
    len = ftell(in);
    rewind(in);
    char check[8];
    char check2[8];
    char values[8]={0x24,0x08,0x00,0xaa,0x15,0x09,0x00,0x04};
    fseek(in,0xc0,SEEK_SET);
    fread(check,1,8,in);
    fseek(in,0xc4,SEEK_SET);
    fread(check2,1,8,in);
    int ret1=0xff;
    int ret2=0xff;
    if ((ret1=memcmp(check,values,8)) && (ret2=memcmp(check2,values,8)))
	{
	fprintf(stderr,"no compatible routerstation bootloader found\n");
	fclose(in);
	return;
	}
    if (!ret1)
	fprintf(stderr,"bootloader rev1 found\n");
    if (!ret2)
	fprintf(stderr,"bootloader rev2 found\n");
    FILE *out = fopen( "/tmp/boot", "w+b" );
    rewind(in);
    for (i=0;i<len;i++)
	putc(getc(in),out);
    fclose(in);
    int ret=1;
    if (!strcmp(argv[1], "200")){
	ret=overclock(out,"200",0x1);
    }
    if (!strcmp(argv[1], "300")){
	ret=overclock(out,"300",0x2);
    }
    if (!strcmp(argv[1], "333")){
	ret=overclock(out,"333",0x3);
    }
    if (!strcmp(argv[1], "400")){
    ret=overclock(out,"400",0x6);
	}
    if (!strcmp(argv[1], "600")){
	ret=overclock(out,"600",0x7);
    }
    if (!strcmp(argv[1], "680")){  //special ubiquiti setting with different ddram clock settings
	ret=overclock(out,"680",0xa); 
    }
    if (!strcmp(argv[1], "720")){
	ret=overclock(out,"720",0x1e);
    }
    if (!strcmp(argv[1], "800")){
	ret=overclock(out,"800",0x1f);
    }
fclose(out);
if (!ret)
    {
    fprintf(stderr,"write new bootloader\n");
    system("mtd -f write /tmp/boot /dev/mtd0");
    fprintf(stderr,"reboot\n");
    }
if (ret==1)
    {
    fprintf(stderr,"valid cpuclk values are 200, 300 ,333, 400, 600, 680, 720, 800\n");
    }
}