[deliverable/linux.git] / drivers / gpu / vga / vga_switcheroo.c

/*
 * Copyright (c) 2010 Red Hat Inc.
 * Author : Dave Airlie <airlied@redhat.com>
 *
 *
 * Licensed under GPLv2
 *
 * vga_switcheroo.c - Support for laptop with dual GPU using one set of outputs

 Switcher interface - methods require for ATPX and DCM
 - switchto - this throws the output MUX switch
 - discrete_set_power - sets the power state for the discrete card

 GPU driver interface
 - set_gpu_state - this should do the equiv of s/r for the card
		  - this should *not* set the discrete power state
 - switch_check  - check if the device is in a position to switch now
 */

#include <linux/module.h>
#include <linux/dmi.h>
#include <linux/seq_file.h>
#include <linux/uaccess.h>
#include <linux/fs.h>
#include <linux/debugfs.h>
#include <linux/fb.h>

#include <linux/pci.h>
#include <linux/vga_switcheroo.h>

#include <linux/vgaarb.h>

struct vga_switcheroo_client {
	struct pci_dev *pdev;
	struct fb_info *fb_info;
	int pwr_state;
	void (*set_gpu_state)(struct pci_dev *pdev, enum vga_switcheroo_state);
	void (*reprobe)(struct pci_dev *pdev);
	bool (*can_switch)(struct pci_dev *pdev);
	int id;
	bool active;
};

static DEFINE_MUTEX(vgasr_mutex);

struct vgasr_priv {

	bool active;
	bool delayed_switch_active;
	enum vga_switcheroo_client_id delayed_client_id;

	struct dentry *debugfs_root;
	struct dentry *switch_file;

	int registered_clients;
	struct vga_switcheroo_client clients[VGA_SWITCHEROO_MAX_CLIENTS];

	struct vga_switcheroo_handler *handler;
};

static int vga_switcheroo_debugfs_init(struct vgasr_priv *priv);
static void vga_switcheroo_debugfs_fini(struct vgasr_priv *priv);

/* only one switcheroo per system */
static struct vgasr_priv vgasr_priv;

int vga_switcheroo_register_handler(struct vga_switcheroo_handler *handler)
{
	mutex_lock(&vgasr_mutex);
	if (vgasr_priv.handler) {
		mutex_unlock(&vgasr_mutex);
		return -EINVAL;
	}

	vgasr_priv.handler = handler;
	mutex_unlock(&vgasr_mutex);
	return 0;
}
EXPORT_SYMBOL(vga_switcheroo_register_handler);

void vga_switcheroo_unregister_handler(void)
{
	mutex_lock(&vgasr_mutex);
	vgasr_priv.handler = NULL;
	mutex_unlock(&vgasr_mutex);
}
EXPORT_SYMBOL(vga_switcheroo_unregister_handler);

static void vga_switcheroo_enable(void)
{
	int i;
	int ret;
	/* call the handler to init */
	vgasr_priv.handler->init();

	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
		ret = vgasr_priv.handler->get_client_id(vgasr_priv.clients[i].pdev);
		if (ret < 0)
			return;

		vgasr_priv.clients[i].id = ret;
	}
	vga_switcheroo_debugfs_init(&vgasr_priv);
	vgasr_priv.active = true;
}

int vga_switcheroo_register_client(struct pci_dev *pdev,
				   void (*set_gpu_state)(struct pci_dev *pdev, enum vga_switcheroo_state),
				   void (*reprobe)(struct pci_dev *pdev),
				   bool (*can_switch)(struct pci_dev *pdev))
{
	int index;

	mutex_lock(&vgasr_mutex);
	/* don't do IGD vs DIS here */
	if (vgasr_priv.registered_clients & 1)
		index = 1;
	else
		index = 0;

	vgasr_priv.clients[index].pwr_state = VGA_SWITCHEROO_ON;
	vgasr_priv.clients[index].pdev = pdev;
	vgasr_priv.clients[index].set_gpu_state = set_gpu_state;
	vgasr_priv.clients[index].reprobe = reprobe;
	vgasr_priv.clients[index].can_switch = can_switch;
	vgasr_priv.clients[index].id = -1;
	if (pdev == vga_default_device())
		vgasr_priv.clients[index].active = true;

	vgasr_priv.registered_clients |= (1 << index);

	/* if we get two clients + handler */
	if (vgasr_priv.registered_clients == 0x3 && vgasr_priv.handler) {
		printk(KERN_INFO "vga_switcheroo: enabled\n");
		vga_switcheroo_enable();
	}
	mutex_unlock(&vgasr_mutex);
	return 0;
}
EXPORT_SYMBOL(vga_switcheroo_register_client);

