实现死锁检测组件

最新推荐文章于 2024-05-28 14:22:00 发布

闲谈社

最新推荐文章于 2024-05-28 14:22:00 发布

阅读量427

点赞数

分类专栏： Linux基础组件文章标签：死锁图搜索 linux

本文链接：https://blog.csdn.net/Ricardo2/article/details/131612005

版权

Linux基础组件专栏收录该内容

11 篇文章 2 订阅

订阅专栏

一、死锁

1.1 死锁的定义

死锁，是指多个线程或者进程在运行过程中因争夺资源而造成的一种僵局，当进程或者线程处于这种僵持状态，若无外力作用，它们将无法再向前推进。

如下图所示，以申请锁资源为例。线程 A 想申请线程 B 的锁，线程 B 想申请线程 C 的锁，线程 C 想申请线程 D 的锁，线程 D 想申请线程 A 的锁，从而构建了一个资源申请环。
在这里插入图片描述
多个线程之间依次想要访问其他线程的资源，这样相互僵持形成的一个资源申请闭环。
死锁的存在是因为有资源申请环的存在，所以只要能检测出资源申请环，就等同于检测出
死锁的存在。

1.2 死锁产生的条件

1.条件互斥：进程/线程要求对所分配的资源进行排它性控制，即在一段时间内某资源仅为一进程/线程所占用。
2.请求与保持：当进程/线程因请求资源而阻塞时，对已获得的资源保持不放。
3.不剥夺：进程/线程已获得的资源在未使用完之前，不能剥夺，只能在使用完时由自己释放。
4.环路等待：在发生死锁时，必然存在一个进程/线程——资源的环形链。

二、死锁检测

为了能检测系统是否发生死锁，需要有两点

用某种数据结构来保存资源的请求和分配信息；
提供一种算法，利用上述信息检测系统是否进入死锁状态。

2.1 资源分配图

在资源分配图中，有两种结点，两种边
1）两种结点

进程结点：对应一个进程
资源结点：对应一类资源，资源可能有多个

2）两种边

进程结点 ——> 资源结点：表示进程想申请几个资源（每条边代表一个）
资源结点 ——> 进程结点：表示已经为进程分配了几个资源（每条边代表一个）

在这里插入图片描述
如图，顶点表示进程；黄色边代表资源申请边；红色边代表资源获取边。
每个进程获取了自己的资源（锁），同时又申请了其他线程的资源（锁）。

上图展示了一个死锁，检测死锁的关键是检测资源申请边有没有构成一回路。

我们可以采用有向图来存储资源申请环，则问题转化为检测有向图是否成环。

2.2 有向图

有向图的构建有矩阵、邻接表两种方式，本文选择邻接表。
如，有向图
在这里插入图片描述
其邻接表

2.3 死锁检测组件 – hook

死锁检测的流程可大致归纳为：
1）hook拦截系统的锁接口（pthread_mutex_lock和pthread_mutex_unlock）；
2）在hook后的锁接口中，利用有向图存储资源分配图的信息
线程A申请线程B的资源锁：若锁未被占有，则占有这个锁；若锁被占有，构建线程A指向线程B的申请边。
3）创建一个线程，定时检测图进程是否有环的存在。

hook类似于C++中的重载。通过hook，系统的pthread_mutex_lock和pthread_mutex_unlock，将指向我们自己定义的函数。

hook重载的函数中，把线程与锁的申请关系构建成有向图。通过线程加锁前、加锁后、释放锁，这3个阶段维护有向图
1）加锁前：检测当前申请的锁是否被占用

若被占用：构建当前线程指向占有锁的线程之间的申请边
若未被占用：继续执行后续（占有这个锁）

2）加锁后：占有这个锁之后，检测

该锁在之前没有被占有过：建立线程id与锁id的对应关系
该锁在之前被占有过：释放之前创建的申请边，建立线程id与锁id的对应关系
该锁在之前没被占有过：建立线程id与锁id的对应关系

