#include <sys/param.h>
#include <sys/systm.h>
#include <sys/buf.h>
#include <sys/kernel.h>
#include <sys/malloc.h>
#include <sys/proc.h>
#include <sys/mount.h>
#include <sys/vnode.h>
#include <sys/time.h>
#include <vm/vm.h>
#include <vm/vm_extern.h>
#include <vm/vm_page2.h>
#include <sys/buf2.h>
#include <vfs/msdosfs/bpb.h>
#include <vfs/msdosfs/direntry.h>
#include <vfs/msdosfs/denode.h>
#include <vfs/msdosfs/fat.h>
#include <vfs/msdosfs/msdosfsmount.h>
static MALLOC_DEFINE(M_MSDOSFSNODE, "MSDOSFS node", "MSDOSFS vnode private part");
static struct denode **dehashtbl;
static u_long dehash;
static struct lwkt_token dehash_token;
#define DEHASH(dev, dcl, doff) (dehashtbl[(minor(dev) + (dcl) + (doff) / \
sizeof(struct direntry)) & dehash])
int
msdosfs_init(struct vfsconf *vfsp)
{
dehash = vfs_inodehashsize();
dehashtbl = kmalloc(sizeof(void *) * dehash,
M_MSDOSFSMNT, M_WAITOK|M_ZERO);
--dehash;
lwkt_token_init(&dehash_token, "msdosihash");
return (0);
}
int
msdosfs_uninit(struct vfsconf *vfsp)
{
if (dehashtbl)
kfree(dehashtbl, M_MSDOSFSMNT);
return (0);
}
static struct denode *
msdosfs_hashget(cdev_t dev, u_long dirclust, u_long diroff)
{
struct denode *dep;
struct vnode *vp;
lwkt_gettoken(&dehash_token);
loop:
for (dep = DEHASH(dev, dirclust, diroff); dep; dep = dep->de_next) {
if (dirclust != dep->de_dirclust
|| diroff != dep->de_diroffset
|| dev != dep->de_dev
|| dep->de_refcnt <= 0) {
continue;
}
vp = DETOV(dep);
if (vget(vp, LK_EXCLUSIVE))
goto loop;
for (dep = DEHASH(dev, dirclust, diroff); dep;
dep = dep->de_next) {
if (dirclust == dep->de_dirclust
&& diroff == dep->de_diroffset
&& dev == dep->de_dev
&& dep->de_refcnt > 0) {
break;
}
}
if (dep == NULL || DETOV(dep) != vp) {
vput(vp);
goto loop;
}
lwkt_reltoken(&dehash_token);
return (dep);
}
lwkt_reltoken(&dehash_token);
return (NULL);
}
static
int
msdosfs_hashins(struct denode *dep)
{
struct denode **depp, *deq;
lwkt_gettoken(&dehash_token);
depp = &DEHASH(dep->de_dev, dep->de_dirclust, dep->de_diroffset);
while ((deq = *depp) != NULL) {
if (deq->de_dev == dep->de_dev &&
deq->de_dirclust == dep->de_dirclust &&
deq->de_diroffset == dep->de_diroffset &&
deq->de_refcnt > 0) {
lwkt_reltoken(&dehash_token);
return(EBUSY);
}
depp = &deq->de_next;
}
dep->de_next = NULL;
*depp = dep;
lwkt_reltoken(&dehash_token);
return(0);
}
static
void
msdosfs_hashrem(struct denode *dep)
{
struct denode **depp, *deq;
lwkt_gettoken(&dehash_token);
depp = &DEHASH(dep->de_dev, dep->de_dirclust, dep->de_diroffset);
while ((deq = *depp) != NULL) {
if (dep == deq)
break;
depp = &deq->de_next;
}
KKASSERT(dep == deq);
*depp = dep->de_next;
dep->de_next = NULL;
lwkt_reltoken(&dehash_token);
}
void
msdosfs_reinsert(struct denode *ip, u_long new_dirclust, u_long new_diroffset)
{
int error;
lwkt_gettoken(&dehash_token);
msdosfs_hashrem(ip);
ip->de_dirclust = new_dirclust;
ip->de_diroffset = new_diroffset;
error = msdosfs_hashins(ip);
KASSERT(!error, ("msdosfs_reinsert: insertion failed %d", error));
lwkt_reltoken(&dehash_token);
}
int
deget(struct msdosfsmount *pmp, u_long dirclust, u_long diroffset,
struct denode **depp)
{
int error;
cdev_t dev = pmp->pm_dev;
struct mount *mntp = pmp->pm_mountp;
struct direntry *direntptr;
struct denode *ldep;
struct vnode *nvp;
struct buf *bp;
mprintf("deget(pmp %p, dirclust %lu, diroffset %lx, depp %p)\n",
pmp, dirclust, diroffset, depp);
if (FAT32(pmp) && dirclust == MSDOSFSROOT)
dirclust = pmp->pm_rootdirblk;
again:
ldep = msdosfs_hashget(dev, dirclust, diroffset);
if (ldep) {
*depp = ldep;
return (0);
}
ldep = kmalloc(sizeof(struct denode), M_MSDOSFSNODE, M_WAITOK | M_ZERO);
error = getnewvnode(VT_MSDOSFS, mntp, &nvp, VLKTIMEOUT, 0);
if (error) {
*depp = NULL;
kfree(ldep, M_MSDOSFSNODE);
return error;
}
ldep->de_vnode = nvp;
ldep->de_flag = 0;
ldep->de_devvp = 0;
ldep->de_dev = dev;
ldep->de_dirclust = dirclust;
ldep->de_diroffset = diroffset;
fc_purge(ldep, 0);
error = msdosfs_hashins(ldep);
if (error == EBUSY) {
nvp->v_type = VBAD;
vx_put(nvp);
kfree(ldep, M_MSDOSFSNODE);
goto again;
} else if (error) {
nvp->v_type = VBAD;
vx_put(nvp);
kfree(ldep, M_MSDOSFSNODE);
*depp = NULL;
return (EINVAL);
}
nvp->v_data = ldep;
ldep->de_pmp = pmp;
ldep->de_refcnt = 1;
if ((dirclust == MSDOSFSROOT ||
(FAT32(pmp) && dirclust == pmp->pm_rootdirblk)) &&
diroffset == MSDOSFSROOT_OFS) {
vsetflags(nvp, VROOT);
ldep->de_Attributes = ATTR_DIRECTORY;
ldep->de_LowerCase = 0;
if (FAT32(pmp)) {
ldep->de_StartCluster = pmp->pm_rootdirblk;
} else {
ldep->de_StartCluster = MSDOSFSROOT;
ldep->de_FileSize = pmp->pm_rootdirsize * DEV_BSIZE;
}
ldep->de_CHun = 0;
ldep->de_CTime = 0x0000;
ldep->de_CDate = (0 << DD_YEAR_SHIFT) | (1 << DD_MONTH_SHIFT)
| (1 << DD_DAY_SHIFT);
ldep->de_ADate = ldep->de_CDate;
ldep->de_MTime = ldep->de_CTime;
ldep->de_MDate = ldep->de_CDate;
} else {
error = readep(pmp, dirclust, diroffset, &bp, &direntptr);
if (error) {
ldep->de_Name[0] = SLOT_DELETED;
nvp->v_type = VBAD;
vx_put(nvp);
*depp = NULL;
return (error);
}
DE_INTERNALIZE(ldep, direntptr);
brelse(bp);
}
if (ldep->de_Attributes & ATTR_DIRECTORY) {
u_long size;
if (diroffset == 0 && ldep->de_StartCluster != dirclust) {
kprintf("deget(): \".\" entry at clust %lu != %lu\n",
dirclust, ldep->de_StartCluster);
ldep->de_StartCluster = dirclust;
}
nvp->v_type = VDIR;
if (ldep->de_StartCluster != MSDOSFSROOT) {
error = pcbmap(ldep, 0xffff, NULL, &size, NULL);
if (error == E2BIG) {
ldep->de_FileSize = de_cn2off(pmp, size);
error = 0;
} else
kprintf("deget(): pcbmap returned %d\n", error);
}
} else
nvp->v_type = VREG;
ldep->de_modrev = init_va_filerev();
ldep->de_devvp = pmp->pm_devvp;
vref(ldep->de_devvp);
vinitvmio(nvp, ldep->de_FileSize, PAGE_SIZE, -1);
vx_downgrade(nvp);
*depp = ldep;
return (0);
}
int
deupdat(struct denode *dep, int waitfor)
{
struct direntry dir;
struct timespec ts;
struct buf *bp;
struct direntry *dirp;
int error;
if (DETOV(dep)->v_mount->mnt_flag & MNT_RDONLY) {
dep->de_flag &= ~(DE_UPDATE | DE_CREATE | DE_ACCESS |
DE_MODIFIED);
return (0);
}
vfs_timestamp(&ts);
DETIMES(dep, &ts, &ts, &ts);
if ((dep->de_flag & DE_MODIFIED) == 0 && waitfor == 0)
return (0);
dep->de_flag &= ~DE_MODIFIED;
if (DETOV(dep)->v_flag & VROOT)
return (EINVAL);
if (dep->de_refcnt <= 0)
return (0);
error = readde(dep, &bp, &dirp);
if (error)
return (error);
DE_EXTERNALIZE(&dir, dep);
if (bcmp(dirp, &dir, sizeof(dir)) == 0) {
if (waitfor == 0 || (bp->b_flags & B_DELWRI) == 0) {
brelse(bp);
return (0);
}
} else
*dirp = dir;
if ((DETOV(dep)->v_mount->mnt_flag & MNT_NOCLUSTERW) == 0)
bp->b_flags |= B_CLUSTEROK;
if (waitfor)
error = bwrite(bp);
else if (vm_paging_min() || buf_dirty_count_severe())
bawrite(bp);
else
bdwrite(bp);
return (error);
}
int
detrunc(struct denode *dep, u_long length, int flags)
{
int error;
int allerror;
u_long eofentry;
u_long chaintofree;
daddr_t bn;
int boff;
int isadir = dep->de_Attributes & ATTR_DIRECTORY;
struct buf *bp;
struct msdosfsmount *pmp = dep->de_pmp;
mprintf("detrunc(): file %s, length %lu, flags %x\n", dep->de_Name,
length, flags);
if ((DETOV(dep)->v_flag & VROOT) && !FAT32(pmp)) {
kprintf("detrunc(): can't truncate root directory, clust %ld, "
"offset %ld\n",
dep->de_dirclust, dep->de_diroffset);
return (EINVAL);
}
if (dep->de_FileSize < length) {
vnode_pager_setsize(DETOV(dep), length);
return (deextend(dep, length));
}
if (length == 0) {
chaintofree = dep->de_StartCluster;
dep->de_StartCluster = 0;
eofentry = ~0;
} else {
error = pcbmap(dep, de_clcount(pmp, length) - 1,
NULL, &eofentry, NULL);
if (error) {
mprintf("detrunc(): pcbmap fails %d\n", error);
return (error);
}
}
fc_purge(dep, de_clcount(pmp, length));
if ((boff = length & pmp->pm_crbomask) != 0) {
if (isadir) {
bn = cntobn(pmp, eofentry);
error = bread(pmp->pm_devvp, de_bn2doff(pmp, bn),
pmp->pm_bpcluster, &bp);
} else {
u_long cn = de_cluster(pmp, length);
error = bread(DETOV(dep), de_cn2doff(pmp, cn),
pmp->pm_bpcluster, &bp);
}
if (error) {
brelse(bp);
mprintf("detrunc(): bread fails %d\n", error);
return (error);
}
memset(bp->b_data + boff, 0, pmp->pm_bpcluster - boff);
if ((flags & IO_SYNC) != 0)
bwrite(bp);
else
bdwrite(bp);
}
dep->de_FileSize = length;
if (!isadir)
dep->de_flag |= DE_UPDATE | DE_MODIFIED;
allerror = vtruncbuf(DETOV(dep), length, pmp->pm_bpcluster);
#ifdef MSDOSFS_DEBUG
if (allerror)
kprintf("detrunc(): vtruncbuf error %d\n", allerror);
#endif
error = deupdat(dep, !DOINGASYNC((DETOV(dep))));
if (error != 0 && allerror == 0)
allerror = error;
mprintf("detrunc(): allerror %d, eofentry %lu\n",
allerror, eofentry);
if (eofentry != ~0) {
error = fatentry(FAT_GET_AND_SET, pmp, eofentry,
&chaintofree, CLUST_EOFE);
if (error) {
mprintf("detrunc(): fatentry errors %d\n", error);
return (error);
}
fc_setcache(dep, FC_LASTFC, de_cluster(pmp, length - 1),
eofentry);
}
if (chaintofree != 0 && !MSDOSFSEOF(pmp, chaintofree))
freeclusterchain(pmp, chaintofree);
return (allerror);
}
int
deextend(struct denode *dep, u_long length)
{
struct msdosfsmount *pmp = dep->de_pmp;
u_long count;
int error;
if ((DETOV(dep)->v_flag & VROOT) && !FAT32(pmp))
return (EINVAL);
if (dep->de_Attributes & ATTR_DIRECTORY)
return (EISDIR);
if (length <= dep->de_FileSize)
panic("deextend: file too large");
count = de_clcount(pmp, length) - de_clcount(pmp, dep->de_FileSize);
if (count > 0) {
if (count > pmp->pm_freeclustercount)
return (ENOSPC);
error = extendfile(dep, count, NULL, NULL, DE_CLEAR);
if (error) {
detrunc(dep, dep->de_FileSize, 0);
return (error);
}
}
dep->de_FileSize = length;
dep->de_flag |= DE_UPDATE | DE_MODIFIED;
return (deupdat(dep, !DOINGASYNC(DETOV(dep))));
}
int
msdosfs_reclaim(struct vop_reclaim_args *ap)
{
struct vnode *vp = ap->a_vp;
struct denode *dep = VTODE(vp);
mprintf("msdosfs_reclaim(): dep %p, file %s, refcnt %ld\n",
dep, dep ? (char *)dep->de_Name : "?", dep ? dep->de_refcnt : -1);
if (prtactive && VREFCNT(vp) > 1)
vprint("msdosfs_reclaim(): pushing active", vp);
vp->v_data = NULL;
if (dep) {
msdosfs_hashrem(dep);
if (dep->de_devvp) {
vrele(dep->de_devvp);
dep->de_devvp = 0;
}
kfree(dep, M_MSDOSFSNODE);
}
return (0);
}
int
msdosfs_inactive(struct vop_inactive_args *ap)
{
struct vnode *vp = ap->a_vp;
struct denode *dep = VTODE(vp);
int error = 0;
mprintf("msdosfs_inactive(): dep %p, de_Name[0] %x\n",
dep, (dep ? dep->de_Name[0] : 0));
if (prtactive && VREFCNT(vp) > 1)
vprint("msdosfs_inactive(): pushing active", vp);
if (dep == NULL || dep->de_Name[0] == SLOT_DELETED || dep->de_Name[0] == SLOT_EMPTY)
goto out;
mprintf("msdosfs_inactive(): dep %p, refcnt %ld, mntflag %llx, MNT_RDONLY %llx\n",
dep, dep->de_refcnt, (unsigned long long)vp->v_mount->mnt_flag,
(unsigned long long)MNT_RDONLY);
if (dep->de_refcnt <= 0 && (vp->v_mount->mnt_flag & MNT_RDONLY) == 0) {
error = detrunc(dep, (u_long) 0, 0);
dep->de_flag |= DE_UPDATE;
dep->de_Name[0] = SLOT_DELETED;
}
deupdat(dep, 0);
out:
mprintf("msdosfs_inactive(): v_refcnt 0x%08x, de_Name[0] %x\n",
vp->v_refcnt, (dep ? dep->de_Name[0] : 0));
if (dep == NULL || dep->de_Name[0] == SLOT_DELETED || dep->de_Name[0] == SLOT_EMPTY)
vrecycle(vp);
return (error);
}