void vga_switcheroo_unregister_client(struct pci_dev *pdev)
{
	int i;

	mutex_lock(&vgasr_mutex);
	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
		if (vgasr_priv.clients[i].pdev == pdev) {
			vgasr_priv.registered_clients &= ~(1 << i);
			break;
		}
	}

	printk(KERN_INFO "vga_switcheroo: disabled\n");
	vga_switcheroo_debugfs_fini(&vgasr_priv);
	vgasr_priv.active = false;
	mutex_unlock(&vgasr_mutex);
}
EXPORT_SYMBOL(vga_switcheroo_unregister_client);

void vga_switcheroo_client_fb_set(struct pci_dev *pdev,
				 struct fb_info *info)
{
	int i;

	mutex_lock(&vgasr_mutex);
	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
		if (vgasr_priv.clients[i].pdev == pdev) {
			vgasr_priv.clients[i].fb_info = info;
			break;
		}
	}
	mutex_unlock(&vgasr_mutex);
}
EXPORT_SYMBOL(vga_switcheroo_client_fb_set);

static int vga_switcheroo_show(struct seq_file *m, void *v)
{
	int i;
	mutex_lock(&vgasr_mutex);
	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
		seq_printf(m, "%d:%s:%c:%s:%s\n", i,
			   vgasr_priv.clients[i].id == VGA_SWITCHEROO_DIS ? "DIS" : "IGD",
			   vgasr_priv.clients[i].active ? '+' : ' ',
			   vgasr_priv.clients[i].pwr_state ? "Pwr" : "Off",
			   pci_name(vgasr_priv.clients[i].pdev));
	}
	mutex_unlock(&vgasr_mutex);
	return 0;
}

static int vga_switcheroo_debugfs_open(struct inode *inode, struct file *file)
{
	return single_open(file, vga_switcheroo_show, NULL);
}

static int vga_switchon(struct vga_switcheroo_client *client)
{
	if (vgasr_priv.handler->power_state)
		vgasr_priv.handler->power_state(client->id, VGA_SWITCHEROO_ON);
	/* call the driver callback to turn on device */
	client->set_gpu_state(client->pdev, VGA_SWITCHEROO_ON);
	client->pwr_state = VGA_SWITCHEROO_ON;
	return 0;
}

static int vga_switchoff(struct vga_switcheroo_client *client)
{
	/* call the driver callback to turn off device */
	client->set_gpu_state(client->pdev, VGA_SWITCHEROO_OFF);
	if (vgasr_priv.handler->power_state)
		vgasr_priv.handler->power_state(client->id, VGA_SWITCHEROO_OFF);
	client->pwr_state = VGA_SWITCHEROO_OFF;
	return 0;
}

/* stage one happens before delay */
static int vga_switchto_stage1(struct vga_switcheroo_client *new_client)
{
	int i;
	struct vga_switcheroo_client *active = NULL;

	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
		if (vgasr_priv.clients[i].active == true) {
			active = &vgasr_priv.clients[i];
			break;
		}
	}
	if (!active)
		return 0;

	if (new_client->pwr_state == VGA_SWITCHEROO_OFF)
		vga_switchon(new_client);

	vga_set_default_device(new_client->pdev);

	return 0;
}

/* post delay */
static int vga_switchto_stage2(struct vga_switcheroo_client *new_client)
{
	int ret;
	int i;
	struct vga_switcheroo_client *active = NULL;

	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
		if (vgasr_priv.clients[i].active == true) {
			active = &vgasr_priv.clients[i];
			break;
		}
	}
	if (!active)
		return 0;

	active->active = false;

	if (new_client->fb_info) {
		struct fb_event event;
		event.info = new_client->fb_info;
		fb_notifier_call_chain(FB_EVENT_REMAP_ALL_CONSOLE, &event);
	}

	ret = vgasr_priv.handler->switchto(new_client->id);
	if (ret)
		return ret;

	if (new_client->reprobe)
		new_client->reprobe(new_client->pdev);

	if (active->pwr_state == VGA_SWITCHEROO_ON)
		vga_switchoff(active);

	new_client->active = true;
	return 0;
}