3）释放锁：清除锁与线程的对应关系

三、死锁设计

3.1 有向图

3.1.1 数据结构

线程既可以拥有资源，也可以申请资源，资源绑定线程。为了实现复用，增加了 type 字段，当 type = RESOURCE，该结构体作为资源使用，放入资源链表； type = PROCESS，该结构体作为线程使用。

1）顶点资源信息，即线程所拥有的资源信息

// 顶点资源信息
struct source_type {

	uint64 id;			// 拥有该资源的线程 id
	enum Type type;		// 顶点类型：线程 or 资源
	uint64 lock_id;		// 资源（锁） id
	int degress;	    // 资源的出度，该资源被多少顶点（线程）申请
};

2）顶点，表示线程；图的资源申请边，表示线程间的资源申请关系。

// 顶点信息
struct vertex {

	struct source_type s;	// 顶点资源信息
	struct vertex *next;	// 指向下一个顶点（邻接表）

};

3）图的结构，资源 (type = RESOURCE) 存储在资源链表，线程（type = PROCESS) 作为图上的顶点。

// 图结构
struct task_graph {

	struct vertex list[MAX];			// 存储顶点于数组中
	int num;							// 顶点的数量

	struct source_type locklist[MAX];	// 存储所有顶点资源（锁）
	int lockidx; 						// 资源（锁）的数量

	pthread_mutex_t mutex;
};

3.1.2 图的基本操作接口

1）创建顶点：create_vertex(struct source_type type)
2）增加顶点：add_vertex(struct source_type type)
3）寻找与资源对应的顶点所在的下标：search_vertex(struct source_type type)
4）增加一条资源申请边：add_edge(struct source_type from, struct source_type to)
5）检查资源i和j之间是否存在边：verify_edge(struct source_type i, struct source_type j)
6）删除资源from到to之间的边：remove_edge(struct source_type from, struct source_type to)


// 创建结点
struct vertex *create_vertex(struct source_type type) {

	struct vertex *tex = (struct vertex *)malloc(sizeof(struct vertex ));

	tex->s = type;
	tex->next = NULL;

	return tex;

}

// 寻找与资源对应的顶点所在的下标
int search_vertex(struct source_type type) {

	int i = 0;

	for (i = 0;i < tg->num;i ++) {

		if (tg->list[i].s.type == type.type && tg->list[i].s.id == type.id) {
			return i;
		}

	}

	return -1;
}

// 增加节点
void add_vertex(struct source_type type) {

	if (search_vertex(type) == -1) {

		tg->list[tg->num].s = type;
		tg->list[tg->num].next = NULL;
		tg->num ++;

	}

}

// 增加一条边
int add_edge(struct source_type from, struct source_type to) {

	add_vertex(from);
	add_vertex(to);

	struct vertex *v = &(tg->list[search_vertex(from)]);

	while (v->next != NULL) {
		v = v->next;
	}

	v->next = create_vertex(to);

}

// 检查资源i和j之间是否存在边
int verify_edge(struct source_type i, struct source_type j) {

	if (tg->num == 0) return 0;

	// 寻找资源i对应的结点下标
	int idx = search_vertex(i);
	if (idx == -1) {
		return 0;
	}

	struct vertex *v = &(tg->list[idx]);

	while (v != NULL) {

		if (v->s.id == j.id) return 1;

		v = v->next;
		
	}

	return 0;

}

// 删除资源from到to之间的边
int remove_edge(struct source_type from, struct source_type to) {

	int idxi = search_vertex(from);
	int idxj = search_vertex(to);

	if (idxi != -1 && idxj != -1) {

		struct vertex *v = &tg->list[idxi];
		struct vertex *remove;

		while (v->next != NULL) {

			if (v->next->s.id == to.id) {

				remove = v->next;
				v->next = v->next->next;

				free(remove);
				break;

			}

			v = v->next;
		}

	}

}

3.1.3 判断有向图是否成环

