539 lines
13 KiB
C
539 lines
13 KiB
C
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/utsname.h>
|
|
#include <sys/mount.h>
|
|
#include <netinet/in.h>
|
|
#include <unistd.h>
|
|
#include <signal.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
|
|
/* Default path we try to mount. "%s" gets replaced by our IP address */
|
|
#define NFS_ROOT "/tftpboot/%s"
|
|
#define NFS_DEF_FILE_IO_BUFFER_SIZE 4096
|
|
#define NFS_MAXPATHLEN 1024
|
|
#define NFS_MNT_PROGRAM 100005
|
|
#define NFS_MNT_PORT 627
|
|
#define NFS_PROGRAM 100003
|
|
#define NFS_PORT 2049
|
|
#define NFS2_VERSION 2
|
|
#define NFS3_VERSION 3
|
|
#define NFS_MNT_PROGRAM 100005
|
|
#define NFS_MNT_VERSION 1
|
|
#define NFS_MNT3_VERSION 3
|
|
#define MNTPROC_MNT 1
|
|
#define MOUNTPROC3_MNT 1
|
|
#define RPC_PMAP_PROGRAM 100000
|
|
#define RPC_PMAP_VERSION 2
|
|
#define RPC_PMAP_PORT 111
|
|
|
|
#define NFS2_FHSIZE 32
|
|
#define NFS3_FHSIZE 64
|
|
|
|
#define RPC_VERSION 2
|
|
|
|
enum rpc_msg_type {
|
|
RPC_CALL = 0,
|
|
RPC_REPLY = 1
|
|
};
|
|
|
|
enum rpc_auth_flavor {
|
|
RPC_AUTH_NULL = 0,
|
|
RPC_AUTH_UNIX = 1,
|
|
RPC_AUTH_SHORT = 2,
|
|
RPC_AUTH_DES = 3,
|
|
RPC_AUTH_KRB = 4,
|
|
};
|
|
|
|
enum rpc_reply_stat {
|
|
RPC_MSG_ACCEPTED = 0,
|
|
RPC_MSG_DENIED = 1
|
|
};
|
|
|
|
#define NFS_MAXFHSIZE 64
|
|
struct nfs_fh {
|
|
unsigned short size;
|
|
unsigned char data[NFS_MAXFHSIZE];
|
|
};
|
|
|
|
struct nfs2_fh {
|
|
char data[NFS2_FHSIZE];
|
|
};
|
|
|
|
#define NFS_MOUNT_VERSION 4
|
|
|
|
struct nfs_mount_data {
|
|
int version;
|
|
int fd;
|
|
struct nfs2_fh old_root;
|
|
int flags;
|
|
int rsize;
|
|
int wsize;
|
|
int timeo;
|
|
int retrans;
|
|
int acregmin;
|
|
int acregmax;
|
|
int acdirmin;
|
|
int acdirmax;
|
|
struct sockaddr_in addr;
|
|
char hostname[256];
|
|
int namlen;
|
|
unsigned int bsize;
|
|
struct nfs_fh root;
|
|
};
|
|
|
|
#define NFS_MOUNT_SOFT 0x0001 /* 1 */
|
|
#define NFS_MOUNT_INTR 0x0002 /* 1 */
|
|
#define NFS_MOUNT_SECURE 0x0004 /* 1 */
|
|
#define NFS_MOUNT_POSIX 0x0008 /* 1 */
|
|
#define NFS_MOUNT_NOCTO 0x0010 /* 1 */
|
|
#define NFS_MOUNT_NOAC 0x0020 /* 1 */
|
|
#define NFS_MOUNT_TCP 0x0040 /* 2 */
|
|
#define NFS_MOUNT_VER3 0x0080 /* 3 */
|
|
#define NFS_MOUNT_KERBEROS 0x0100 /* 3 */
|
|
#define NFS_MOUNT_NONLM 0x0200 /* 3 */
|
|
#define NFS_MOUNT_BROKEN_SUID 0x0400 /* 4 */
|
|
#define NFS_MOUNT_FLAGMASK 0xFFFF
|
|
|
|
static char nfs_root_name[256];
|
|
static u_int32_t root_server_addr;
|
|
static char root_server_path[256];
|
|
|
|
/* Address of NFS server */
|
|
static u_int32_t servaddr;
|
|
|
|
/* Name of directory to mount */
|
|
static char nfs_path[NFS_MAXPATHLEN];
|
|
|
|
/* NFS-related data */
|
|
static struct nfs_mount_data nfs_data = {
|
|
.version = NFS_MOUNT_VERSION,
|
|
.flags = NFS_MOUNT_NONLM, /* No lockd in nfs root yet */
|
|
.rsize = NFS_DEF_FILE_IO_BUFFER_SIZE,
|
|
.wsize = NFS_DEF_FILE_IO_BUFFER_SIZE,
|
|
.bsize = 0,
|
|
.timeo = 7,
|
|
.retrans = 3,
|
|
.acregmin = 3,
|
|
.acregmax = 60,
|
|
.acdirmin = 30,
|
|
.acdirmax = 60,
|
|
};
|
|
static int nfs_port = -1;
|
|
static int mount_port;
|
|
|
|
/***************************************************************************
|
|
|
|
Parsing of options
|
|
|
|
***************************************************************************/
|
|
|
|
/*
|
|
* The following integer options are recognized
|
|
*/
|
|
static struct nfs_int_opts {
|
|
const char *name;
|
|
int *val;
|
|
} root_int_opts[] = {
|
|
{ "port", &nfs_port },
|
|
{ "rsize", &nfs_data.rsize },
|
|
{ "wsize", &nfs_data.wsize },
|
|
{ "timeo", &nfs_data.timeo },
|
|
{ "retrans", &nfs_data.retrans },
|
|
{ "acregmin", &nfs_data.acregmin },
|
|
{ "acregmax", &nfs_data.acregmax },
|
|
{ "acdirmin", &nfs_data.acdirmin },
|
|
{ "acdirmax", &nfs_data.acdirmax },
|
|
{ NULL, NULL }
|
|
};
|
|
|
|
/*
|
|
* And now the flag options
|
|
*/
|
|
static struct nfs_bool_opts {
|
|
const char *name;
|
|
int and_mask;
|
|
int or_mask;
|
|
} root_bool_opts[] = {
|
|
{ "soft", ~NFS_MOUNT_SOFT, NFS_MOUNT_SOFT },
|
|
{ "hard", ~NFS_MOUNT_SOFT, 0 },
|
|
{ "intr", ~NFS_MOUNT_INTR, NFS_MOUNT_INTR },
|
|
{ "nointr", ~NFS_MOUNT_INTR, 0 },
|
|
{ "posix", ~NFS_MOUNT_POSIX, NFS_MOUNT_POSIX },
|
|
{ "noposix", ~NFS_MOUNT_POSIX, 0 },
|
|
{ "cto", ~NFS_MOUNT_NOCTO, 0 },
|
|
{ "nocto", ~NFS_MOUNT_NOCTO, NFS_MOUNT_NOCTO },
|
|
{ "ac", ~NFS_MOUNT_NOAC, 0 },
|
|
{ "noac", ~NFS_MOUNT_NOAC, NFS_MOUNT_NOAC },
|
|
{ "lock", ~NFS_MOUNT_NONLM, 0 },
|
|
{ "nolock", ~NFS_MOUNT_NONLM, NFS_MOUNT_NONLM },
|
|
#ifdef CONFIG_NFS_V3
|
|
{ "v2", ~NFS_MOUNT_VER3, 0 },
|
|
{ "v3", ~NFS_MOUNT_VER3, NFS_MOUNT_VER3 },
|
|
#endif
|
|
{ "udp", ~NFS_MOUNT_TCP, 0 },
|
|
{ "tcp", ~NFS_MOUNT_TCP, NFS_MOUNT_TCP },
|
|
{ "broken_suid",~NFS_MOUNT_BROKEN_SUID, NFS_MOUNT_BROKEN_SUID },
|
|
{ NULL, 0, 0 }
|
|
};
|
|
/*
|
|
* Parse option string.
|
|
*/
|
|
static void root_nfs_parse(char *name, char *buf)
|
|
{
|
|
char *options, *val, *cp;
|
|
|
|
if ((options = strchr(name, ','))) {
|
|
*options++ = 0;
|
|
cp = strtok(options, ",");
|
|
while (cp) {
|
|
if ((val = strchr(cp, '='))) {
|
|
struct nfs_int_opts *opts = root_int_opts;
|
|
*val++ = '\0';
|
|
while (opts->name && strcmp(opts->name, cp))
|
|
opts++;
|
|
if (opts->name)
|
|
*(opts->val) = (int) strtoul(val, NULL, 10);
|
|
} else {
|
|
struct nfs_bool_opts *opts = root_bool_opts;
|
|
while (opts->name && strcmp(opts->name, cp))
|
|
opts++;
|
|
if (opts->name) {
|
|
nfs_data.flags &= opts->and_mask;
|
|
nfs_data.flags |= opts->or_mask;
|
|
}
|
|
}
|
|
cp = strtok(NULL, ",");
|
|
}
|
|
}
|
|
if (name[0] && strcmp(name, "default")) {
|
|
strncpy(buf, name, NFS_MAXPATHLEN-1);
|
|
buf[NFS_MAXPATHLEN-1] = 0;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Prepare the NFS data structure and parse all options.
|
|
*/
|
|
static int root_nfs_name(char *name)
|
|
{
|
|
char buf[NFS_MAXPATHLEN];
|
|
struct utsname uname_buf;
|
|
|
|
/* Set some default values */
|
|
strcpy(buf, NFS_ROOT);
|
|
|
|
/* Process options received from the remote server */
|
|
root_nfs_parse(root_server_path, buf);
|
|
|
|
/* Override them by options set on kernel command-line */
|
|
root_nfs_parse(name, buf);
|
|
|
|
uname(&uname_buf);
|
|
if (strlen(buf) + strlen(uname_buf.nodename) > NFS_MAXPATHLEN) {
|
|
printf("nfsroot: Pathname for remote directory too long.\n");
|
|
return -1;
|
|
}
|
|
sprintf(nfs_path, buf, uname_buf.nodename);
|
|
|
|
return 1;
|
|
}
|
|
|
|
/***************************************************************************
|
|
|
|
Routines to actually mount the root directory
|
|
|
|
***************************************************************************/
|
|
|
|
/*
|
|
* Construct sockaddr_in from address and port number.
|
|
*/
|
|
static inline void
|
|
set_sockaddr(struct sockaddr_in *sin, u_int32_t addr, u_int16_t port)
|
|
{
|
|
memset(sin, 0, sizeof(*sin));
|
|
sin->sin_family = AF_INET;
|
|
sin->sin_addr.s_addr = addr;
|
|
sin->sin_port = port;
|
|
}
|
|
|
|
/*
|
|
* Extremely crude RPC-over-UDP call. We get an already encoded request
|
|
* to pass, we do that and put the reply into buffer. That (and callers
|
|
* below - getport, getfh2 and getfh3) should be replaced with proper
|
|
* librpc use. Now, if we only had one that wasn't bloated as a dead
|
|
* gnu that had lied for a while under the sun...
|
|
*/
|
|
|
|
static u_int32_t XID;
|
|
static int flag;
|
|
static void timeout(int n)
|
|
{
|
|
(void)n;
|
|
flag = 1;
|
|
}
|
|
static int do_call(struct sockaddr_in *sin, u_int32_t msg[], u_int32_t rmsg[],
|
|
u_int32_t len, u_int32_t rlen)
|
|
{
|
|
struct sockaddr_in from;
|
|
int slen = sizeof(struct sockaddr_in);
|
|
struct timeval tv = {1, 0};
|
|
int n;
|
|
int fd;
|
|
|
|
sysv_signal(SIGALRM, timeout);
|
|
fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
if (fd < 0)
|
|
goto Esocket;
|
|
setsockopt(fd, SOL_SOCKET, SO_SNDTIMEO, (void*)&tv, sizeof(tv));
|
|
len *= 4;
|
|
if (sendto(fd, msg, len, 0, (struct sockaddr *)sin, slen)!=(int)len)
|
|
goto Esend;
|
|
setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, (void*)&tv, sizeof(tv));
|
|
alarm(0);
|
|
flag = 0;
|
|
alarm(5);
|
|
rlen *= 4;
|
|
do {
|
|
slen = sizeof(from);
|
|
n = recvfrom(fd, rmsg, rlen, 0, (struct sockaddr*)&from, &slen);
|
|
if (flag || n < 0)
|
|
goto Erecv;
|
|
} while (memcmp(&from, sin, sizeof(from)) || rmsg[0] != msg[0]);
|
|
|
|
if (n < 6*4 || n % 4 || ntohl(rmsg[1]) != 1 || rmsg[2] ||
|
|
rmsg[3] || rmsg[4] || rmsg[5])
|
|
goto Einval;
|
|
alarm(0);
|
|
close(fd);
|
|
return n / 4 - 6;
|
|
|
|
Esend: printf("rpc: write failed\n");
|
|
goto out;
|
|
Erecv: printf("rpc: read failed\n");
|
|
goto out;
|
|
Einval: printf("rpc: invalid response\n");
|
|
goto out;
|
|
Esocket:printf("rpc: can't create socket\n");
|
|
return -1;
|
|
out:
|
|
alarm(0);
|
|
close(fd);
|
|
return -1;
|
|
}
|
|
|
|
enum {
|
|
PMAP_GETPORT = 3
|
|
};
|
|
|
|
static void do_header(u_int32_t msg[], u_int32_t prog, u_int32_t vers, u_int32_t proc)
|
|
{
|
|
msg[0] = XID++;
|
|
msg[1] = htonl(RPC_CALL);
|
|
msg[2] = htonl(RPC_VERSION);
|
|
msg[3] = htonl(prog);
|
|
msg[4] = htonl(vers);
|
|
msg[5] = htonl(proc);
|
|
msg[6] = htonl(RPC_AUTH_NULL);
|
|
msg[7] = htonl(0);
|
|
msg[8] = htonl(RPC_AUTH_NULL);
|
|
msg[9] = htonl(0);
|
|
}
|
|
|
|
static int getport(u_int32_t prog, u_int32_t vers, u_int32_t prot)
|
|
{
|
|
struct sockaddr_in sin;
|
|
unsigned msg[14];
|
|
unsigned rmsg[7];
|
|
int n;
|
|
set_sockaddr(&sin, servaddr, htons(RPC_PMAP_PORT));
|
|
do_header(msg, RPC_PMAP_PROGRAM, RPC_PMAP_VERSION, PMAP_GETPORT);
|
|
msg[10] = htonl(prog);
|
|
msg[11] = htonl(vers);
|
|
msg[12] = htonl(prot);
|
|
msg[13] = htonl(0);
|
|
n = do_call(&sin, msg, rmsg, 14, 7);
|
|
if (n <= 0)
|
|
return -1;
|
|
else
|
|
return ntohl(rmsg[6]);
|
|
}
|
|
|
|
static int getfh2(void)
|
|
{
|
|
struct sockaddr_in sin;
|
|
unsigned msg[10+1+256/4];
|
|
unsigned rmsg[6 + 1 + NFS2_FHSIZE/4];
|
|
int n;
|
|
int len = strlen(nfs_path);
|
|
set_sockaddr(&sin, servaddr, mount_port);
|
|
|
|
if (len > 255) {
|
|
printf("nfsroot: pathname is too long");
|
|
return -1;
|
|
}
|
|
memset(msg, 0, sizeof(msg));
|
|
do_header(msg, NFS_MNT_PROGRAM, NFS_MNT_VERSION, MNTPROC_MNT);
|
|
msg[10] = htonl(len);
|
|
strcpy((char*)&msg[11], nfs_path);
|
|
n = do_call(&sin, msg, rmsg, 11 + (len + 3)/4, 7 + NFS2_FHSIZE/4);
|
|
if (n < 0)
|
|
return -1;
|
|
if (n != NFS2_FHSIZE/4 + 1)
|
|
goto Esize;
|
|
if (rmsg[6]) {
|
|
printf("nfsroot: mountd returned an error (%d)",htonl(rmsg[6]));
|
|
return -1;
|
|
}
|
|
nfs_data.root.size = NFS2_FHSIZE;
|
|
memcpy(nfs_data.root.data, &rmsg[7], NFS2_FHSIZE);
|
|
return 0;
|
|
Esize:
|
|
printf("nfsroot: bad fhandle size");
|
|
return -1;
|
|
}
|
|
|
|
static int getfh3(void)
|
|
{
|
|
struct sockaddr_in sin;
|
|
unsigned msg[10+1+256/4];
|
|
unsigned rmsg[6 + 1 + 1 + NFS3_FHSIZE/4];
|
|
int n;
|
|
int len = strlen(nfs_path);
|
|
int size;
|
|
set_sockaddr(&sin, servaddr, mount_port);
|
|
|
|
if (len > 255) {
|
|
printf("nfsroot: pathname is too long");
|
|
return -1;
|
|
}
|
|
memset(msg, 0, sizeof(msg));
|
|
do_header(msg, NFS_MNT_PROGRAM, NFS_MNT3_VERSION, MOUNTPROC3_MNT);
|
|
msg[10] = htonl(len);
|
|
strcpy((char*)&msg[11], nfs_path);
|
|
n = do_call(&sin, msg, rmsg, 11 + (len + 3)/4, 8 + NFS3_FHSIZE/4);
|
|
if (n < 0)
|
|
return -1;
|
|
if (n <= 2)
|
|
goto Esize;
|
|
if (rmsg[6]) {
|
|
printf("nfsroot: mountd returned an error (%d)",htonl(rmsg[6]));
|
|
return -1;
|
|
}
|
|
size = ntohl(rmsg[7]);
|
|
if (size > NFS3_FHSIZE || n != 2 + size/4)
|
|
goto Esize;
|
|
nfs_data.root.size = size;
|
|
memcpy(nfs_data.root.data, &rmsg[8], size);
|
|
return 0;
|
|
Esize:
|
|
printf("nfsroot: bad fhandle size");
|
|
return -1;
|
|
}
|
|
|
|
/*
|
|
* Use portmapper to find mountd and nfsd port numbers if not overriden
|
|
* by the user. Use defaults if portmapper is not available.
|
|
* XXX: Is there any nfs server with no portmapper?
|
|
*/
|
|
static int root_nfs_ports(void)
|
|
{
|
|
int port;
|
|
int nfsd_ver, mountd_ver;
|
|
int proto;
|
|
|
|
if (nfs_data.flags & NFS_MOUNT_VER3) {
|
|
nfsd_ver = NFS3_VERSION;
|
|
mountd_ver = NFS_MNT3_VERSION;
|
|
} else {
|
|
nfsd_ver = NFS2_VERSION;
|
|
mountd_ver = NFS_MNT_VERSION;
|
|
}
|
|
|
|
proto = (nfs_data.flags & NFS_MOUNT_TCP) ? IPPROTO_TCP : IPPROTO_UDP;
|
|
|
|
if (nfs_port < 0) {
|
|
if ((port = getport(NFS_PROGRAM, nfsd_ver, proto)) < 0) {
|
|
printf("nfsroot: Unable to get nfsd port "
|
|
"number from server, using default\n");
|
|
port = NFS_PORT;
|
|
}
|
|
nfs_port = htons(port);
|
|
printf("nfsroot: Portmapper on server returned %d "
|
|
"as nfsd port\n", port);
|
|
}
|
|
|
|
if ((port = getport(NFS_MNT_PROGRAM, mountd_ver, proto)) < 0) {
|
|
printf("nfsroot: Unable to get mountd port "
|
|
"number from server, using default\n");
|
|
port = NFS_MNT_PORT;
|
|
}
|
|
mount_port = htons(port);
|
|
printf("nfsroot: mountd port is %d\n", port);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int main(void)
|
|
{
|
|
unsigned char *p;
|
|
struct timeval tv;
|
|
char *s;
|
|
|
|
/* FIX: use getopt() instead of this */
|
|
|
|
s = getenv("root_server_addr");
|
|
if (s)
|
|
root_server_addr = strtoul(s, NULL, 10);
|
|
s = getenv("root_server_path");
|
|
if (s)
|
|
strncpy(root_server_path, s, 255);
|
|
s = getenv("nfs_root_name");
|
|
if (s)
|
|
strncpy(nfs_root_name, s, 255);
|
|
|
|
/*
|
|
* Decode the root directory path name and NFS options from
|
|
* the kernel command line. This has to go here in order to
|
|
* be able to use the client IP address for the remote root
|
|
* directory (necessary for pure RARP booting).
|
|
*/
|
|
if (root_nfs_name(nfs_root_name) < 0)
|
|
return 0;
|
|
if ((servaddr = root_server_addr) == INADDR_NONE) {
|
|
printf("nfsroot: No NFS server available, giving up.\n");
|
|
return 0;
|
|
}
|
|
|
|
p = (char *) &servaddr;
|
|
sprintf(nfs_data.hostname, "%d.%d.%d.%d", p[0], p[1], p[2], p[3]);
|
|
|
|
#ifdef NFSROOT_DEBUG
|
|
printf("nfsroot: Mounting %s on server %s as root\n",
|
|
nfs_path, nfs_data.hostname);
|
|
printf("nfsroot: rsize = %d, wsize = %d, timeo = %d, retrans = %d\n",
|
|
nfs_data.rsize, nfs_data.wsize, nfs_data.timeo, nfs_data.retrans);
|
|
printf("nfsroot: acreg (min,max) = (%d,%d), acdir (min,max) = (%d,%d)\n",
|
|
nfs_data.acregmin, nfs_data.acregmax,
|
|
nfs_data.acdirmin, nfs_data.acdirmax);
|
|
printf("nfsroot: nfsd port = %d, mountd port = %d, flags = %08x\n",
|
|
nfs_port, mount_port, nfs_data.flags);
|
|
#endif
|
|
|
|
gettimeofday(&tv, NULL);
|
|
XID = (tv.tv_sec << 15) ^ tv.tv_usec;
|
|
|
|
if (root_nfs_ports() < 0)
|
|
return 0;
|
|
if (nfs_data.flags & NFS_MOUNT_VER3) {
|
|
if (getfh3())
|
|
return 0;
|
|
} else {
|
|
if (getfh2())
|
|
return 0;
|
|
}
|
|
set_sockaddr((struct sockaddr_in *) &nfs_data.addr, servaddr, nfs_port);
|
|
return mount("/dev/root", "/mnt", "nfs", 0, &nfs_data) == 0;
|
|
}
|