编写内核模块，获取磁盘的disk_stats信息

motorn

我正在写一个内核模块，基于linux 2.6.32。这个模块的功能是获取所有磁盘的disk_stats，最后展示的结果与文件/sys/block/sda/stat的内容相似。

想问一下，对于block device，有没有类似于for_each_netdev的宏/函数，因为一台机器可能有多个磁盘，我需要遍历这些磁盘。或者，有没有什么方法可以根据块设备的名字（比如sda）获取到相应的数据结构。一块磁盘的disk_stats，存放在struct hd_struct，如何获取到这个hd_struct？

我看了linux2.6.32/block/genhd.c，/sys/block/sda/stat的内容是由genhd.c的diskstats_show函数产生的，不过这个函数被调用的时候，第二个参数应该是什么？如何获取到这个参数？

对于block device，linux 2.4有一个全局变量gendisk_head，但是linux2.6好像找不到了。

求大牛帮忙，感激不尽。

Tinnal

回复 1# motorn

你要的东西就在genhd.c这个文件里。

1. /*
   
2. * aggregate disk stat collector.  Uses the same stats that the sysfs
   
3. * entries do, above, but makes them available through one seq_file.
   
4. * Watching a few disks may be efficient through sysfs, but watching
   
5. * all of them will be more efficient through this interface.
   
6. *
   
7. * The output looks suspiciously like /proc/partitions with a bunch of
   
8. * extra fields.
   
9. */
   
10. 
    
11. /* iterator */
    
12. static void *diskstats_start(struct seq_file *part, loff_t *pos)
    
13. {
    
14.         loff_t k = *pos;
    
15.         struct list_head *p;
    
16. 
    
17.         mutex_lock(&block_subsys_lock);
    
18.         list_for_each(p, &block_subsys.list)
    
19.                 if (!k--)
    
20.                         return list_entry(p, struct gendisk, kobj.entry);
    
21.         return NULL;
    
22. }
    
23. 
    
24. static void *diskstats_next(struct seq_file *part, void *v, loff_t *pos)
    
25. {
    
26.         struct list_head *p = ((struct gendisk *)v)->kobj.entry.next;
    
27.         ++*pos;
    
28.         return p==&block_subsys.list ? NULL :
    
29.                 list_entry(p, struct gendisk, kobj.entry);
    
30. }
    
31. 
    
32. static void diskstats_stop(struct seq_file *part, void *v)
    
33. {
    
34.         mutex_unlock(&block_subsys_lock);
    
35. }

复制代码

你太不认真的。你怎么只看diskstats_show，不看和他一套的函数呢。

如果对seq_operations不熟，自己去baidu.com!!

motorn

回复 2# Tinnal


    你好，你提到的diskstats_{start, next, stop}，在linux 2.6.25及以前的版本中存在，之后就不存在了，看这里。不过我在linux 2.6.32中找到了类似的一组函数：disk_seqf_{start, next, stop}，看这里，这组函数从linux 2.6.31开始出现。

    但是有一个问题：对于linux 2.6.25，diskstats_start的实现中用到了一个变量block_subsys，它在drivers/block/genhd.c中定义的，在这里，可以看到，它是static的，即文件内可见。所以，你写个内核模块装载到内核之后，你是看不到这个变量的。所以也就没法把这几个函数“依葫芦画瓢”地再实现一遍。（为什么要再实现一遍？因为genhd.c里这几个函数是static的，而你要在你的模块里获得类似的功能）

    同样的，对于linux 2.6.32，disk_seqf_start的实现中用到了一个变量block_class，它也是在block/genhd.c中定义的，看这里，可以看到，它虽不是static的，但是它在定义之后并没有EXPORT_SYMBOL。linux内核在2.4之后，凡是没有EXPORT_SYMBOL的全局变量，不管它是不是static的，它都只是文件可见，不是真正的全局（即整个模块可见）。所以，我们遇到了同样的问题。

motorn

回复 2# Tinnal

1. #ifdef CONFIG_PROC_FS
   
2. /* iterator */
   
3. static void *disk_seqf_start(struct seq_file *seqf, loff_t *pos)
   
4. {
   
5.         loff_t skip = *pos;
   
6.         struct class_dev_iter *iter;
   
7.         struct device *dev;
   
8. 
   
9.         iter = kmalloc(sizeof(*iter), GFP_KERNEL);
   
10.         if (!iter)
    
11.                 return ERR_PTR(-ENOMEM);
    
12. 
    
13.         seqf->private = iter;
    
14.         class_dev_iter_init(iter, &block_class, NULL, &disk_type);
    
15.         do {
    
16.                 dev = class_dev_iter_next(iter);
    
17.                 if (!dev)
    
18.                         return NULL;
    
19.         } while (skip--);
    
20. 
    
21.         return dev_to_disk(dev);
    
22. }
    
23. 
    
24. static void *disk_seqf_next(struct seq_file *seqf, void *v, loff_t *pos)
    
25. {
    
26.         struct device *dev;
    
27. 
    
28.         (*pos)++;
    
29.         dev = class_dev_iter_next(seqf->private);
    
30.         if (dev)
    
31.                 return dev_to_disk(dev);
    
32. 
    
33.         return NULL;
    
34. }
    
35. 
    
36. static void disk_seqf_stop(struct seq_file *seqf, void *v)
    
37. {
    
38.         struct class_dev_iter *iter = seqf->private;
    
39. 
    
40.         /* stop is called even after start failed :-( */
    
41.         if (iter) {
    
42.                 class_dev_iter_exit(iter);
    
43.                 kfree(iter);
    
44.         }
    
45. }
    

复制代码

Tinnal

回复 4# motorn

这个函数是你想要的。

1. void class_dev_iter_init(struct class_dev_iter *iter, struct class *class,
   
2.                          struct device *start, const struct device_type *type)
   
3. {
   
4.         struct klist_node *start_knode = NULL;
   
5. 
   
6.         if (start)
   
7.                 start_knode = &start->knode_class;
   
8.         klist_iter_init_node(&class->p->class_devices, &iter->ki, start_knode);
   
9.         iter->type = type;
   
10. }
    
11. EXPORT_SYMBOL_GPL(class_dev_iter_init);

复制代码

这也是你要的。

1. void class_dev_iter_init(struct class_dev_iter *iter, struct class *class,
   
2.                          struct device *start, const struct device_type *type)
   
3. {
   
4.         struct klist_node *start_knode = NULL;
   
5. 
   
6.         if (start)
   
7.                 start_knode = &start->knode_class;
   
8.         klist_iter_init_node(&class->p->class_devices, &iter->ki, start_knode);
   
9.         iter->type = type;
   
10. }
    
11. EXPORT_SYMBOL_GPL(class_dev_iter_init);

复制代码

至于其他，你自己写呀，大哥。

1. struct class block_class = {
   
2.         .name                = "block",
   
3. };
   

复制代码

这个这么简单的的定义你也非得要用别人的？ 这些函数都只是让你参考，也不是让你直接去用，而且也没必要去抄。这几个函数代码量又不大，你抄来干嘛。
你要真纠结，就把他们EXPORT_SYMBOL_GPL。{:2_174:}