// 打印环  cycle : 1 --> 2 --> 3 --> 4 --> 2
void print_deadlock(void) {

	int i = 0;

	printf("cycle : ");
	for (i = 0;i < k-1;i ++) {

		printf("%ld --> ", tg->list[path[i]].s.id);

	}

	printf("%ld\n", tg->list[path[i]].s.id);

}

int DFS(int idx) {

	struct vertex *ver = &tg->list[idx];
	// 如果当前结点已经访问过，说明存在环
	if (visited[idx] == 1) {

		path[k++] = idx;
		print_deadlock();
		deadlock = 1;
		
		return 0;
	}

	// 否则将标记置为1，代表访问了，继续DFS
	visited[idx] = 1;
	path[k++] = idx;

	while (ver->next != NULL) {

		DFS(search_vertex(ver->next->s));
		k --;
		
		ver = ver->next;

	}

	return 1;

}

//从list[idx]的顶点开始，检测是否存在环
int search_for_cycle(int idx) {

	struct vertex *ver = &tg->list[idx];
	visited[idx] = 1;	// 标记idx处顶点被访问过
	k = 0;		
	path[k++] = idx;	// 记录起点

	while (ver->next != NULL) {

		// 把visited数组的开始到idx-1处的值都置为0
		int i = 0;
		for (i = 0;i < tg->num;i ++) {
			if (i == idx) continue;
			
			visited[i] = 0;
		}

		// 把path数组的第i到MAX处的值都置为-1
		for (i = 1;i <= MAX;i ++) {
			path[i] = -1;
		}
		k = 1;

		DFS(search_vertex(ver->next->s));
		ver = ver->next;
	}

}

3.2 hook

3.2.1 重载系统的锁接口

// ·······································hook··············································
// 1.定义原类型
typedef int (*pthread_mutex_lock_t)(pthread_mutex_t *mutex);
pthread_mutex_lock_t pthread_mutex_lock_f = NULL;

typedef int (*pthread_mutex_unlock_t)(pthread_mutex_t *mutex);
pthread_mutex_unlock_t pthread_mutex_unlock_f = NULL;

// 2.实现
int pthread_mutex_lock(pthread_mutex_t *mutex) {

	
	pthread_t selfid = pthread_self();
	
	// 获取锁之前，检测这个锁是否被占用。
	// 有的话就创建资源申请边，没有的话就跳过
	lock_before((uint64_t)selfid, (uint64_t)mutex);
	
	pthread_mutex_lock_f(mutex);

	// 获取锁之后，检测这个锁是否存在与资源列表中。
	// 存在的话，移除自己对这个锁的申请边，表示请求得到满足了；不存在的话，把这个锁加入资源列表中。
	lock_after((uint64_t)selfid, (uint64_t)mutex);

	
}

int pthread_mutex_unlock(pthread_mutex_t *mutex) {

	pthread_mutex_unlock_f(mutex);

	pthread_t selfid = pthread_self();
	unlock_after((uint64_t)selfid, (uint64_t)mutex);
	
}

// 3. 初始化
void init_hook(void) {

	if (!pthread_mutex_lock_f)
		pthread_mutex_lock_f = dlsym(RTLD_NEXT, "pthread_mutex_lock");

	if (!pthread_mutex_unlock_f)
		pthread_mutex_unlock_f = dlsym(RTLD_NEXT, "pthread_mutex_unlock");
	
}

3.2.2 加锁前——lock_before

获取资源（锁）前，检测该资源是否被其他线程占用。

如果被占用，则创建一条资源申请边，表示当前进程正在向拥有资源的线程申请该资源。
如果没有被占用，则跳过。