static ssize_t
vga_switcheroo_debugfs_write(struct file *filp, const char __user *ubuf,
			     size_t cnt, loff_t *ppos)
{
	char usercmd[64];
	const char *pdev_name;
	int i, ret;
	bool delay = false, can_switch;
	bool just_mux = false;
	int client_id = -1;
	struct vga_switcheroo_client *client = NULL;

	if (cnt > 63)
		cnt = 63;

	if (copy_from_user(usercmd, ubuf, cnt))
		return -EFAULT;

	mutex_lock(&vgasr_mutex);

	if (!vgasr_priv.active) {
		cnt = -EINVAL;
		goto out;
	}

	/* pwr off the device not in use */
	if (strncmp(usercmd, "OFF", 3) == 0) {
		for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
			if (vgasr_priv.clients[i].active)
				continue;
			if (vgasr_priv.clients[i].pwr_state == VGA_SWITCHEROO_ON)
				vga_switchoff(&vgasr_priv.clients[i]);
		}
		goto out;
	}
	/* pwr on the device not in use */
	if (strncmp(usercmd, "ON", 2) == 0) {
		for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
			if (vgasr_priv.clients[i].active)
				continue;
			if (vgasr_priv.clients[i].pwr_state == VGA_SWITCHEROO_OFF)
				vga_switchon(&vgasr_priv.clients[i]);
		}
		goto out;
	}

	/* request a delayed switch - test can we switch now */
	if (strncmp(usercmd, "DIGD", 4) == 0) {
		client_id = VGA_SWITCHEROO_IGD;
		delay = true;
	}

	if (strncmp(usercmd, "DDIS", 4) == 0) {
		client_id = VGA_SWITCHEROO_DIS;
		delay = true;
	}

	if (strncmp(usercmd, "IGD", 3) == 0)
		client_id = VGA_SWITCHEROO_IGD;

	if (strncmp(usercmd, "DIS", 3) == 0)
		client_id = VGA_SWITCHEROO_DIS;

	if (strncmp(usercmd, "MIGD", 4) == 0) {
		just_mux = true;
		client_id = VGA_SWITCHEROO_IGD;
	}
	if (strncmp(usercmd, "MDIS", 4) == 0) {
		just_mux = true;
		client_id = VGA_SWITCHEROO_DIS;
	}

	if (client_id == -1)
		goto out;

	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
		if (vgasr_priv.clients[i].id == client_id) {
			client = &vgasr_priv.clients[i];
			break;
		}
	}

	vgasr_priv.delayed_switch_active = false;

	if (just_mux) {
		ret = vgasr_priv.handler->switchto(client_id);
		goto out;
	}

	if (client->active == true)
		goto out;

	/* okay we want a switch - test if devices are willing to switch */
	can_switch = true;
	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
		can_switch = vgasr_priv.clients[i].can_switch(vgasr_priv.clients[i].pdev);
		if (can_switch == false) {
			printk(KERN_ERR "vga_switcheroo: client %d refused switch\n", i);
			break;
		}
	}

	if (can_switch == false && delay == false)
		goto out;

	if (can_switch == true) {
		pdev_name = pci_name(client->pdev);
		ret = vga_switchto_stage1(client);
		if (ret)
			printk(KERN_ERR "vga_switcheroo: switching failed stage 1 %d\n", ret);

		ret = vga_switchto_stage2(client);
		if (ret)
			printk(KERN_ERR "vga_switcheroo: switching failed stage 2 %d\n", ret);

	} else {
		printk(KERN_INFO "vga_switcheroo: setting delayed switch to client %d\n", client->id);
		vgasr_priv.delayed_switch_active = true;
		vgasr_priv.delayed_client_id = client_id;

		ret = vga_switchto_stage1(client);
		if (ret)
			printk(KERN_ERR "vga_switcheroo: delayed switching stage 1 failed %d\n", ret);
	}

out:
	mutex_unlock(&vgasr_mutex);
	return cnt;
}

static const struct file_operations vga_switcheroo_debugfs_fops = {
	.owner = THIS_MODULE,
	.open = vga_switcheroo_debugfs_open,
	.write = vga_switcheroo_debugfs_write,
	.read = seq_read,
	.llseek = seq_lseek,
	.release = single_release,
};

static void vga_switcheroo_debugfs_fini(struct vgasr_priv *priv)
{
	if (priv->switch_file) {
		debugfs_remove(priv->switch_file);
		priv->switch_file = NULL;
	}
	if (priv->debugfs_root) {
		debugfs_remove(priv->debugfs_root);
		priv->debugfs_root = NULL;
	}
}

