#include	<stdio.h>
#include	<sys/types.h>
#include	<sys/uio.h>
#include	<sys/fcntl.h>
#include	<unistd.h>
#include	<errno.h>

/* ロックファイルデータ構造体 */
typedef	struct	{
	int	pid;
	int	flag;
}LOCK_FILE;

/* プロセス内ロック情報 */
typedef struct	{
	int	flag;	/* 0:未使用 */
	char	table_name[256];
	int	fd;
}LOCK_DATA;
typedef struct	{
	LOCK_DATA	*lock;
	int	lock_size;
	int	lock_no;
}LOCK_MEMORY;
LOCK_MEMORY	LockMemory={NULL,0,0};

LOCK_DATA *LockMemoryGet(char *table_name,int create_flag)
{
int	i,no,free_no;
char	buf[512];

	/* ロック情報の検索 */
	no=-1;
	free_no=-1;
	for(i=0;i<LockMemory.lock_no;i++){
		if(LockMemory.lock[i].flag==0){	/* 空き */
			free_no=i;
		}
		else if(strcmp(LockMemory.lock[i].table_name,table_name)==0){
			no=i;
			break;
		}
	}

	/* 該当ロック情報無しで、作成フラグがオンの場合：ロック情報追加 */
	if(no==-1&&create_flag==1){
		if(free_no==-1){
			if(LockMemory.lock_size==0){
				LockMemory.lock_size=64;
				LockMemory.lock=(LOCK_DATA *)calloc(LockMemory.lock_size,sizeof(LOCK_DATA));
			}
			else{
				LockMemory.lock_size+=64;
				LockMemory.lock=(LOCK_DATA *)realloc(LockMemory.lock,LockMemory.lock_size*sizeof(LOCK_DATA));
			}
			no=LockMemory.lock_no;
			LockMemory.lock_no++;
		}
		else{
			no=free_no;
		}
		LockMemory.lock[no].flag=1;
		strcpy(LockMemory.lock[no].table_name,table_name);
		sprintf(buf,"%s.lock",table_name);
		LockMemory.lock[no].fd=open(buf,O_RDWR|O_CREAT,0666);
	}

	if(no==-1){
		return(NULL);
	}
	else{
		return(&LockMemory.lock[no]);
	}
}

int LockTable(char *table_name,int flag)
{
LOCK_DATA	*lock_data;
LOCK_FILE	lock_file;

	/* プロセス内ロック情報を得る */
	lock_data=LockMemoryGet(table_name,1);

	/* ロック：他プロセスがロックしていたら待つ */
	lockf(lock_data->fd,F_LOCK,0);

	/* 現在のロックファイル情報を取得 */
	lseek(lock_data->fd,0,SEEK_SET);
	lock_file.pid=0;
	lock_file.flag=0;
	read(lock_data->fd,&lock_file,sizeof(lock_file));

	if(lock_file.pid==getpid()){	/* 自プロセスの情報があり */
		if(lock_file.flag<flag){	/* 今回のロック情報の方が強ければセット */
			lock_file.flag=flag;
		}
	}
	else{	/* 他プロセスの情報がある：ごみ */
		lock_file.pid=getpid();
		lock_file.flag=flag;
	}

	/* ロックファイル情報を書き込み */
	lseek(lock_data->fd,0,SEEK_SET);
	write(lock_data->fd,&lock_file,sizeof(lock_file));

	return(1);
}

int UnlockTable(char *table_name,int flag)
{
LOCK_DATA	*lock_data;
LOCK_FILE	lock_file;

	/* プロセス内ロック情報を得る */
	lock_data=LockMemoryGet(table_name,0);
	if(lock_data!=NULL){	/* ロック情報あり */
		/* 現在のロックファイル情報を取得 */
		lseek(lock_data->fd,0,SEEK_SET);
		lock_file.pid=0;
		lock_file.flag=0;
		read(lock_data->fd,&lock_file,sizeof(lock_file));

		if(lock_file.pid==getpid()){	/* 自プロセスの情報があり */
			if(lock_file.flag<=flag){	/* 解除しようとするフラグと同じかそれ以下ならセット */
				lock_file.pid=0;
				lock_file.flag=0;
			}
		}
		else{	/* 他プロセスの情報がある：ごみ */
			lock_file.pid=0;
			lock_file.flag=0;
		}

		if(lock_file.flag==0){	/* アンロックになる場合 */
			/* ロックファイル情報を書き込み */
			lseek(lock_data->fd,0,SEEK_SET);
			write(lock_data->fd,&lock_file,sizeof(lock_file));

			/* ファイルのロック解除 */
			lockf(lock_data->fd,F_ULOCK,0);

			/* ロック情報クリア */
			close(lock_data->fd);
			lock_data->flag=0;
		}
	}

	return(1);
}

int TestTable(char *table_name)
{
LOCK_DATA	*lock_data;
LOCK_FILE	lock_file;
int	ret;

	/* プロセス内ロック情報を得る */
	lock_data=LockMemoryGet(table_name,1);
	ret=lockf(lock_data->fd,F_TEST,0);
	if(ret==-1){	/* 他プロセスがロック中 */
		/* ロック情報クリア */
		close(lock_data->fd);
		lock_data->flag=0;
		return(0);
	}
	else{	/* ロックされていない */
		/* 現在のロックファイル情報を取得 */
		lseek(lock_data->fd,0,SEEK_SET);
		lock_file.pid=0;
		lock_file.flag=0;
		read(lock_data->fd,&lock_file,sizeof(lock_file));
		if(lock_file.pid==getpid()){	/* 自プロセスがロック中 */
		}
		else{	/* 自プロセスはロックしていない */
			/* ロック情報クリア */
			close(lock_data->fd);
			lock_data->flag=0;
		}
		return(1);
	}
}