void lock_before(uint64_t tid, uint64_t lockaddr) {
	/*
	1. 	if (lockaddr) {
			tid --> lockaddr.tid;
	   	}
	*/

	int idx = 0;

	for (idx = 0;idx < tg->lockidx;idx ++) {

		// 1. 如果被占用
		if (tg->locklist[idx].lock_id == lockaddr) { 
			
			// 创建申请该资源的顶点（PROCESS类型）
			struct source_type from;
			from.id = tid;
			from.type = PROCESS;
			add_vertex(from);

			// 创建拥有该资源的顶点（PROCESS类型)
			struct source_type to;
			to.id = tg->locklist[idx].id;
			to.type = PROCESS;
			add_vertex(to);

			// 申请该资源的结点数量+1
			tg->locklist[idx].degress ++;

			// 如果两个顶点间不存在资源申请边，增加一条边
			if (!verify_edge(from, to))
				add_edge(from, to);

		}

	}
	
	
}

3.2.3 加锁前——lock_before

线程获取资源（锁）后，检查该资源是否存在（资源链表中是否存在）

若该资源之前不存在，添加资源到资源链表中
若该资源已经存在，则移除之前创建的自己对该资源的申请边，表示请求已经得到满足

void lock_after(uint64_t tid, uint64_t lockaddr) {

	/*
		if (!lockaddr) {

			tid --> lockaddr;

		} else {

			lockaddr.tid = tid;
			tid -> lockaddr;

		}
		
	 */
	int idx = 0;
	// 1. 若该资源（锁）不存在，添加资源到资源链表中
	if (-1 == (idx = search_lock(lockaddr))) {// 

		// 寻找资源链表中空闲的位置并添加该资源
		int eidx = search_empty_lock(lockaddr);

		tg->locklist[eidx].id = tid;
		tg->locklist[eidx].lock_id = lockaddr;

		// 资源数量+1
		tg->lockidx ++;
		
	} 
	else {	// 2. 该资源（锁）存在，需要移除自己的请求边

		// 申请该资源的顶点（PROCESS类型）
		struct source_type from;
		from.id = tid;
		from.type = PROCESS;
		add_vertex(from);

		// 拥有该资源的顶点（PROCESS类型）
		struct source_type to;
		to.id = tg->locklist[idx].id;
		to.type = PROCESS;
		add_vertex(to);

		// 申请该资源的顶点数-1
		tg->locklist[idx].degress --;

		// 如果存在该资源申请边，则删除
		if (verify_edge(from, to))
			remove_edge(from, to);

		// 线程占用该资源（锁）
		tg->locklist[idx].id = tid;
		
	}
	 
	
}

3.2.4 加锁后——unlock_after

线程释放该资源后，检查该资源是否还被其他线程申请，没有则将其从资源链表中移除。

void unlock_after(uint64_t tid, uint64_t lockaddr) {

	// lockaddr.tid = 0;
	// 此锁已被unlock
	int idx = search_lock(lockaddr);

	// 若该资源没有线程申请，则将其从资源链表中移除
	if (tg->locklist[idx].degress == 0) {
		tg->locklist[idx].id = 0;
		tg->locklist[idx].lock_id = 0;
	}
	
}

3.3 死锁检测线程

创建死锁检测线程，在程序的运行期间时刻监控线程与锁之间的关系。

// 检测死锁
void check_dead_lock(void) {

	int i = 0;

	deadlock = 0;
	for (i = 0;i < tg->num;i ++) {
		if (deadlock == 1) break;
		// 检测是否存在环，即死锁
		search_for_cycle(i);
	}

	if (deadlock == 0) {
		printf("no deadlock\n");
	}

}

// 检测死锁的线程
static void *thread_routine(void *args) {

	while (1) {

		sleep(5);	//每隔5秒检测一次
		check_dead_lock();

	}

}

// 开启死锁检测
void start_check(void) {

	tg = (struct task_graph*)malloc(sizeof(struct task_graph));
	tg->num = 0;
	tg->lockidx = 0;
	
	pthread_t tid;

	// 创建检测死锁的线程
	pthread_create(&tid, NULL, thread_routine, NULL);

}

3.4 画图解释

1、
在这里插入图片描述
2、

