写一个块设备驱动(5)

 

第5章

+---------------------------------------------------+ |                 写一个块设备驱动                  | +---------------------------------------------------+ | 作者：赵磊                                        | | email: [email protected]                      | +---------------------------------------------------+ | 文章版权归原作者所有。                            | | 大家可以自由转载这篇文章，但原版权信息必须保留。  | | 如需用于商业用途，请务必与原作者联系，若因未取得  | | 授权而收起的版权争议，由侵权者自行负责。          | +---------------------------------------------------+ 既然上一章结束时我们已经预告了本章的内容， 那么本章中我们就让这个块设备有能力告知操作系统它的“物理结构”。 当然，对于基于内存的块设备来说，什么样的物理结构并不重要， 这就如同从酒吧带mm回家时不需要打听她的姓名一样。 但如果不幸遇到的是兼职，并且带她去不入流的招待所时， 建议最好还是先串供一下姓名、生日和职业等信息， 以便JJ查房时可以伪装成情侣。 同样，如果要实现的是真实的物理块设备驱动， 那么返回设备的物理结构时大概不能这么随意。 对于块设备驱动程序而言，我们现在需要关注那条目前只有一行的struct block_device_operations simp_blkdev_fops结构。 到目前为止，它存在的目的仅仅是因为它必须存在，但马上我们将发现它存在的另一个目的：为块设备驱动添加获得块设备物理结构的接口。 对于具有极强钻研精神的极品读者来说，大概在第一章中就会自己去看struct block_device_operations结构，然后将发现这个结构其实还挺复杂： struct block_device_operations {         int (*open) (struct block_device *, fmode_t);         int (*release) (struct gendisk *, fmode_t);         int (*locked_ioctl) (struct block_device *, fmode_t, unsigned, unsigned long);         int (*ioctl) (struct block_device *, fmode_t, unsigned, unsigned long);         int (*compat_ioctl) (struct block_device *, fmode_t, unsigned, unsigned long);         int (*direct_access) (struct block_device *, sector_t,                                                 void **, unsigned long *);         int (*media_changed) (struct gendisk *);         int (*revalidate_disk) (struct gendisk *);         int (*getgeo)(struct block_device *, struct hd_geometry *);         struct module *owner; }; 在前几章中，我们邂逅过其中的owner成员变量，它用于存储这个结构的所有者，也就是我们的模块，因此我们做了如下的赋值： .owner                = THIS_MODULE, 而这一章中，我们将与它的同胞妹妹------getgeo也亲密接触一下。 我们要做的是： 1：在block_device_operations中增加getgeo成员变量初值的设定，指向我们的“获得块设备物理结构”函数。 2：实现我们的“获得块设备物理结构”函数。 第一步很简单，我们暂且为“获得块设备物理结构”函数取个名字叫simp_blkdev_getgeo()吧，也避免了在下文中把这么一大堆汉字拷来拷去。 在simp_blkdev_fops中添加.getgeo指向simp_blkdev_getgeo，也就是把simp_blkdev_fops结构改成这个样子： struct block_device_operations simp_blkdev_fops = {         .owner                = THIS_MODULE,         .getgeo                = simp_blkdev_getgeo, }; 第二步难一些，但也难不到哪去，在代码中的struct block_device_operations simp_blkdev_fops这行之前找个空点的场子，把如下函数插进去： static int simp_blkdev_getgeo(struct block_device *bdev,                 struct hd_geometry *geo) {         /*          * capacity        heads        sectors        cylinders          * 0~16M        1        1        0~32768          * 16M~512M        1        32        1024~32768          * 512M~16G        32        32        1024~32768          * 16G~...        255        63        2088~...          */         if (SIMP_BLKDEV_BYTES < 16 * 1024 * 1024) {                 geo->heads = 1;                 geo->sectors = 1;         } else if (SIMP_BLKDEV_BYTES < 512 * 1024 * 1024) {                 geo->heads = 1;                 geo->sectors = 32;         } else if (SIMP_BLKDEV_BYTES < 16ULL * 1024 * 1024 * 1024) {                 geo->heads = 32;                 geo->sectors = 32;         } else {                 geo->heads = 255;                 geo->sectors = 63;         }         geo->cylinders = SIMP_BLKDEV_BYTES>>9/geo->heads/geo->sectors;         return 0; } 因为这里我们用到了struct hd_geometry结构，所以还要增加一行#include <linux/hdreg.h>。 这个函数的目的，是选择适当的物理结构信息装入struct hd_geometry *geo结构。 当然，为了克服上一章中只能分成2个区的问题，我们应该尽可能增加磁道的数量。 希望读者不要理解成分几个区就需要几个磁道，这意味着一个磁道一个区，也意味着每个区必须一般大小。 由于分区总是以磁道为边界，尽可能增加磁道的数量不仅仅是为了让块设备容纳更多的分区， 更重要的是让分区的实际大小更接近于分区时的指定值，也就是提高实际做出的分区容量的精度。 不过对于设置的物理结构值，还存在一个限制，就是struct hd_geometry中的数值上限。 我们看struct hd_geometry的内容： struct hd_geometry {         unsigned char heads;         unsigned char sectors;         unsigned short cylinders;         unsigned long start; }; unsigned char的磁头数和每磁道扇区数决定了其255的上限，同样，unsigned short的磁道数决定了其65535的上限。 这还不算，但在前一章中，我们知道对于现代硬盘，磁头数和每磁道扇区数通常取的值是255和63， 再组合上这里的65535的磁道数上限，hd_geometry能够表示的最大块设备容量是255*63*65535*512/1024/1024/1024=502G。 显然目前linux支持的最大硬盘容量大于502G，那么对于这类块设备，内核是如何通过hd_geometry结构表示其物理结构的呢？ 诀窍不在内核，而在于用户态程序如fdisk等通过内核调用获得hd_geometry结构后， 会舍弃hd_geometry.cylinders内容，取而代之的是直接通过hd_geometry中的磁头数和每磁道扇区数以及硬盘大小去计算磁道数。 因此对于超过502G的硬盘，由于用户程序得出的磁道数与hd_geometry.cylinders无关，所以我们往往在fdisk中能看到这块硬盘的磁道数大于65535。 刚才扯远了，现在言归正题，我们决定让这个函数对于任何尺寸的块设备，总是试图返回比较漂亮的物理结构。 漂亮意味着返回的物理结构既要保证拥有足够多的磁道，也要保证磁头数和每磁道扇区数不超过255和63，同时最好使用程序员看起来比较顺眼的数字， 如：1、2、4、8、16、32、64等。 当然，我们也希望找到某个One Shot公式适用于所有大小的块设备，但很遗憾目前作者没找到，因此采用了分段计算的方法： 首先考虑容量很小的块设备：   即使磁头数和每磁道扇区数都是1，磁道数也不够多时，我们会将磁头数和每磁道扇区数都固定为1，以使磁道数尽可能多，以提高分区的精度。   因此磁道数随块设备容量而上升。   虽然我们已经知道了磁道数其实可以超过unsigned short的65535上限，但在这里却没有必要，因此我们要给磁道数设置一个上限。   因为不想让上限超过65535，同时还希望上限也是一个程序员喜欢的数字，因此这里选择了32768。   当然，当磁道数超过32768时，已经意味着块设备容量不那么小了，也就没有必要使用这种情况中如此苛刻的磁头数和每磁道扇区数了。   简单来说，当块设备容量小于1个磁头、每磁道1扇区和32768个磁道对应的容量--也就是16M时，我们将按照这种情况处理。 然后假设块设备容量已经大于16M了：   我们希望保证块设备包含足够多的磁道，这里我们认为1024个磁道应该不少了。   磁道的最小值发生在块设备容量为16M的时候，这时使用1024作为磁道数，可以计算出磁头数*每磁道扇区数=32。   这里暂且把磁头数和每磁道扇区数固定为1和32，而让磁道数随着块设备容量的增大而增加。   同时，我们还是磁道的上限设置成32768，这时的块设备容量为512M。   总结来说，当块设备容量在16M和512M之间时，我们把磁头数和每磁道扇区数固定为1和32。 然后对于容量大于512M的块设备：   与上述处理相似，当块设备容量在512M和16G之间时，我们把磁头数和每磁道扇区数固定为32和32。 最后的一种情况：   块设备已经足够大了，大到即使我们使用磁头数和每磁道扇区数的上限，   也能获得足够多的磁道数。这时把磁头数和每磁道扇区数固定为255和63。   至于磁道数就算出多少是多少了，即使超过unsigned short的上限也无所谓，反正用不着。 随着这个函数解说到此结束，我们对代码的修改也结束了。 现在开始试验： 编译和加载： # make make -C /lib/modules/2.6.27.4/build SUBDIRS=/mnt/host_test/simp_blkdev/simp_blkdev_step05 modules make[1]: Entering directory `/mnt/ltt-kernel'   CC [M]  /mnt/host_test/simp_blkdev/simp_blkdev_step05/simp_blkdev.o   Building modules, stage 2.   MODPOST 1 modules   CC      /mnt/host_test/simp_blkdev/simp_blkdev_step05/simp_blkdev.mod.o   LD [M]  /mnt/host_test/simp_blkdev/simp_blkdev_step05/simp_blkdev.ko make[1]: Leaving directory `/mnt/ltt-kernel' # insmod simp_blkdev.ko # 用fdisk打开设备文件 # fdisk /dev/simp_blkdev Device contains neither a valid DOS partition table, nor Sun, SGI or OSF disklabel Building a new DOS disklabel. Changes will remain in memory only, until you decide to write them. After that, of course, the previous content won't be recoverable. Warning: invalid flag 0x0000 of partition table 4 will be corrected by w(rite) Command (m for help): 看看设备的物理结构： Command (m for help): p Disk /dev/simp_blkdev: 16 MB, 16777216 bytes 1 heads, 32 sectors/track, 1024 cylinders Units = cylinders of 32 * 512 = 16384 bytes            Device Boot      Start         End      Blocks   Id  System Command (m for help): 我们发现，现在的设备有1个磁头、32扇区每磁道、1024个磁道。 这是符合代码中的处理的。 本章的内容也不是太难，连同上一章，我们已经休息2章了。 聪明的读者可能已经猜到作者打算说什么了。 不错，下一章会有一个surprise。 <未完，待续>