mirror of
https://github.com/opnsense/src.git
synced 2026-04-29 10:11:09 -04:00
Add create_geom and destroy_geom methods.
This commit is contained in:
parent
67ffbc63f3
commit
29c331bfaa
1 changed files with 159 additions and 0 deletions
|
|
@ -118,6 +118,163 @@ g_bde_access(struct g_provider *pp, int dr, int dw, int de)
|
|||
return (g_access_rel(cp, dr, dw, de));
|
||||
}
|
||||
|
||||
static int
|
||||
g_bde_create_geom(struct gctl_req *req, struct g_class *mp, struct g_provider *pp)
|
||||
{
|
||||
struct g_geom *gp;
|
||||
struct g_consumer *cp;
|
||||
struct g_bde_key *kp;
|
||||
int error, i;
|
||||
u_int sectorsize;
|
||||
off_t mediasize;
|
||||
struct g_bde_softc *sc;
|
||||
void *pass;
|
||||
void *key;
|
||||
|
||||
if (pp == NULL)
|
||||
return (gctl_error(req, "Provider needed"));
|
||||
g_trace(G_T_TOPOLOGY, "g_bde_create_geom(%s, %s)", mp->name, pp->name);
|
||||
g_topology_assert();
|
||||
gp = NULL;
|
||||
|
||||
pass = gctl_get_param(req, "pass", &i);
|
||||
if (pass == NULL || i != SHA512_DIGEST_LENGTH) {
|
||||
if (pass != NULL) {
|
||||
bzero(pass, i);
|
||||
g_free(pass);
|
||||
}
|
||||
return (gctl_error(req, "No usable key presented"));
|
||||
}
|
||||
key = gctl_get_param(req, "key", &i);
|
||||
if (key != NULL && i != 16) {
|
||||
bzero(key, i);
|
||||
bzero(pass, SHA512_DIGEST_LENGTH);
|
||||
g_free(pass);
|
||||
g_free(key);
|
||||
return (gctl_error(req, "Invalid key presented"));
|
||||
}
|
||||
|
||||
gp = g_new_geomf(mp, "%s.bde", pp->name);
|
||||
gp->start = g_bde_start;
|
||||
gp->orphan = g_bde_orphan;
|
||||
gp->access = g_bde_access;
|
||||
gp->spoiled = g_std_spoiled;
|
||||
cp = g_new_consumer(gp);
|
||||
g_attach(cp, pp);
|
||||
error = g_access_rel(cp, 1, 1, 1);
|
||||
if (error) {
|
||||
g_detach(cp);
|
||||
g_destroy_consumer(cp);
|
||||
g_destroy_geom(gp);
|
||||
return (error);
|
||||
}
|
||||
g_topology_unlock();
|
||||
g_waitidle();
|
||||
while (1) {
|
||||
sectorsize = cp->provider->sectorsize;
|
||||
mediasize = cp->provider->mediasize;
|
||||
sc = g_malloc(sizeof(struct g_bde_softc), M_WAITOK | M_ZERO);
|
||||
gp->softc = sc;
|
||||
sc->geom = gp;
|
||||
sc->consumer = cp;
|
||||
|
||||
error = g_bde_decrypt_lock(sc, pass, key,
|
||||
mediasize, sectorsize, NULL);
|
||||
bzero(pass, SHA512_DIGEST_LENGTH);
|
||||
if (key != NULL)
|
||||
bzero(key, 16);
|
||||
bzero(sc->sha2, sizeof sc->sha2);
|
||||
if (error)
|
||||
break;
|
||||
kp = &sc->key;
|
||||
|
||||
/* Initialize helper-fields */
|
||||
kp->keys_per_sector = kp->sectorsize / G_BDE_SKEYLEN;
|
||||
kp->zone_cont = kp->keys_per_sector * kp->sectorsize;
|
||||
kp->zone_width = kp->zone_cont + kp->sectorsize;
|
||||
kp->media_width = kp->sectorN - kp->sector0 -
|
||||
G_BDE_MAXKEYS * kp->sectorsize;
|
||||
|
||||
/* Our external parameters */
|
||||
sc->zone_cont = kp->zone_cont;
|
||||
sc->mediasize = g_bde_max_sector(kp);
|
||||
sc->sectorsize = kp->sectorsize;
|
||||
|
||||
TAILQ_INIT(&sc->freelist);
|
||||
TAILQ_INIT(&sc->worklist);
|
||||
mtx_init(&sc->worklist_mutex, "g_bde_worklist", NULL, MTX_DEF);
|
||||
mtx_lock(&Giant);
|
||||
/* XXX: error check */
|
||||
kthread_create(g_bde_worker, gp, &sc->thread, 0, 0,
|
||||
"g_bde %s", gp->name);
|
||||
mtx_unlock(&Giant);
|
||||
g_topology_lock();
|
||||
pp = g_new_providerf(gp, gp->name);
|
||||
pp->flags |= G_PF_CANDELETE;
|
||||
pp->stripesize = kp->zone_cont;
|
||||
pp->stripeoffset = 0;
|
||||
pp->mediasize = sc->mediasize;
|
||||
pp->sectorsize = sc->sectorsize;
|
||||
g_error_provider(pp, 0);
|
||||
g_topology_unlock();
|
||||
break;
|
||||
}
|
||||
g_topology_lock();
|
||||
if (error == 0) {
|
||||
return (0);
|
||||
} else {
|
||||
g_access_rel(cp, -1, -1, -1);
|
||||
}
|
||||
g_detach(cp);
|
||||
g_destroy_consumer(cp);
|
||||
if (gp->softc != NULL)
|
||||
g_free(gp->softc);
|
||||
g_destroy_geom(gp);
|
||||
return (error);
|
||||
}
|
||||
|
||||
|
||||
static int
|
||||
g_bde_destroy_geom(struct gctl_req *req, struct g_class *mp, struct g_geom *gp)
|
||||
{
|
||||
struct g_consumer *cp;
|
||||
struct g_provider *pp;
|
||||
int error;
|
||||
struct g_bde_softc *sc;
|
||||
|
||||
g_trace(G_T_TOPOLOGY, "g_bde_destroy_geom(%s, %s)", mp->name, gp->name);
|
||||
g_topology_assert();
|
||||
/*
|
||||
* Orderly detachment.
|
||||
*/
|
||||
KASSERT(gp != NULL, ("NULL geom"));
|
||||
pp = LIST_FIRST(&gp->provider);
|
||||
KASSERT(pp != NULL, ("NULL provider"));
|
||||
if (pp->acr > 0 || pp->acw > 0 || pp->ace > 0)
|
||||
return (EBUSY);
|
||||
g_orphan_provider(pp, ENXIO);
|
||||
sc = gp->softc;
|
||||
cp = LIST_FIRST(&gp->consumer);
|
||||
KASSERT(cp != NULL, ("NULL consumer"));
|
||||
sc->dead = 1;
|
||||
wakeup(sc);
|
||||
error = g_access_rel(cp, -1, -1, -1);
|
||||
KASSERT(error == 0, ("error on close"));
|
||||
g_detach(cp);
|
||||
g_destroy_consumer(cp);
|
||||
g_topology_unlock();
|
||||
while (sc->dead != 2 && !LIST_EMPTY(&pp->consumers))
|
||||
tsleep(sc, PRIBIO, "g_bdedie", hz);
|
||||
g_waitidle();
|
||||
g_topology_lock();
|
||||
g_destroy_provider(pp);
|
||||
mtx_destroy(&sc->worklist_mutex);
|
||||
bzero(&sc->key, sizeof sc->key);
|
||||
g_free(sc);
|
||||
g_destroy_geom(gp);
|
||||
return (0);
|
||||
}
|
||||
|
||||
static int
|
||||
g_bde_config(struct g_configargs *ga)
|
||||
{
|
||||
|
|
@ -273,6 +430,8 @@ g_bde_config(struct g_configargs *ga)
|
|||
static struct g_class g_bde_class = {
|
||||
.name = BDE_CLASS_NAME,
|
||||
.config = g_bde_config,
|
||||
.create_geom = g_bde_create_geom,
|
||||
.destroy_geom = g_bde_destroy_geom,
|
||||
G_CLASS_INITIALIZER
|
||||
};
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue