/* (C)opyright MMIV-MMVI Anselm R. Garbe <garbeam at gmail dot com>
 * (C)opyright MMVI Kris Maglione <fbsdaemon at gmail dot com>
 * See LICENSE file for license details.
 */

#include <sys/types.h>

#define uchar	_ixpuchar
#define ushort	_ixpushort
#define uint	_ixpuint
#define ulong	_ixpulong
#define vlong	_ixpvlong
#define uvlong	_ixpuvlong
typedef unsigned char	uchar;
typedef unsigned short	ushort;
typedef unsigned int	uint;
typedef unsigned long	ulong;
typedef long long	vlong;
typedef unsigned long long	uvlong;

#define IXP_VERSION	"9P2000"
#define IXP_NOTAG	(ushort)~0U	/*Dummy tag */
#define IXP_NOFID	(uint)~0	/*No auth */

enum {	IXP_MAX_VERSION = 32,
	IXP_MAX_ERROR = 128,
	IXP_MAX_CACHE = 32,
	IXP_MAX_MSG = 8192,
	IXP_MAX_FLEN = 128,
	IXP_MAX_ULEN = 32,
	IXP_MAX_WELEM = 16 };

/* 9P message types */
enum {	TVERSION = 100,
	RVERSION,
	TAUTH = 102,
	RAUTH,
	TATTACH = 104,
	RATTACH,
	TERROR = 106, /* illegal */
	RERROR,
	TFLUSH = 108,
	RFLUSH,
	TWALK = 110,
	RWALK,
	TOPEN = 112,
	ROPEN,
	TCREATE = 114,
	RCREATE,
	TREAD = 116,
	RREAD,
	TWRITE = 118,
	RWRITE,
	TCLUNK = 120,
	RCLUNK,
	TREMOVE = 122,
	RREMOVE,
	TSTAT = 124,
	RSTAT,
	TWSTAT = 126,
	RWSTAT,
};

/* modes */
enum {	IXP_OREAD = 0x00,
	IXP_OWRITE = 0x01,
	IXP_ORDWR = 0x02,
	IXP_OEXEC = 0x03,
	IXP_OEXCL = 0x04,
	IXP_OTRUNC = 0x10,
	IXP_OREXEC = 0x20,
	IXP_ORCLOSE = 0x40,
	IXP_OAPPEND = 0x80,
};

/* qid.types */
enum {	IXP_QTDIR = 0x80,
	IXP_QTAPPEND = 0x40,
	IXP_QTEXCL = 0x20,
	IXP_QTMOUNT = 0x10,
	IXP_QTAUTH = 0x08,
	IXP_QTTMP = 0x04,
	IXP_QTSYMLINK = 0x02,
	IXP_QTLINK = 0x01,
	IXP_QTFILE = 0x00,
};

/* from libc.h in p9p */
enum {	P9OREAD		= 0,	/* open for read */
	P9OWRITE	= 1,	/* write */
	P9ORDWR		= 2,	/* read and write */
	P9OEXEC		= 3,	/* execute, == read but check execute permission */
	P9OTRUNC	= 16,	/* or'ed in (except for exec), truncate file first */
	P9OCEXEC	= 32,	/* or'ed in, close on exec */
	P9ORCLOSE	= 64,	/* or'ed in, remove on close */
	P9ODIRECT	= 128,	/* or'ed in, direct access */
	P9ONONBLOCK	= 256,	/* or'ed in, non-blocking call */
	P9OEXCL		= 0x1000,	/* or'ed in, exclusive use (create only) */
	P9OLOCK		= 0x2000,	/* or'ed in, lock after opening */
	P9OAPPEND		= 0x4000	/* or'ed in, append only */
};

/* bits in Qid.type */
enum {	P9QTDIR		= 0x80,		/* type bit for directories */
	P9QTAPPEND	= 0x40,		/* type bit for append only files */
	P9QTEXCL	= 0x20,		/* type bit for exclusive use files */
	P9QTMOUNT	= 0x10,		/* type bit for mounted channel */
	P9QTAUTH	= 0x08,		/* type bit for authentication file */
	P9QTTMP		= 0x04,		/* type bit for non-backed-up file */
	P9QTSYMLINK	= 0x02,		/* type bit for symbolic link */
	P9QTFILE	= 0x00		/* type bits for plain file */
};