3、

四、代码实现及测试

创建5个线程，1申请2、2申请3、3申请4、4申请5、5申请1
出现死锁
在这里插入图片描述

修改1申请2、2申请3、3申请4、4申请5
检测完成，未出现死锁
在这里插入图片描述


// build: gcc -o deadlock deadlock.c -lpthread -ldl


#define _GNU_SOURCE
#include <dlfcn.h>


#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <unistd.h>

#include <stdint.h>


//·········································有向图的结构和基本操作接口····························································
#if 1 	// 图结构


typedef unsigned long int uint64;


#define MAX		100

enum Type {PROCESS, RESOURCE};

// 顶点资源信息
struct source_type {

	uint64 id;			// 拥有该资源的线程 id
	enum Type type;		// 顶点类型：线程 or 资源
	uint64 lock_id;		// 资源（锁） id
	int degress;	    // 资源的出度，该资源被多少顶点（线程）申请
};

// 顶点信息
struct vertex {

	struct source_type s;	// 顶点资源信息
	struct vertex *next;	// 指向下一个顶点（邻接表）

};

// 图结构
struct task_graph {

	struct vertex list[MAX];			// 存储顶点于数组中
	int num;							// 顶点的数量

	struct source_type locklist[MAX];	// 存储所有顶点资源（锁）
	int lockidx; 						// 资源（锁）的数量

	pthread_mutex_t mutex;
};

struct task_graph *tg = NULL;
int path[MAX+1];
int visited[MAX];
int k = 0;
int deadlock = 0;	// 死锁标记

// 创建顶点
struct vertex *create_vertex(struct source_type type) {

	struct vertex *tex = (struct vertex *)malloc(sizeof(struct vertex ));

	tex->s = type;
	tex->next = NULL;

	return tex;

}

// 寻找与资源对应的顶点所在的下标
int search_vertex(struct source_type type) {

	int i = 0;

	for (i = 0;i < tg->num;i ++) {

		if (tg->list[i].s.type == type.type && tg->list[i].s.id == type.id) {
			return i;
		}

	}

	return -1;
}

// 增加节点
void add_vertex(struct source_type type) {

	if (search_vertex(type) == -1) {

		tg->list[tg->num].s = type;
		tg->list[tg->num].next = NULL;
		tg->num ++;

	}

}

// 增加一条边
int add_edge(struct source_type from, struct source_type to) {

	add_vertex(from);
	add_vertex(to);

	struct vertex *v = &(tg->list[search_vertex(from)]);

	while (v->next != NULL) {
		v = v->next;
	}

	v->next = create_vertex(to);

}

// 检查资源i和j之间是否存在边
int verify_edge(struct source_type i, struct source_type j) {

	if (tg->num == 0) return 0;

	// 寻找资源i对应的结点下标
	int idx = search_vertex(i);
	if (idx == -1) {
		return 0;
	}

	struct vertex *v = &(tg->list[idx]);

	while (v != NULL) {

		if (v->s.id == j.id) return 1;

		v = v->next;
		
	}

	return 0;

}

// 删除资源from到to之间的边
int remove_edge(struct source_type from, struct source_type to) {

	int idxi = search_vertex(from);
	int idxj = search_vertex(to);

	if (idxi != -1 && idxj != -1) {

		struct vertex *v = &tg->list[idxi];
		struct vertex *remove;

		while (v->next != NULL) {

			if (v->next->s.id == to.id) {

				remove = v->next;
				v->next = v->next->next;

				free(remove);
				break;

			}

			v = v->next;
		}

	}

}

// 打印环  cycle : 1 --> 2 --> 3 --> 4 --> 2
void print_deadlock(void) {

	int i = 0;

	printf("deadlock : ");
	for (i = 0;i < k-1;i ++) {

		printf("%ld --> ", tg->list[path[i]].s.id);

	}

	printf("%ld\n", tg->list[path[i]].s.id);

}

