#include <sys/param.h>
#include <sys/stdarg.h>
#include <netinet/in.h>
#include <stand.h>
#include <sys/disk.h>
#include "disk.h"
#include "libofw.h"
static int ofwd_init(void);
static int ofwd_strategy(void *devdata, int flag, daddr_t dblk,
size_t size, char *buf, size_t *rsize);
static int ofwd_open(struct open_file *f, ...);
static int ofwd_close(struct open_file *f);
static int ofwd_ioctl(struct open_file *f, u_long cmd, void *data);
static int ofwd_print(int verbose);
static char * ofwd_fmtdev(struct devdesc *);
static int ofwd_parsedev(struct devdesc **, const char *, const char **);
static bool ofwd_match(struct devsw *, const char *);
struct devsw ofwdisk = {
.dv_name = "block",
.dv_type = DEVT_OFDISK,
.dv_init = ofwd_init,
.dv_strategy = ofwd_strategy,
.dv_open = ofwd_open,
.dv_close = ofwd_close,
.dv_ioctl = ofwd_ioctl,
.dv_print = ofwd_print,
.dv_cleanup = nullsys,
.dv_match = ofwd_match,
.dv_fmtdev = ofwd_fmtdev,
.dv_parsedev = ofwd_parsedev,
};
static struct ofw_devdesc *kdp;
static int
ofwd_init(void)
{
return (0);
}
static int
ofwd_strategy(void *devdata, int flag __unused, daddr_t dblk, size_t size,
char *buf, size_t *rsize)
{
struct ofw_devdesc *dp = (struct ofw_devdesc *)devdata;
daddr_t pos;
int n;
if (dp != kdp) {
if (kdp != NULL) {
#if !defined(__powerpc__)
OF_close(kdp->d_handle);
#endif
kdp = NULL;
}
if ((dp->d_handle = OF_open(dp->d_path)) == -1)
return (ENOENT);
kdp = dp;
}
pos = dblk * 512;
do {
if (OF_seek(dp->d_handle, pos) < 0)
return (EIO);
n = OF_read(dp->d_handle, buf, size);
if (n < 0 && n != -2)
return (EIO);
} while (n == -2);
*rsize = size;
return (0);
}
static int
ofwd_open(struct open_file *f, ...)
{
struct ofw_devdesc *dp;
va_list vl;
va_start(vl, f);
dp = va_arg(vl, struct ofw_devdesc *);
va_end(vl);
if (dp != kdp) {
if (kdp != NULL) {
OF_close(kdp->d_handle);
kdp = NULL;
}
if ((dp->d_handle = OF_open(dp->d_path)) == -1) {
printf("%s: Could not open %s\n", __func__,
dp->d_path);
return (ENOENT);
}
kdp = dp;
}
return (0);
}
static int
ofwd_close(struct open_file *f)
{
struct ofw_devdesc *dev = f->f_devdata;
if (dev == kdp) {
#if !defined(__powerpc__)
OF_close(dev->d_handle);
#endif
kdp = NULL;
}
return (0);
}
static int
ofwd_ioctl(struct open_file *f, u_long cmd, void *data)
{
struct ofw_devdesc *dev = f->f_devdata;
int block_size;
unsigned int n;
switch (cmd) {
case DIOCGSECTORSIZE:
block_size = OF_block_size(dev->d_handle);
*(u_int *)data = block_size;
break;
case DIOCGMEDIASIZE:
block_size = OF_block_size(dev->d_handle);
n = OF_blocks(dev->d_handle);
*(uint64_t *)data = ((uint64_t)n * block_size);
break;
default:
return (ENOTTY);
}
return (0);
}
static int
ofwd_print(int verbose __unused)
{
uintmax_t block_size, n;
int ret;
char line[80];
if ((ret = pager_output("block devices:\n")) != 0)
return (ret);
if (kdp != NULL) {
block_size = OF_block_size(kdp->d_handle);
n = OF_blocks(kdp->d_handle);
snprintf(line, sizeof(line),
" %s: OFW block device (%ju X %ju): %ju bytes\n",
kdp->d_path, n, block_size, n * block_size);
if ((ret = pager_output(line)) != 0)
return (ret);
} else {
if ((ret = pager_output(" none are open, so no info\n")) != 0)
return (ret);
}
return (0);
}
static bool
ofwd_match(struct devsw *devsw, const char *devspec)
{
const char *path;
return (ofw_path_to_handle(devspec, devsw->dv_name, &path) != -1);
}
static char *
ofwd_fmtdev(struct devdesc *idev)
{
struct ofw_devdesc *dev = (struct ofw_devdesc *)idev;
return (dev->d_path);
}
static int
ofwd_parsedev(struct devdesc **dev, const char *devspec, const char **path)
{
return (ofw_common_parsedev(dev, devspec, path, ofwdisk.dv_name));
}