首页
社区
课程
招聘
[原创]一起来写控制普通键盘上的3盏LED的驱动
发表于: 2011-2-12 19:57 9936

[原创]一起来写控制普通键盘上的3盏LED的驱动

2011-2-12 19:57
9936

最近在看ROOTKITS方面的书,有一章直接操纵硬件,是关于键盘上的LED指示器的,感觉对我这样的菜鸟很有帮助,就按照着文章的例程,写下了这篇文章。由于我在ROOTKITS这方面还只能算个菜鸟中的菜鸟,所以文章中的所谓还希望各位看官们,不吝赐教嗷。
说实话不知道应该从哪里讲起,我这方面的知识很零碎……硬着头皮上吧,一边写一边百度,额。
其实光明正大的控制一个键盘很简单。一般的键盘控制器在0X60和0X64上是可寻址的,也就是说这两个端口(PORT)是进入键盘芯片的端口。如果像我一样使用WDK,则可以使用:
READ_PORT_UCHAR(PORT);
WRITE_PORT_UCHAR(PORT,COMMAND/DATA);
这两个函数对这两个端口进行读写操作。
通过这两个端口我们可以做很多事情,比如键盘窃听什么的^^当然还有我们今天的主角,控制键盘上LED指示器。
键盘和鼠标是共享0X60端口的,所以在使用之前,系统要先读取0X64端口,判断是鼠标还是键盘。话说回来。笔记本这方面也应该一样的吧。

设置LED的命令是 0XED。而 OXED 使用的数据格式是一个八位的数据是:
  ■■■■■■■■        (话说一个■表示一位,即0或者1,最后三位控制键盘上 SCROLL LOCK ,NUM LOCK,CAPS LOCK3个LED灯的亮暗)
            ↑↑↑
            ↑↑CAPS LOCK
            ↑↑
            ↑NUM LOCK
            ↑
            ↑
            SCROLL LOCK
而我们所要做的就是使用上面提到的2个函数:
WRITE_PORT_UCHAR(0X60,0XED);
WRITE_PORT_UCHAR(0X60,00000111b); 比如这样表示3个灯全亮。
上面两个就是传说中的核心代码……要是文章能这样结束那该多好呀~~~不过硬件可不是那么好对付的。
如果键盘忙于处理击键动作,那么这种方式很可能无法成功。对话硬件一个很重要的地方就是要等到硬件芯片就绪,如果芯片未就绪就试图向它发送命令、数据,比较理想的情况是什么都不会发生。偶尔么就崩溃了。
程序中出现了一个叫做Dpc的东西,看不懂可以先放放,而且我也不怎么懂,等下我们一起学吧。
前置知识介绍完了,开始动手吧。 对了 这个针对8042键盘,也就是一般是普通台式的键盘,话说笔记本上的3个LED灯就不一样了,比如我这台上面最左边的是表示硬盘读写状态的LED。
写完用WDK编译一下就可以用 InstDrv 加载测试了,呵呵,来一次与硬件的对话吧。
===========================================
#include "ntddk.h"  //不解释
#include <stdio.h>  //同上

PUCHAR KEYBOARD_PORT_60 = (PUCHAR)0x60;
PUCHAR KEYBOARD_PORT_64 = (PUCHAR)0x64; //定义两个端口

#define IBUFFER_FULL 0X02  //寄存器标志位,等下会与READ_PORT_UCHAR()的返回值进行&运算,来判断是否没有输入,至于为什么是00000010b,我不知道
#define OBUFFER_FULL 0X01  //估计是与键盘输的数据有关吧,因为WDK文档上讲READ_PORT_UCHAR()的返回值是从指定端口的读取的字节。
                           //键盘输出的字符毕竟有限。
PKTIMER gTimer;  //计时器
PKDPC gDPCP;     //Dpc,后面细说
UCHAR g_key_bits=0;    //这个是控制LED灯循环的

