#ifndef _VX_VS_CONTEXT_H
#define _VX_VS_CONTEXT_H

#include <linux/kernel.h>
#include "vserver/debug.h"


#define get_vx_info(i)	__get_vx_info(i,__FILE__,__LINE__,__HERE__)

static inline struct vx_info *__get_vx_info(struct vx_info *vxi,
	const char *_file, int _line, void *_here)
{
	if (!vxi)
		return NULL;

	vxlprintk(VXD_CBIT(xid, 2), "get_vx_info(%p[#%d.%d])",
		vxi, vxi?vxi->vx_id:0, vxi?atomic_read(&vxi->vx_usecnt):0,
		_file, _line);
	__vxh_get_vx_info(vxi, _here);

	atomic_inc(&vxi->vx_usecnt);
	return vxi;
}


extern void free_vx_info(struct vx_info *);

#define put_vx_info(i)	__put_vx_info(i,__FILE__,__LINE__,__HERE__)

static inline void __put_vx_info(struct vx_info *vxi,
	const char *_file, int _line, void *_here)
{
	if (!vxi)
		return;

	vxlprintk(VXD_CBIT(xid, 2), "put_vx_info(%p[#%d.%d])",
		vxi, vxi?vxi->vx_id:0, vxi?atomic_read(&vxi->vx_usecnt):0,
		_file, _line);
	__vxh_put_vx_info(vxi, _here);

	if (atomic_dec_and_test(&vxi->vx_usecnt))
		free_vx_info(vxi);
}


#define init_vx_info(p,i) __init_vx_info(p,i,__FILE__,__LINE__,__HERE__)

static inline void __init_vx_info(struct vx_info **vxp, struct vx_info *vxi,
	const char *_file, int _line, void *_here)
{
	if (vxi) {
		vxlprintk(VXD_CBIT(xid, 3),
			"init_vx_info(%p[#%d.%d])",
			vxi, vxi?vxi->vx_id:0,
			vxi?atomic_read(&vxi->vx_usecnt):0,
			_file, _line);
		__vxh_init_vx_info(vxi, vxp, _here);

		atomic_inc(&vxi->vx_usecnt);
	}
	*vxp = vxi;
}


#define set_vx_info(p,i) __set_vx_info(p,i,__FILE__,__LINE__,__HERE__)

static inline void __set_vx_info(struct vx_info **vxp, struct vx_info *vxi,
	const char *_file, int _line, void *_here)
{
	struct vx_info *vxo;

	if (!vxi)
		return;

	vxlprintk(VXD_CBIT(xid, 3), "set_vx_info(%p[#%d.%d])",
		vxi, vxi?vxi->vx_id:0,
		vxi?atomic_read(&vxi->vx_usecnt):0,
		_file, _line);
	__vxh_set_vx_info(vxi, vxp, _here);

	atomic_inc(&vxi->vx_usecnt);
	vxo = xchg(vxp, vxi);
	BUG_ON(vxo);
}


#define clr_vx_info(p) __clr_vx_info(p,__FILE__,__LINE__,__HERE__)

static inline void __clr_vx_info(struct vx_info **vxp,
	const char *_file, int _line, void *_here)
{
	struct vx_info *vxo;

	vxo = xchg(vxp, NULL);
	if (!vxo)
		return;

	vxlprintk(VXD_CBIT(xid, 3), "clr_vx_info(%p[#%d.%d])",
		vxo, vxo?vxo->vx_id:0,
		vxo?atomic_read(&vxo->vx_usecnt):0,
		_file, _line);
	__vxh_clr_vx_info(vxo, vxp, _here);

	if (atomic_dec_and_test(&vxo->vx_usecnt))
		free_vx_info(vxo);
}


#define claim_vx_info(v,p) \
	__claim_vx_info(v,p,__FILE__,__LINE__,__HERE__)

static inline void __claim_vx_info(struct vx_info *vxi,
	struct task_struct *task,
	const char *_file, int _line, void *_here)
{
	vxlprintk(VXD_CBIT(xid, 3), "claim_vx_info(%p[#%d.%d.%d]) %p",
		vxi, vxi?vxi->vx_id:0,
		vxi?atomic_read(&vxi->vx_usecnt):0,
		vxi?atomic_read(&vxi->vx_tasks):0,
		task, _file, _line);
	__vxh_claim_vx_info(vxi, task, _here);

	atomic_inc(&vxi->vx_tasks);
}


extern void unhash_vx_info(struct vx_info *);

#define release_vx_info(v,p) \
	__release_vx_info(v,p,__FILE__,__LINE__,__HERE__)

static inline void __release_vx_info(struct vx_info *vxi,
	struct task_struct *task,
	const char *_file, int _line, void *_here)
{
	vxlprintk(VXD_CBIT(xid, 3), "release_vx_info(%p[#%d.%d.%d]) %p",
		vxi, vxi?vxi->vx_id:0,
		vxi?atomic_read(&vxi->vx_usecnt):0,
		vxi?atomic_read(&vxi->vx_tasks):0,
		task, _file, _line);
	__vxh_release_vx_info(vxi, task, _here);

	might_sleep();

	if (atomic_dec_and_test(&vxi->vx_tasks))
		unhash_vx_info(vxi);
}


#define task_get_vx_info(p) \
	__task_get_vx_info(p,__FILE__,__LINE__,__HERE__)

static inline struct vx_info *__task_get_vx_info(struct task_struct *p,
	const char *_file, int _line, void *_here)
{
	struct vx_info *vxi;

	task_lock(p);
	vxlprintk(VXD_CBIT(xid, 5), "task_get_vx_info(%p)",
		p, _file, _line);
	vxi = __get_vx_info(p->vx_info, _file, _line, _here);
	task_unlock(p);
	return vxi;
}


static inline void __wakeup_vx_info(struct vx_info *vxi)
{
	if (waitqueue_active(&vxi->vx_wait))
		wake_up_interruptible(&vxi->vx_wait);
}

extern void exit_vx_info(struct task_struct *, int);

static inline
struct task_struct *vx_child_reaper(struct task_struct *p)
{
	struct vx_info *vxi = p->vx_info;
	struct task_struct *reaper = child_reaper;

	if (!vxi)
		goto out;

	BUG_ON(!p->vx_info->vx_reaper);

	/* child reaper for the guest reaper */
	if (vxi->vx_reaper == p)
		goto out;

	reaper = vxi->vx_reaper;
out:
	return reaper;
}


#else
#warning duplicate inclusion
#endif