wifipineapple-openwrt/package/openwrt/libnvram/nvram_linux.c

321 lines
6.4 KiB
C

/*
* NVRAM variable manipulation (Linux user mode half)
*
* Copyright 2004, Broadcom Corporation
* All Rights Reserved.
*
* THIS SOFTWARE IS OFFERED "AS IS", AND BROADCOM GRANTS NO WARRANTIES OF ANY
* KIND, EXPRESS OR IMPLIED, BY STATUTE, COMMUNICATION OR OTHERWISE. BROADCOM
* SPECIFICALLY DISCLAIMS ANY IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A SPECIFIC PURPOSE OR NONINFRINGEMENT CONCERNING THIS SOFTWARE.
*
* $Id$
*/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <error.h>
#include <string.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/mman.h>
#include <typedefs.h>
#include <bcmnvram.h>
#include <nvram_convert.h>
#include <shutils.h>
#include <utils.h>
#define PATH_DEV_NVRAM "/dev/nvram"
/* Globals */
static int nvram_fd = -1;
static char *nvram_buf = NULL;
int check_action(void);
int file_to_buf(char *path, char *buf, int len);
int
nvram_init(void *unused)
{
if ((nvram_fd = open(PATH_DEV_NVRAM, O_RDWR)) < 0)
goto err;
/* Map kernel string buffer into user space */
if ((nvram_buf = mmap(NULL, NVRAM_SPACE, PROT_READ, MAP_SHARED, nvram_fd, 0)) == MAP_FAILED) {
close(nvram_fd);
nvram_fd = -1;
goto err;
}
return 0;
err:
perror(PATH_DEV_NVRAM);
return errno;
}
char *
nvram_get(const char *name)
{
size_t count = strlen(name) + 1;
char tmp[100], *value;
unsigned long *off = (unsigned long *) tmp;
if (nvram_fd < 0)
if (nvram_init(NULL))
return NULL;
if (count > sizeof(tmp)) {
if (!(off = malloc(count)))
return NULL;
}
/* Get offset into mmap() space */
strcpy((char *) off, name);
count = read(nvram_fd, off, count);
if (count == sizeof(unsigned long))
value = &nvram_buf[*off];
else
value = NULL;
if (count < 0)
perror(PATH_DEV_NVRAM);
if (off != (unsigned long *) tmp)
free(off);
return value;
}
int
nvram_getall(char *buf, int count)
{
int ret;
if (nvram_fd < 0)
if ((ret = nvram_init(NULL)))
return ret;
if (count == 0)
return 0;
/* Get all variables */
*buf = '\0';
ret = read(nvram_fd, buf, count);
if (ret < 0)
perror(PATH_DEV_NVRAM);
return (ret == count) ? 0 : ret;
}
static int
_nvram_set(const char *name, const char *value)
{
size_t count = strlen(name) + 1;
char tmp[100], *buf = tmp;
int ret;
if (nvram_fd < 0)
if ((ret = nvram_init(NULL)))
return ret;
/* Unset if value is NULL */
if (value)
count += strlen(value) + 1;
if (count > sizeof(tmp)) {
if (!(buf = malloc(count)))
return -ENOMEM;
}
if (value)
sprintf(buf, "%s=%s", name, value);
else
strcpy(buf, name);
ret = write(nvram_fd, buf, count);
if (ret < 0)
perror(PATH_DEV_NVRAM);
if (buf != tmp)
free(buf);
return (ret == count) ? 0 : ret;
}
int
nvram_set(const char *name, const char *value)
{
extern struct nvram_convert nvram_converts[];
struct nvram_convert *v;
int ret;
ret = _nvram_set(name, value);
for(v = nvram_converts ; v->name ; v++) {
if(!strcmp(v->name, name)){
if(strcmp(v->wl0_name,"")) _nvram_set(v->wl0_name, value);
if(strcmp(v->d11g_name,"")) _nvram_set(v->d11g_name, value);
}
}
return ret;
}
int
nvram_unset(const char *name)
{
return _nvram_set(name, NULL);
}
int
nvram_commit(void)
{
int ret;
cprintf("nvram_commit(): start\n");
if((check_action() == ACT_IDLE) ||
(check_action() == ACT_SW_RESTORE) ||
(check_action() == ACT_HW_RESTORE)){
if (nvram_fd < 0)
if ((ret = nvram_init(NULL)))
return ret;
ret = ioctl(nvram_fd, NVRAM_MAGIC, NULL);
if (ret < 0)
perror(PATH_DEV_NVRAM);
cprintf("nvram_commit(): end\n");
}
else
cprintf("nvram_commit(): nothing to do...\n");
return ret;
}
int file2nvram(char *filename, char *varname) {
FILE *fp;
int c,count;
int i=0,j=0;
char mem[10000],buf[30000];
if ( !(fp=fopen(filename,"rb") ))
return 0;
count=fread(mem,1,sizeof(mem),fp);
fclose(fp);
for (j=0;j<count;j++) {
if (i > sizeof(buf)-3 )
break;
c=mem[j];
if (c >= 32 && c <= 126 && c != '\\' && c != '~') {
buf[i++]=(unsigned char) c;
} else if (c==0) {
buf[i++]='~';
} else {
buf[i++]='\\';
sprintf(buf+i,"%02X",c);
i+=2;
}
}
if (i==0) return 0;
buf[i]=0;
//fprintf(stderr,"================ > file2nvram %s = [%s] \n",varname,buf);
nvram_set(varname,buf);
//nvram_commit(); //Barry adds for test
}
int nvram2file(char *varname, char *filename) {
FILE *fp;
int c,tmp;
int i=0,j=0;
char *buf;
char mem[10000];
if ( !(fp=fopen(filename,"wb") ))
return 0;
buf=strdup(nvram_safe_get(varname));
//fprintf(stderr,"=================> nvram2file %s = [%s] \n",varname,buf);
while ( buf[i] && j < sizeof(mem)-3 ) {
if (buf[i] == '\\') {
i++;
tmp=buf[i+2];
buf[i+2]=0;
sscanf(buf+i,"%02X",&c);
buf[i+2]=tmp;
i+=2;
mem[j]=c;j++;
} else if (buf[i] == '~') {
mem[j]=0;j++;
i++;
} else {
mem[j]=buf[i];j++;
i++;
}
}
if (j<=0) return j;
j=fwrite(mem,1,j,fp);
fclose(fp);
free(buf);
return j;
}
int
check_action(void)
{
char buf[80] = "";
if(file_to_buf(ACTION_FILE, buf, sizeof(buf))){
if(!strcmp(buf, "ACT_TFTP_UPGRADE")){
cprintf("Upgrading from tftp now, quiet exit....\n");
return ACT_TFTP_UPGRADE;
}
else if(!strcmp(buf, "ACT_WEBS_UPGRADE")){
cprintf("Upgrading from web (https) now, quiet exit....\n");
return ACT_WEBS_UPGRADE;
}
else if(!strcmp(buf, "ACT_WEB_UPGRADE")){
cprintf("Upgrading from web (http) now, quiet exit....\n");
return ACT_WEB_UPGRADE;
}
else if(!strcmp(buf, "ACT_SW_RESTORE")){
cprintf("Receive restore command from web, quiet exit....\n");
return ACT_SW_RESTORE;
}
else if(!strcmp(buf, "ACT_HW_RESTORE")){
cprintf("Receive restore commond from resetbutton, quiet exit....\n");
return ACT_HW_RESTORE;
}
}
//fprintf(stderr, "Waiting for upgrading....\n");
return ACT_IDLE;
}
int
file_to_buf(char *path, char *buf, int len)
{
FILE *fp;
memset(buf, 0 , len);
if ((fp = fopen(path, "r"))) {
fgets(buf, len, fp);
fclose(fp);
return 1;
}
return 0;
}