#include <linux/timer.h>
 #include <linux/types.h>
 #include <linux/platform_device.h>
+#include <linux/leds.h>
 
 /*
  * Number of attempts to read data from queue per poll;
 /* BIOS subsystem IDs */
 #define WIFI           0x35
 #define BLUETOOTH      0x34
+#define MAIL_LED       0x31
 
 MODULE_AUTHOR("Miloslav Trmac <mitr@volny.cz>");
 MODULE_DESCRIPTION("Wistron laptop button driver");
 MODULE_LICENSE("GPL v2");
-MODULE_VERSION("0.2");
+MODULE_VERSION("0.3");
 
 static int force; /* = 0; */
 module_param(force, bool, 0);
 static const struct key_entry *keymap; /* = NULL; Current key map */
 static int have_wifi;
 static int have_bluetooth;
+static int have_leds;
 
 static int __init dmi_matched(struct dmi_system_id *dmi)
 {
                else if (key->type == KE_BLUETOOTH)
                        have_bluetooth = 1;
        }
+       have_leds = key->code & (FE_MAIL_LED | FE_WIFI_LED);
+
        return 1;
 }
 
        input_sync(input_dev);
 }
 
+
+ /* led management */
+static void wistron_mail_led_set(struct led_classdev *led_cdev,
+                               enum led_brightness value)
+{
+       bios_set_state(MAIL_LED, (value != LED_OFF) ? 1 : 0);
+}
+
+/* same as setting up wifi card, but for laptops on which the led is managed */
+static void wistron_wifi_led_set(struct led_classdev *led_cdev,
+                               enum led_brightness value)
+{
+       bios_set_state(WIFI, (value != LED_OFF) ? 1 : 0);
+}
+
+static struct led_classdev wistron_mail_led = {
+       .name                   = "mail:green",
+       .brightness_set         = wistron_mail_led_set,
+};
+
+static struct led_classdev wistron_wifi_led = {
+       .name                   = "wifi:red",
+       .brightness_set         = wistron_wifi_led_set,
+};
+
+static void __devinit wistron_led_init(struct device *parent)
+{
+       if (have_leds & FE_WIFI_LED) {
+               u16 wifi = bios_get_default_setting(WIFI);
+               if (wifi & 1) {
+                       wistron_wifi_led.brightness = (wifi & 2) ? LED_FULL : LED_OFF;
+                       if (led_classdev_register(parent, &wistron_wifi_led))
+                               have_leds &= ~FE_WIFI_LED;
+                       else
+                               bios_set_state(WIFI, wistron_wifi_led.brightness);
+
+               } else
+                       have_leds &= ~FE_WIFI_LED;
+       }
+
+       if (have_leds & FE_MAIL_LED) {
+               /* bios_get_default_setting(MAIL) always retuns 0, so just turn the led off */
+               wistron_mail_led.brightness = LED_OFF;
+               if (led_classdev_register(parent, &wistron_mail_led))
+                       have_leds &= ~FE_MAIL_LED;
+               else
+                       bios_set_state(MAIL_LED, wistron_mail_led.brightness);
+       }
+}
+
+static void __devexit wistron_led_remove(void)
+{
+       if (have_leds & FE_MAIL_LED)
+               led_classdev_unregister(&wistron_mail_led);
+
+       if (have_leds & FE_WIFI_LED)
+               led_classdev_unregister(&wistron_wifi_led);
+}
+
+static inline void wistron_led_suspend(void)
+{
+       if (have_leds & FE_MAIL_LED)
+               led_classdev_suspend(&wistron_mail_led);
+
+       if (have_leds & FE_WIFI_LED)
+               led_classdev_suspend(&wistron_wifi_led);
+}
+
+static inline void wistron_led_resume(void)
+{
+       if (have_leds & FE_MAIL_LED)
+               led_classdev_resume(&wistron_mail_led);
+
+       if (have_leds & FE_WIFI_LED)
+               led_classdev_resume(&wistron_wifi_led);
+}
+
  /* Driver core */
 
 static int wifi_enabled;
                        bios_set_state(BLUETOOTH, bluetooth_enabled);
        }
 
+       wistron_led_init(&dev->dev);
        poll_bios(1); /* Flush stale event queue and arm timer */
 
        return 0;
 static int __devexit wistron_remove(struct platform_device *dev)
 {
        del_timer_sync(&poll_timer);
+       wistron_led_remove();
        input_unregister_device(input_dev);
        bios_detach();
 
        if (have_bluetooth)
                bios_set_state(BLUETOOTH, 0);
 
+       wistron_led_suspend();
        return 0;
 }
 
        if (have_bluetooth)
                bios_set_state(BLUETOOTH, bluetooth_enabled);
 
+       wistron_led_resume();
        poll_bios(1);
 
        return 0;