static int vga_switcheroo_debugfs_init(struct vgasr_priv *priv)
{
	/* already initialised */
	if (priv->debugfs_root)
		return 0;
	priv->debugfs_root = debugfs_create_dir("vgaswitcheroo", NULL);

	if (!priv->debugfs_root) {
		printk(KERN_ERR "vga_switcheroo: Cannot create /sys/kernel/debug/vgaswitcheroo\n");
		goto fail;
	}

	priv->switch_file = debugfs_create_file("switch", 0644,
						priv->debugfs_root, NULL, &vga_switcheroo_debugfs_fops);
	if (!priv->switch_file) {
		printk(KERN_ERR "vga_switcheroo: cannot create /sys/kernel/debug/vgaswitcheroo/switch\n");
		goto fail;
	}
	return 0;
fail:
	vga_switcheroo_debugfs_fini(priv);
	return -1;
}

int vga_switcheroo_process_delayed_switch(void)
{
	struct vga_switcheroo_client *client = NULL;
	const char *pdev_name;
	bool can_switch = true;
	int i;
	int ret;
	int err = -EINVAL;

	mutex_lock(&vgasr_mutex);
	if (!vgasr_priv.delayed_switch_active)
		goto err;

	printk(KERN_INFO "vga_switcheroo: processing delayed switch to %d\n", vgasr_priv.delayed_client_id);

	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
		if (vgasr_priv.clients[i].id == vgasr_priv.delayed_client_id)
			client = &vgasr_priv.clients[i];
		can_switch = vgasr_priv.clients[i].can_switch(vgasr_priv.clients[i].pdev);
		if (can_switch == false) {
			printk(KERN_ERR "vga_switcheroo: client %d refused switch\n", i);
			break;
		}
	}

	if (can_switch == false || client == NULL)
		goto err;

	pdev_name = pci_name(client->pdev);
	ret = vga_switchto_stage2(client);
	if (ret)
		printk(KERN_ERR "vga_switcheroo: delayed switching failed stage 2 %d\n", ret);

	vgasr_priv.delayed_switch_active = false;
	err = 0;
err:
	mutex_unlock(&vgasr_mutex);
	return err;
}
EXPORT_SYMBOL(vga_switcheroo_process_delayed_switch);
Commit	Line	Data
6a9ee8af DA	1	/*
	2	* Copyright (c) 2010 Red Hat Inc.
	3	* Author : Dave Airlie <airlied@redhat.com>
	4	*
	5	*
	6	* Licensed under GPLv2
	7	*
	8	* vga_switcheroo.c - Support for laptop with dual GPU using one set of outputs
	9
	10	Switcher interface - methods require for ATPX and DCM
	11	- switchto - this throws the output MUX switch
	12	- discrete_set_power - sets the power state for the discrete card
	13
	14	GPU driver interface
	15	- set_gpu_state - this should do the equiv of s/r for the card
	16	- this should not set the discrete power state
	17	- switch_check - check if the device is in a position to switch now
	18	*/
	19
	20	#include <linux/module.h>
	21	#include <linux/dmi.h>
	22	#include <linux/seq_file.h>
	23	#include <linux/uaccess.h>
	24	#include <linux/fs.h>
	25	#include <linux/debugfs.h>
	26	#include <linux/fb.h>
	27
6a9ee8af DA	28	#include <linux/pci.h>
	29	#include <linux/vga_switcheroo.h>
	30