/* bits in Dir.mode */
#define P9DMDIR		0x80000000	/* mode bit for directories */
#define P9DMAPPEND	0x40000000	/* mode bit for append only files */
#define P9DMEXCL	0x20000000	/* mode bit for exclusive use files */
#define P9DMMOUNT	0x10000000	/* mode bit for mounted channel */
#define P9DMAUTH	0x08000000	/* mode bit for authentication file */
#define P9DMTMP		0x04000000	/* mode bit for non-backed-up file */
#define P9DMSYMLINK	0x02000000	/* mode bit for symbolic link (Unix, 9P2000.u) */
#define P9DMDEVICE	0x00800000	/* mode bit for device file (Unix, 9P2000.u) */
#define P9DMNAMEDPIPE	0x00200000	/* mode bit for named pipe (Unix, 9P2000.u) */
#define P9DMSOCKET	0x00100000	/* mode bit for socket (Unix, 9P2000.u) */
#define P9DMSETUID	0x00080000	/* mode bit for setuid (Unix, 9P2000.u) */
#define P9DMSETGID	0x00040000	/* mode bit for setgid (Unix, 9P2000.u) */
	
enum {	P9DMREAD		= 0x4,		/* mode bit for read permission */
	P9DMWRITE		= 0x2,		/* mode bit for write permission */
	P9DMEXEC		= 0x1		/* mode bit for execute permission */
};


typedef struct Qid Qid;
struct Qid {
	uchar type;
	uint version;
	uvlong path;
	/* internal use only */
	uchar dir_type;
};

#include <ixp_fcall.h>

/* stat structure */
typedef struct Stat {
	ushort type;
	uint dev;
	Qid qid;
	uint mode;
	uint atime;
	uint mtime;
	uvlong length;
	char *name;
	char *uid;
	char *gid;
	char *muid;
} Stat;

typedef struct IXPServer IXPServer;
typedef struct IXPConn IXPConn;
typedef struct Intmap Intmap;

typedef struct Intlist Intlist;
struct Intmap {
	ulong nhash;
	Intlist	**hash;
};

struct IXPConn {
	IXPServer	*srv;
	void		*aux;
	int		fd;
	void		(*read) (IXPConn *);
	void		(*close) (IXPConn *);
	char		closed;

	/* Implementation details */
	/* do not use */
	IXPConn		*next;
};

struct IXPServer {
	int running;
	IXPConn *conn;
	int maxfd;
	fd_set rd;
};

typedef struct IXPClient {
	int fd;
	uint root_fid;
	Qid root_qid;
	Fcall ifcall;
	Fcall ofcall;
	char *errstr;
} IXPClient;

typedef struct P9Conn P9Conn;
typedef struct Fid {
	P9Conn		*conn;
	Intmap		*map;
	char		*uid;
	void		*aux;
	ulong	fid;
	Qid		qid;
	signed char	omode;
} Fid;

typedef struct P9Req P9Req;
struct P9Req {
	P9Conn	*conn;
	Fid	*fid;
	Fid	*newfid;
	P9Req	*oldreq;
	Fcall	ifcall;
	Fcall	ofcall;
	void	*aux;
};

typedef struct P9Srv {
	void (*attach)(P9Req *r);
	void (*clunk)(P9Req *r);
	void (*create)(P9Req *r);
	void (*flush)(P9Req *r);
	void (*open)(P9Req *r);
	void (*read)(P9Req *r);
	void (*remove)(P9Req *r);
	void (*stat)(P9Req *r);
	void (*walk)(P9Req *r);
	void (*write)(P9Req *r);
	void (*freefid)(Fid *f);
} P9Srv;

/* client.c */
extern int ixp_client_dial(IXPClient *c, char *address, uint rootfid);
extern void ixp_client_hangup(IXPClient *c);
extern int ixp_client_remove(IXPClient *c, uint newfid, char *filepath);
extern int ixp_client_create(IXPClient *c, uint dirfid, char *name,
		uint perm, uchar mode);
extern int ixp_client_walk(IXPClient *c, uint newfid, char *filepath);
extern int ixp_client_stat(IXPClient *c, uint newfid, char *filepath);
extern int ixp_client_walkopen(IXPClient *c, uint newfid, char *filepath,
		uchar mode);