BOOLEAN WaitForKeyboard()
{
   INT i=100;     //循环的次数
   UCHAR mychar;   
   do
     {
       mychar=READ_PORT_UCHAR(KEYBOARD_PORT_64);      
       KeStallExecutionProcessor(666);             //这个函数据说能使CPU暂停特定微秒数。于是键盘可以完成它先前的工作。至于为什么
                                                   //是666毫秒。。因为书上这么写的。
       if(!(mychar&IBUFFER_FULL)) break;
     }
   while(i--);
   if(i) return TRUE;
   return FALSE;
}
VOID DrainOutputBuffer()                                 //这个函数的用途是如果键盘缓冲区含有击键动作,它会将缓冲区排干,防止干                                                          // 扰LED灯
{                                                        
  int i=100;      //循环次数
  UCHAR c;
   do
   {
       c=READ_PORT_UCHAR(KEYBOARD_PORT_64);              
       KeStallExecutionProcessor(666);                   //依然666。
       if(!(c&OBUFFER_FULL)) break;   
   }
   while(i--);
}
BOOLEAN SendKeyboardCommand(                              //等下会被下一个函数SetLEDS调用,开始传给0X60端口命令和数据了 ,文章开
                             IN UCHAR theCommand          //始有说过怎么传的 还记得吧。
                           )
{
   if(TRUE == WaitForKeyboard())                          //如果键盘就绪 就开始传递。
    {
      WRITE_PORT_UCHAR(KEYBOARD_PORT_60,theCommand);
      return TRUE;
    }
   return FALSE;
   
}
VOID SetLEDS(                                   //这个函数调用上面一个函数。
              UCHAR theLEDS
            )
{
  SendKeyboardCommand(0XED);
  SendKeyboardCommand(theLEDS);
}

VOID OnUnload(                                      //卸载驱动的函数。取消该取消的,释放该释放的。
               IN PDRIVER_OBJECT DriverObject
             )
{
      KeCancelTimer(gTimer); //取消计时器
      ExFreePool(gTimer);    //释放内存池
      ExFreePool(gDPCP);     //同上
}

VOID timerDPC(                                           //这个函数是KeInitializeDpc的一个参数,等下看KeInitializeDpc的时候一起                                                                //讲吧。
               IN PKDPC Dpc,
               IN PVOID DeferredContext,
               IN PVOID sys1,
               IN PVOID sys2
             )
{
  SetLEDS(g_key_bits++);
  if(g_key_bits>0X07) g_key_bits=0;
}                                                      

NTSTATUS DriverEntry(
                     IN PDRIVER_OBJECT theDriverObject,
                     IN PUNICODE_STRING theRegistryPath
                    )
{   
    LARGE_INTEGER timeout;                          
    theDriverObject->DriverUnload=OnUnload;              //申明驱动卸载例程
    gTimer=ExAllocatePool(NonPagedPool,sizeof(KTIMER));  //分配内存,由于只是个测试,就用NonPagedPool参数了。
    gDPCP=ExAllocatePool(NonPagedPool,sizeof(KDPC));    //一样的。
    timeout.QuadPart=-10;                           //为什么是负数,后面讲。
    KeInitializeTimer(gTimer);                      //安装计时器。
    KeInitializeDpc(gDPCP,timerDPC,NULL);           //安装Dpc。Dpc这方面我也不懂。等下在后面我们一起学吧。
    KeSetTimerEx(gTimer,timeout,300,gDPCP);         //我的理解是配置计时器。
   
      return STATUS_SUCCESS;
   
   
}

==============================================================================
程序就是上面这样。
我们倒着看吧。先看看KeSetTimerEx(gTimer,timeout,300,gDPCP)这个函数。
WDK上的帮助文件如是说:
The KeSetTimerEx routine sets the absolute or relative interval at which a timer object is to be set to a signaled state, optionally supplies a CustomTimerDpc routine to be executed when that interval expires, and optionally supplies a recurring interval for the timer.

BOOLEAN
  KeSetTimerEx(
    IN PKTIMER  Timer,
    IN LARGE_INTEGER  DueTime,
    IN LONG  Period OPTIONAL,
    IN PKDPC  Dpc OPTIONAL
    );