2fbe8c7c MG	31	#include <linux/vgaarb.h>
2fbe8c7c MG	32
6a9ee8af DA	33	struct vga_switcheroo_client {
	34	struct pci_dev *pdev;
	35	struct fb_info *fb_info;
	36	int pwr_state;
	37	void (set_gpu_state)(struct pci_dev pdev, enum vga_switcheroo_state);
8d608aa6	38	void (reprobe)(struct pci_dev pdev);
6a9ee8af DA	39	bool (can_switch)(struct pci_dev pdev);
	40	int id;
	41	bool active;
	42	};
	43
	44	static DEFINE_MUTEX(vgasr_mutex);
	45
	46	struct vgasr_priv {
	47
	48	bool active;
	49	bool delayed_switch_active;
	50	enum vga_switcheroo_client_id delayed_client_id;
	51
	52	struct dentry *debugfs_root;
	53	struct dentry *switch_file;
	54
	55	int registered_clients;
	56	struct vga_switcheroo_client clients[VGA_SWITCHEROO_MAX_CLIENTS];
	57
	58	struct vga_switcheroo_handler *handler;
	59	};
	60
	61	static int vga_switcheroo_debugfs_init(struct vgasr_priv *priv);
	62	static void vga_switcheroo_debugfs_fini(struct vgasr_priv *priv);
	63
	64	/* only one switcheroo per system */
	65	static struct vgasr_priv vgasr_priv;
	66
	67	int vga_switcheroo_register_handler(struct vga_switcheroo_handler *handler)
	68	{
	69	mutex_lock(&vgasr_mutex);
	70	if (vgasr_priv.handler) {
	71	mutex_unlock(&vgasr_mutex);
	72	return -EINVAL;
	73	}
	74
	75	vgasr_priv.handler = handler;
	76	mutex_unlock(&vgasr_mutex);
	77	return 0;
	78	}
	79	EXPORT_SYMBOL(vga_switcheroo_register_handler);
	80
	81	void vga_switcheroo_unregister_handler(void)
	82	{
	83	mutex_lock(&vgasr_mutex);
	84	vgasr_priv.handler = NULL;
	85	mutex_unlock(&vgasr_mutex);
	86	}
	87	EXPORT_SYMBOL(vga_switcheroo_unregister_handler);
	88
	89	static void vga_switcheroo_enable(void)
	90	{
	91	int i;
	92	int ret;
	93	/* call the handler to init */
	94	vgasr_priv.handler->init();
	95
	96	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
	97	ret = vgasr_priv.handler->get_client_id(vgasr_priv.clients[i].pdev);
	98	if (ret < 0)
	99	return;
	100
	101	vgasr_priv.clients[i].id = ret;
	102	}