int DFS(int idx) {

	struct vertex *ver = &tg->list[idx];
	// 如果当前结点已经访问过，说明存在环
	if (visited[idx] == 1) {

		path[k++] = idx;
		print_deadlock();
		deadlock = 1;
		
		return 0;
	}

	// 否则将标记置为1，代表访问了，继续DFS
	visited[idx] = 1;
	path[k++] = idx;

	while (ver->next != NULL) {

		DFS(search_vertex(ver->next->s));
		k --;
		
		ver = ver->next;

	}

	return 1;

}

//从list[idx]的顶点开始，检测是否存在环
int search_for_cycle(int idx) {

	struct vertex *ver = &tg->list[idx];
	visited[idx] = 1;	// 标记idx处顶点被访问过
	k = 0;		
	path[k++] = idx;	// 记录起点

	while (ver->next != NULL) {

		// 把visited数组的开始到idx-1处的值都置为0
		int i = 0;
		for (i = 0;i < tg->num;i ++) {
			if (i == idx) continue;
			
			visited[i] = 0;
		}

		// 把path数组的第i到MAX处的值都置为-1
		for (i = 1;i <= MAX;i ++) {
			path[i] = -1;
		}
		k = 1;

		DFS(search_vertex(ver->next->s));
		ver = ver->next;
	}

}


#endif

// 



#if 1

// 搜索锁lock是否在资源列表中，若有则返回下标位置
int search_lock(uint64 lock) {

	int i = 0;
	
	for (i = 0;i < tg->lockidx;i ++) {
		//如果lock存在就返回它在锁列表中的下标位置
		if (tg->locklist[i].lock_id == lock) {
			return i;
		}
	}

	return -1;
}

// 寻找资源列表中的一个空位置
int search_empty_lock(uint64 lock) {

	int i = 0;
	
	for (i = 0;i < tg->lockidx;i ++) {
		
		if (tg->locklist[i].lock_id == 0) {
			return i;
		}
	}

	return tg->lockidx;

}


/*
获取资源（锁）前，检测该资源是否被其他线程占用。
1. 如果被占用，则创建一条资源申请边，表示当前进程正在向拥有资源的线程申请该资源。
2. 如果没有被占用，则跳过。
*/
void lock_before(uint64_t tid, uint64_t lockaddr) {
	/*
	1. 	if (lockaddr) {
			tid --> lockaddr.tid;
	   	}
	*/

	int idx = 0;

	for (idx = 0;idx < tg->lockidx;idx ++) {

		// 1. 如果被占用
		if (tg->locklist[idx].lock_id == lockaddr) { 
			
			// 创建申请该资源的顶点（PROCESS类型）
			struct source_type from;
			from.id = tid;
			from.type = PROCESS;
			add_vertex(from);

			// 创建拥有该资源的顶点（PROCESS类型)
			struct source_type to;
			to.id = tg->locklist[idx].id;
			to.type = PROCESS;
			add_vertex(to);

			// 申请该资源的结点数量+1
			tg->locklist[idx].degress ++;

			// 如果两个顶点间不存在资源申请边，增加一条边
			if (!verify_edge(from, to))
				add_edge(from, to);

		}

	}
	
	
}