extern int ixp_client_open(IXPClient *c, uint newfid, uchar mode);
extern int ixp_client_read(IXPClient *c, uint fid,
		uvlong offset, void *result,
		uint res_len);
extern int ixp_client_write(IXPClient *c, uint fid,
		uvlong offset,
		uint count, uchar *data);
extern int ixp_client_close(IXPClient *c, uint fid);
extern int ixp_client_do_fcall(IXPClient * c);

/* convert.c */
extern void ixp_pack_u8(uchar **msg, int *msize, uchar val);
extern void ixp_unpack_u8(uchar **msg, int *msize, uchar *val);
extern void ixp_pack_u16(uchar **msg, int *msize, ushort val);
extern void ixp_unpack_u16(uchar **msg, int *msize, ushort *val);
extern void ixp_pack_u32(uchar **msg, int *msize, uint val);
extern void ixp_unpack_u32(uchar **msg, int *msize, uint *val);
extern void ixp_pack_u64(uchar **msg, int *msize, uvlong val);
extern void ixp_unpack_u64(uchar **msg, int *msize, uvlong *val);
extern void ixp_pack_string(uchar **msg, int *msize, const char *s);
extern void ixp_unpack_strings(uchar **msg, int *msize, ushort n, char **strings);
extern void ixp_unpack_string(uchar **msg, int *msize, char **string, ushort *len);
extern void ixp_pack_data(uchar **msg, int *msize, uchar *data,
		uint datalen);
extern void ixp_unpack_data(uchar **msg, int *msize, uchar **data,
		uint datalen);
extern void ixp_pack_prefix(uchar *msg, uint size,
		uchar id, ushort tag);
extern void ixp_unpack_prefix(uchar **msg, uint *size,
		uchar *id, ushort *tag);
extern void ixp_pack_qid(uchar **msg, int *msize, Qid *qid);
extern void ixp_unpack_qid(uchar **msg, int *msize, Qid *qid);
extern void ixp_pack_stat(uchar **msg, int *msize, Stat *stat);
extern void ixp_unpack_stat(uchar **msg, int *msize, Stat *stat);

/* request.c */
extern void respond(P9Req *r, char *error);
extern void serve_9pcon(IXPConn *c);

/* intmap.c */
extern void initmap(Intmap *m, ulong nhash, void *hash);
extern void incref_map(Intmap *m);
extern void decref_map(Intmap *m);
extern void freemap(Intmap *map, void (*destroy)(void*));
extern void execmap(Intmap *map, void (*destroy)(void*));
extern void * lookupkey(Intmap *map, ulong id);
extern void * insertkey(Intmap *map, ulong id, void *v);
extern int caninsertkey(Intmap *map, ulong id, void *v);
extern void * deletekey(Intmap *map, ulong id);

/* message.c */
extern ushort ixp_sizeof_stat(Stat *stat);
extern uint ixp_fcall2msg(void *msg, Fcall *fcall, uint msglen);
extern uint ixp_msg2fcall(Fcall *call, void *msg, uint msglen);

/* server.c */
extern IXPConn *ixp_server_open_conn(IXPServer *s, int fd, void *aux,
		void (*read)(IXPConn *c), void (*close)(IXPConn *c));
extern void ixp_server_close_conn(IXPConn *c);
extern char *ixp_server_loop(IXPServer *s);
extern uint ixp_server_receive_fcall(IXPConn *c, Fcall *fcall);
extern int ixp_server_respond_fcall(IXPConn *c, Fcall *fcall);
extern int ixp_server_respond_error(IXPConn *c, Fcall *fcall, char *errstr);
extern void ixp_server_close(IXPServer *s);

/* socket.c */
extern int ixp_connect_sock(char *address);
extern int ixp_create_sock(char *address, char **errstr);

/* transport.c */
extern uint ixp_send_message(int fd, void *msg, uint msize, char **errstr);
extern uint ixp_recv_message(int fd, void *msg, uint msglen, char **errstr);

/* util.c */
extern void * ixp_emalloc(uint size);
extern void * ixp_emallocz(uint size);
extern void  ixp_eprint(const char *errstr, ...);
extern void * ixp_erealloc(void *ptr, uint size);
extern char * ixp_estrdup(const char *str);
extern uint ixp_tokenize(char **result, uint reslen, char *str, char delim);
extern uint ixp_strlcat(char *dst, const char *src, uint siz);