103	vga_switcheroo_debugfs_init(&vgasr_priv);
104	vgasr_priv.active = true;
105	}
106
107	int vga_switcheroo_register_client(struct pci_dev *pdev,
108	void (set_gpu_state)(struct pci_dev pdev, enum vga_switcheroo_state),
8d608aa6	109	void (reprobe)(struct pci_dev pdev),
6a9ee8af DA	110	bool (can_switch)(struct pci_dev pdev))
	111	{
	112	int index;
	113
	114	mutex_lock(&vgasr_mutex);
	115	/* don't do IGD vs DIS here */
	116	if (vgasr_priv.registered_clients & 1)
	117	index = 1;
	118	else
	119	index = 0;
	120
	121	vgasr_priv.clients[index].pwr_state = VGA_SWITCHEROO_ON;
	122	vgasr_priv.clients[index].pdev = pdev;
	123	vgasr_priv.clients[index].set_gpu_state = set_gpu_state;
8d608aa6	124	vgasr_priv.clients[index].reprobe = reprobe;
6a9ee8af DA	125	vgasr_priv.clients[index].can_switch = can_switch;
6a9ee8af DA	126	vgasr_priv.clients[index].id = -1;
2fbe8c7c	127	if (pdev == vga_default_device())
6a9ee8af DA	128	vgasr_priv.clients[index].active = true;
	129
	130	vgasr_priv.registered_clients \|= (1 << index);
	131
	132	/* if we get two clients + handler */
	133	if (vgasr_priv.registered_clients == 0x3 && vgasr_priv.handler) {
	134	printk(KERN_INFO "vga_switcheroo: enabled\n");
	135	vga_switcheroo_enable();
	136	}
	137	mutex_unlock(&vgasr_mutex);
	138	return 0;
	139	}
	140	EXPORT_SYMBOL(vga_switcheroo_register_client);
	141
	142	void vga_switcheroo_unregister_client(struct pci_dev *pdev)
	143	{
	144	int i;
	145
	146	mutex_lock(&vgasr_mutex);
	147	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
	148	if (vgasr_priv.clients[i].pdev == pdev) {
	149	vgasr_priv.registered_clients &= ~(1 << i);
	150	break;
	151	}
	152	}
	153
	154	printk(KERN_INFO "vga_switcheroo: disabled\n");
	155	vga_switcheroo_debugfs_fini(&vgasr_priv);
	156	vgasr_priv.active = false;
	157	mutex_unlock(&vgasr_mutex);
	158	}
	159	EXPORT_SYMBOL(vga_switcheroo_unregister_client);
	160
	161	void vga_switcheroo_client_fb_set(struct pci_dev *pdev,
	162	struct fb_info *info)
	163	{
	164	int i;
	165
	166	mutex_lock(&vgasr_mutex);
	167	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
	168	if (vgasr_priv.clients[i].pdev == pdev) {
	169	vgasr_priv.clients[i].fb_info = info;
	170	break;
	171	}
	172	}
	173	mutex_unlock(&vgasr_mutex);
	174	}
	175	EXPORT_SYMBOL(vga_switcheroo_client_fb_set);
	176
	177	static int vga_switcheroo_show(struct seq_file m, void v)
	178	{
	179	int i;
	180	mutex_lock(&vgasr_mutex);
	181	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
6c2df40e DA	182	seq_printf(m, "%d:%s:%c:%s:%s\n", i,
6c2df40e DA	183	vgasr_priv.clients[i].id == VGA_SWITCHEROO_DIS ? "DIS" : "IGD",
6a9ee8af DA	184	vgasr_priv.clients[i].active ? '+' : ' ',
	185	vgasr_priv.clients[i].pwr_state ? "Pwr" : "Off",
	186	pci_name(vgasr_priv.clients[i].pdev));
	187	}
	188	mutex_unlock(&vgasr_mutex);
	189	return 0;
	190	}
	191
	192	static int vga_switcheroo_debugfs_open(struct inode inode, struct file file)
	193	{
	194	return single_open(file, vga_switcheroo_show, NULL);
	195	}
	196
	197	static int vga_switchon(struct vga_switcheroo_client *client)
	198	{
5cfb3c3a DA	199	if (vgasr_priv.handler->power_state)
5cfb3c3a DA	200	vgasr_priv.handler->power_state(client->id, VGA_SWITCHEROO_ON);
6a9ee8af DA	201	/* call the driver callback to turn on device */
	202	client->set_gpu_state(client->pdev, VGA_SWITCHEROO_ON);
	203	client->pwr_state = VGA_SWITCHEROO_ON;
	204	return 0;
	205	}
	206
	207	static int vga_switchoff(struct vga_switcheroo_client *client)
	208	{
	209	/* call the driver callback to turn off device */
	210	client->set_gpu_state(client->pdev, VGA_SWITCHEROO_OFF);
5cfb3c3a DA	211	if (vgasr_priv.handler->power_state)
5cfb3c3a DA	212	vgasr_priv.handler->power_state(client->id, VGA_SWITCHEROO_OFF);
6a9ee8af DA	213	client->pwr_state = VGA_SWITCHEROO_OFF;
	214	return 0;
	215	}
	216
66b37c67 DA	217	/* stage one happens before delay */
66b37c67 DA	218	static int vga_switchto_stage1(struct vga_switcheroo_client *new_client)
6a9ee8af	219	{
6a9ee8af DA	220	int i;
	221	struct vga_switcheroo_client *active = NULL;
	222
6a9ee8af DA	223	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
	224	if (vgasr_priv.clients[i].active == true) {
	225	active = &vgasr_priv.clients[i];
	226	break;
	227	}
	228	}
	229	if (!active)
	230	return 0;
	231
6a9ee8af DA	232	if (new_client->pwr_state == VGA_SWITCHEROO_OFF)
	233	vga_switchon(new_client);
	234
2fbe8c7c MG	235	vga_set_default_device(new_client->pdev);
2fbe8c7c MG	236
66b37c67 DA	237	return 0;
	238	}
	239
	240	/* post delay */
	241	static int vga_switchto_stage2(struct vga_switcheroo_client *new_client)
	242	{
	243	int ret;
	244	int i;
	245	struct vga_switcheroo_client *active = NULL;
	246
	247	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
	248	if (vgasr_priv.clients[i].active == true) {
	249	active = &vgasr_priv.clients[i];
	250	break;
	251	}
	252	}
	253	if (!active)
	254	return 0;
	255
	256	active->active = false;
6a9ee8af DA	257
	258	if (new_client->fb_info) {
	259	struct fb_event event;
	260	event.info = new_client->fb_info;
	261	fb_notifier_call_chain(FB_EVENT_REMAP_ALL_CONSOLE, &event);
	262	}
	263
	264	ret = vgasr_priv.handler->switchto(new_client->id);
	265	if (ret)
	266	return ret;
	267