/*
线程获取资源（锁）后，检查该资源是否存在（资源链表中是否存在）
1. 若该资源之前不存在，添加资源到资源链表中
2. 若该资源已经存在，则移除自己对该资源的申请边，表示请求已经得到满足
*/
void lock_after(uint64_t tid, uint64_t lockaddr) {

	/*
		if (!lockaddr) {

			tid --> lockaddr;

		} else {

			lockaddr.tid = tid;
			tid -> lockaddr;

		}
		
	 */
	int idx = 0;
	// 1. 若该资源（锁）不存在，添加资源到资源链表中
	if (-1 == (idx = search_lock(lockaddr))) {// 

		// 寻找资源链表中空闲的位置并添加该资源
		int eidx = search_empty_lock(lockaddr);

		tg->locklist[eidx].id = tid;
		tg->locklist[eidx].lock_id = lockaddr;

		// 资源数量+1
		tg->lockidx ++;
		
	} 
	else {	// 2. 该资源（锁）存在，需要移除自己的请求边

		// 申请该资源的顶点（PROCESS类型）
		struct source_type from;
		from.id = tid;
		from.type = PROCESS;
		add_vertex(from);

		// 拥有该资源的顶点（PROCESS类型）
		struct source_type to;
		to.id = tg->locklist[idx].id;
		to.type = PROCESS;
		add_vertex(to);

		// 申请该资源的顶点数-1
		tg->locklist[idx].degress --;

		// 如果存在该资源申请边，则删除
		if (verify_edge(from, to))
			remove_edge(from, to);

		// 线程占用该资源（锁）
		tg->locklist[idx].id = tid;
		
	}
	 
	
}

// 线程释放该资源后，检查该资源是否还被线程申请，没有则将其从资源链表中移除。
void unlock_after(uint64_t tid, uint64_t lockaddr) {

	// lockaddr.tid = 0;
	// 此锁已被unlock
	int idx = search_lock(lockaddr);

	// 若该资源没有线程申请，则将其从资源链表中移除
	if (tg->locklist[idx].degress == 0) {
		tg->locklist[idx].id = 0;
		tg->locklist[idx].lock_id = 0;
	}
	
}

// ····································死锁检测的线程·········································

// 检测死锁
void check_dead_lock(void) {

	int i = 0;

	deadlock = 0;
	for (i = 0;i < tg->num;i ++) {
		if (deadlock == 1) break;
		// 检测是否存在环，即死锁
		search_for_cycle(i);
	}

	if (deadlock == 0) {
		printf("no deadlock\n");
	}

}

// 检测死锁的线程
static void *thread_routine(void *args) {

	while (1) {

		sleep(5);	//每隔5秒检测一次
		check_dead_lock();

	}

}

// 开启死锁检测
void start_check(void) {

	tg = (struct task_graph*)malloc(sizeof(struct task_graph));
	tg->num = 0;
	tg->lockidx = 0;
	
	pthread_t tid;

	// 创建检测死锁的线程
	pthread_create(&tid, NULL, thread_routine, NULL);

}
// ··························································································

// ·······································hook··············································
// 1.定义原类型
typedef int (*pthread_mutex_lock_t)(pthread_mutex_t *mutex);
pthread_mutex_lock_t pthread_mutex_lock_f = NULL;

typedef int (*pthread_mutex_unlock_t)(pthread_mutex_t *mutex);
pthread_mutex_unlock_t pthread_mutex_unlock_f = NULL;

// 2.实现
int pthread_mutex_lock(pthread_mutex_t *mutex) {

	
	pthread_t selfid = pthread_self();
	
	// 获取锁之前，检测这个锁是否被占用。
	// 有的话就创建资源申请边，没有的话就跳过
	lock_before((uint64_t)selfid, (uint64_t)mutex);
	
	pthread_mutex_lock_f(mutex);

	// 获取锁之后，检测这个锁是否存在与资源列表中。
	// 存在的话，移除自己对这个锁的申请边，表示请求得到满足了；不存在的话，把这个锁加入资源列表中。
	lock_after((uint64_t)selfid, (uint64_t)mutex);

	
}

int pthread_mutex_unlock(pthread_mutex_t *mutex) {

	pthread_mutex_unlock_f(mutex);

	pthread_t selfid = pthread_self();
	unlock_after((uint64_t)selfid, (uint64_t)mutex);
	
}

// 3. 初始化
void init_hook(void) {

	if (!pthread_mutex_lock_f)
		pthread_mutex_lock_f = dlsym(RTLD_NEXT, "pthread_mutex_lock");

	if (!pthread_mutex_unlock_f)
		pthread_mutex_unlock_f = dlsym(RTLD_NEXT, "pthread_mutex_unlock");
	
}