我的理解是对计时器的配置。 第一个参数就我们之前用KeInitializeTimer安装的计时器。第二个参数是之前提到的类型是LARGE_INTEGER的timeout,表示计时器的循环响应(“响应”这个词我想了好久,额)时间,如果timeout.QuadPart是正数,就表示绝对响应时间,这里不说开去。这里我们用了timeout.QuadPart=-10,表示相对于系统当前时间的响应时间,也就是10微秒后响应。后面两个参数都是可选的。Period申明一个周期多少微秒,第四个Dpc么依然放到后面讲吧。

再上面一个函数就是安装我提到好几次的Dpc了------KeInitializeDpc(gDPCP,timerDPC,NULL) WDK文档里面说了等于没说,额。

首先谈谈DPC(Deffered Procedure Call)延迟过程调用。
当一段内核代码运行于一个被提升的IRQL上时,其他任何拥有相同或较低IRQL值的中断都不能执行的(在相同CUP上)。然而,如果在很高的IRQL上执行的代码太多,系统总体性能就会下降。时间关键的时间处理会被延迟、并导致更加灾难性的结果。
为了避免这些问题发生,就需要对内核模式代码进行设计,使其尽可能在最低级的IRQL上执行尽量多的代码。延迟过程调用就是这样一个重要策略。
DPC结构允许一个任务从一个高级的IRQL上被触发,但是不被执行。当服务硬件在一个驱动程序上中断时,这种延迟执行是必要的,因为在一个给定任务能被延迟的情况下,没有必要中断低级IRQL代码的执行。

然后是KeInitializeDpc(gDPCP,timerDPC,NULL)这边有3个参数,第一个就是个DPC,第二个是一个PKDEFERRED_ROUTINE结构,可以理解成被延迟的程序,这里就是timerDPC。这个PKDEFERRED_ROUTINE有4个参数……不讲开去了,本来好好的控制键盘LED,浅显易懂,往深了讲还是再开一帖的好。第三个参数为NULL

另外在写这篇文章的时候,偶然查到了liudanking的【讨论】简单的键盘sniff,为何无法正常工作?
http://bbs.pediy.com/showthread.php?t=113648恩,应该看的和我是同一本书吧,哈,那是个sniff键盘的程序,的确运行不起来,不过当时没在意~~这个问题值得思考,我也再想想吧


[培训]内核驱动高级班,冲击BAT一流互联网大厂工作,每周日13:00-18:00直播授课

上传的附件:
收藏
免费 7
支持
分享
最新回复 (7)
雪    币: 89
活跃值: (185)
能力值: ( LV9,RANK:270 )
在线值:
发帖
回帖
粉丝
2
有创意………………

沙发
2011-2-12 21:03
0
雪    币: 284
活跃值: (16)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
3
可以给个优秀
2011-2-12 21:06
0
雪    币: 578
活跃值: (14)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
4
可以考虑在某个程序里加入,当程序执行操作时,键盘上3个灯就开始玩起"跑马灯"
2011-2-12 22:25
0
雪    币: 123
活跃值: (295)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
5
把程序传上来玩玩。
2011-2-12 22:31
0
雪    币: 190
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
6
请问楼主看得是哪本书啊?
2011-2-13 08:49
0
雪    币: 137
活跃值: (26)
能力值: ( LV4,RANK:50 )
在线值:
发帖
回帖
粉丝
7
《ROOTKITS----Windows内核的安全防护》
2011-2-13 14:42
0
雪    币: 137
活跃值: (26)
能力值: ( LV4,RANK:50 )
在线值:
发帖
回帖
粉丝
8
这个都能精华……版主您太看得起我了。
这个应该是大二时候写的东西,现在看起来真的非常稚嫩,读写0x60 和0x64而已~~
谢谢鼓励。谢谢鼓励。
2012-6-3 15:31
0
游客
登录 | 注册 方可回帖
返回
//