8d608aa6 DA	268	if (new_client->reprobe)
	269	new_client->reprobe(new_client->pdev);
	270
6a9ee8af DA	271	if (active->pwr_state == VGA_SWITCHEROO_ON)
	272	vga_switchoff(active);
	273
	274	new_client->active = true;
	275	return 0;
	276	}
	277
	278	static ssize_t
	279	vga_switcheroo_debugfs_write(struct file filp, const char __user ubuf,
	280	size_t cnt, loff_t *ppos)
	281	{
	282	char usercmd[64];
	283	const char *pdev_name;
	284	int i, ret;
	285	bool delay = false, can_switch;
851ab954	286	bool just_mux = false;
6a9ee8af DA	287	int client_id = -1;
	288	struct vga_switcheroo_client *client = NULL;
	289
	290	if (cnt > 63)
	291	cnt = 63;
	292
	293	if (copy_from_user(usercmd, ubuf, cnt))
	294	return -EFAULT;
	295
	296	mutex_lock(&vgasr_mutex);
	297
8c88e50b JS	298	if (!vgasr_priv.active) {
	299	cnt = -EINVAL;
	300	goto out;
	301	}
6a9ee8af DA	302
	303	/* pwr off the device not in use */
	304	if (strncmp(usercmd, "OFF", 3) == 0) {
	305	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
	306	if (vgasr_priv.clients[i].active)
	307	continue;
	308	if (vgasr_priv.clients[i].pwr_state == VGA_SWITCHEROO_ON)
	309	vga_switchoff(&vgasr_priv.clients[i]);
	310	}
	311	goto out;
	312	}
	313	/* pwr on the device not in use */
	314	if (strncmp(usercmd, "ON", 2) == 0) {
	315	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
	316	if (vgasr_priv.clients[i].active)
	317	continue;
	318	if (vgasr_priv.clients[i].pwr_state == VGA_SWITCHEROO_OFF)
	319	vga_switchon(&vgasr_priv.clients[i]);
	320	}
	321	goto out;
	322	}
	323
	324	/* request a delayed switch - test can we switch now */
	325	if (strncmp(usercmd, "DIGD", 4) == 0) {
	326	client_id = VGA_SWITCHEROO_IGD;
	327	delay = true;
	328	}
	329
	330	if (strncmp(usercmd, "DDIS", 4) == 0) {
	331	client_id = VGA_SWITCHEROO_DIS;
	332	delay = true;
	333	}
	334
	335	if (strncmp(usercmd, "IGD", 3) == 0)
	336	client_id = VGA_SWITCHEROO_IGD;
	337
	338	if (strncmp(usercmd, "DIS", 3) == 0)
	339	client_id = VGA_SWITCHEROO_DIS;
	340
fea6f330	341	if (strncmp(usercmd, "MIGD", 4) == 0) {
851ab954 DA	342	just_mux = true;
	343	client_id = VGA_SWITCHEROO_IGD;
	344	}
fea6f330	345	if (strncmp(usercmd, "MDIS", 4) == 0) {
851ab954 DA	346	just_mux = true;
	347	client_id = VGA_SWITCHEROO_DIS;
	348	}
	349
6a9ee8af DA	350	if (client_id == -1)
	351	goto out;
	352
	353	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
	354	if (vgasr_priv.clients[i].id == client_id) {
	355	client = &vgasr_priv.clients[i];
	356	break;
	357	}
	358	}
	359
	360	vgasr_priv.delayed_switch_active = false;
851ab954 DA	361
	362	if (just_mux) {
	363	ret = vgasr_priv.handler->switchto(client_id);
	364	goto out;
	365	}
	366
a67b8887 FM	367	if (client->active == true)
	368	goto out;
	369