// ··························································································

#endif

// 

#if 1//sample

pthread_mutex_t r1 = PTHREAD_MUTEX_INITIALIZER;
pthread_mutex_t r2 = PTHREAD_MUTEX_INITIALIZER;
pthread_mutex_t r3 = PTHREAD_MUTEX_INITIALIZER;
pthread_mutex_t r4 = PTHREAD_MUTEX_INITIALIZER;
pthread_mutex_t r5 = PTHREAD_MUTEX_INITIALIZER;



void *t1_cb(void *arg) {
	pthread_t selfid = pthread_self(); 
	printf("thread_routine 1 : %ld \n", selfid);

	pthread_mutex_lock(&r1);
	sleep(1);	// 休眠，允许期间其他线程能运行
	pthread_mutex_lock(&r2);
	
	pthread_mutex_unlock(&r2);
	pthread_mutex_unlock(&r1);

}

void *t2_cb(void *arg) {
	pthread_t selfid = pthread_self(); 
	printf("thread_routine 2 : %ld \n", selfid);

	pthread_mutex_lock(&r2);
	sleep(1);
	pthread_mutex_lock(&r3);

	pthread_mutex_unlock(&r3);
	pthread_mutex_unlock(&r2);

}

void *t3_cb(void *arg) {
	pthread_t selfid = pthread_self(); 
	printf("thread_routine 3 : %ld \n", selfid);
	
	pthread_mutex_lock(&r3);
	sleep(1);
	pthread_mutex_lock(&r4);

	pthread_mutex_unlock(&r4);
	pthread_mutex_unlock(&r3);

}

void *t4_cb(void *arg) {
	pthread_t selfid = pthread_self(); 
	printf("thread_routine 4 : %ld \n", selfid);
	
	pthread_mutex_lock(&r4);
	sleep(1);
	pthread_mutex_lock(&r5);

	pthread_mutex_unlock(&r5);
	pthread_mutex_unlock(&r4);

}

void *t5_cb(void *arg) {
	pthread_t selfid = pthread_self(); 
	printf("thread_routine 5 : %ld \n", selfid);
	
	pthread_mutex_lock(&r5);
	sleep(1);
	pthread_mutex_lock(&r1);

	pthread_mutex_unlock(&r1);
	pthread_mutex_unlock(&r5);

}




// deadlock
// 

int main() {

	init_hook();

	// 启动检测死锁的线程
	start_check();
	printf("start_check\n");

	pthread_t t1, t2, t3, t4, t5;

	pthread_create(&t1, NULL, t1_cb, NULL);
	pthread_create(&t2, NULL, t2_cb, NULL);
	pthread_create(&t3, NULL, t3_cb, NULL);
	pthread_create(&t4, NULL, t4_cb, NULL);
	pthread_create(&t5, NULL, t5_cb, NULL);


	pthread_join(t1, NULL);
	pthread_join(t2, NULL);
	pthread_join(t3, NULL);
	pthread_join(t4, NULL);
	pthread_join(t5, NULL);

	printf("complete\n");

}

#endif

闲谈社

关注

0
点赞
踩
2

收藏

觉得还不错? 一键收藏
打赏
0
评论
实现死锁检测组件

死锁，是指多个线程或者进程在运行过程中因争夺资源而造成的一种僵局，当进程或者线程处于这种僵持状态，若无外力作用，它们将无法再向前推进。如下图所示，以申请锁资源为例。线程 A 想申请线程 B 的锁，线程 B 想申请线程 C 的锁，线程 C 想申请线程 D 的锁，线程 D 想申请线程 A 的锁，从而构建了一个资源申请环。多个线程之间依次想要访问其他线程的资源，这样相互僵持形成的一个资源申请闭环。死锁的存在是因为有资源申请环的存在，所以只要能检测出资源申请环，就等同于检测出死锁的存在。
复制链接

扫一扫