6a9ee8af DA	370	/* okay we want a switch - test if devices are willing to switch */
	371	can_switch = true;
	372	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
	373	can_switch = vgasr_priv.clients[i].can_switch(vgasr_priv.clients[i].pdev);
	374	if (can_switch == false) {
	375	printk(KERN_ERR "vga_switcheroo: client %d refused switch\n", i);
	376	break;
	377	}
	378	}
	379
	380	if (can_switch == false && delay == false)
	381	goto out;
	382
	383	if (can_switch == true) {
	384	pdev_name = pci_name(client->pdev);
66b37c67 DA	385	ret = vga_switchto_stage1(client);
	386	if (ret)
	387	printk(KERN_ERR "vga_switcheroo: switching failed stage 1 %d\n", ret);
	388
	389	ret = vga_switchto_stage2(client);
6a9ee8af	390	if (ret)
66b37c67 DA	391	printk(KERN_ERR "vga_switcheroo: switching failed stage 2 %d\n", ret);
66b37c67 DA	392
6a9ee8af DA	393	} else {
	394	printk(KERN_INFO "vga_switcheroo: setting delayed switch to client %d\n", client->id);
	395	vgasr_priv.delayed_switch_active = true;
	396	vgasr_priv.delayed_client_id = client_id;
	397
66b37c67 DA	398	ret = vga_switchto_stage1(client);
	399	if (ret)
	400	printk(KERN_ERR "vga_switcheroo: delayed switching stage 1 failed %d\n", ret);
6a9ee8af DA	401	}
	402
	403	out:
	404	mutex_unlock(&vgasr_mutex);
	405	return cnt;
	406	}
	407
	408	static const struct file_operations vga_switcheroo_debugfs_fops = {
	409	.owner = THIS_MODULE,
	410	.open = vga_switcheroo_debugfs_open,
	411	.write = vga_switcheroo_debugfs_write,
	412	.read = seq_read,
	413	.llseek = seq_lseek,
	414	.release = single_release,
	415	};
	416
	417	static void vga_switcheroo_debugfs_fini(struct vgasr_priv *priv)
	418	{
	419	if (priv->switch_file) {
	420	debugfs_remove(priv->switch_file);
	421	priv->switch_file = NULL;
	422	}
	423	if (priv->debugfs_root) {
	424	debugfs_remove(priv->debugfs_root);
	425	priv->debugfs_root = NULL;
	426	}
	427	}
	428
	429	static int vga_switcheroo_debugfs_init(struct vgasr_priv *priv)
	430	{
	431	/* already initialised */
	432	if (priv->debugfs_root)
	433	return 0;
	434	priv->debugfs_root = debugfs_create_dir("vgaswitcheroo", NULL);
	435
	436	if (!priv->debugfs_root) {
	437	printk(KERN_ERR "vga_switcheroo: Cannot create /sys/kernel/debug/vgaswitcheroo\n");
	438	goto fail;
	439	}
	440
	441	priv->switch_file = debugfs_create_file("switch", 0644,
	442	priv->debugfs_root, NULL, &vga_switcheroo_debugfs_fops);
	443	if (!priv->switch_file) {
	444	printk(KERN_ERR "vga_switcheroo: cannot create /sys/kernel/debug/vgaswitcheroo/switch\n");
	445	goto fail;
	446	}
	447	return 0;
	448	fail:
	449	vga_switcheroo_debugfs_fini(priv);
	450	return -1;
	451	}
	452
	453	int vga_switcheroo_process_delayed_switch(void)
	454	{
	455	struct vga_switcheroo_client *client = NULL;
	456	const char *pdev_name;
	457	bool can_switch = true;
	458	int i;
	459	int ret;
	460	int err = -EINVAL;
	461
	462	mutex_lock(&vgasr_mutex);
	463	if (!vgasr_priv.delayed_switch_active)
	464	goto err;
465
466	printk(KERN_INFO "vga_switcheroo: processing delayed switch to %d\n", vgasr_priv.delayed_client_id);
467
468	for (i = 0; i < VGA_SWITCHEROO_MAX_CLIENTS; i++) {
469	if (vgasr_priv.clients[i].id == vgasr_priv.delayed_client_id)
470	client = &vgasr_priv.clients[i];
471	can_switch = vgasr_priv.clients[i].can_switch(vgasr_priv.clients[i].pdev);
472	if (can_switch == false) {
473	printk(KERN_ERR "vga_switcheroo: client %d refused switch\n", i);
474	break;
475	}
476	}
477
478	if (can_switch == false \|\| client == NULL)
479	goto err;
480
481	pdev_name = pci_name(client->pdev);
66b37c67	482	ret = vga_switchto_stage2(client);
6a9ee8af	483	if (ret)
66b37c67	484	printk(KERN_ERR "vga_switcheroo: delayed switching failed stage 2 %d\n", ret);
6a9ee8af DA	485
	486	vgasr_priv.delayed_switch_active = false;
	487	err = 0;
	488	err:
	489	mutex_unlock(&vgasr_mutex);
	490	return err;
	491	}
	492	EXPORT_SYMBOL(vga_switcheroo_process_delayed_switch);
	493