Chinaunix首页 | 论坛 | 博客

birdofpreybirdofprey.blog.chinaunix.net

首页　| 　博文目录　| 　关于我

niao5929

  • 博客访问： 7116134
  • 博文数量： 3857
  • 博客积分： 6409
  • 博客等级： 准将
  • 技术积分： 15948
  • 用 户 组： 普通用户
  • 注册时间： 2008-09-02 16:48
个人简介

迷彩 潜伏 隐蔽 伪装

文章分类

全部博文（3857）

文章存档

2017年（5）

2016年（63）

2015年（927）

2014年（677）

2013年（807）

2012年（1241）

2011年（67）

2010年（7）

2009年（36）

2008年（28）

我的朋友
最近访客
推荐博文
相关博文
C语言在linux系统中获取cpuid, mac地址完成代码

分类： C/C++

2014-09-04 09:49:37

原文地址：C语言在linux系统中获取cpuid, mac地址完成代码 作者：zhongbai

在网上找了好长时间关于获取CPUID的代码，都不是完全版本，回去还得改，自己写好了，是借鉴dmidecode中的代码成完的，希望多多提意见，谢谢。
这样做的缺点是必须得有root权限才可以，其它用户的权限是不能进行访问/dev/mem这个文件。
types.h

点击(此处)折叠或打开

  1. #ifndef TYPES_H
  2. #define TYPES_H


  5. typedef unsigned char u8;
  6. typedef unsigned short u16;
  7. typedef signed short i16;
  8. typedef unsigned int u32;

  10. #ifdef BIGENDIAN
  11. typedef struct {
  12.     u32 h;
  13.     u32 l;
  14. } u64;
  15. #else
  16. typedef struct {
  17.     u32 l;
  18.     u32 h;
  19. } u64;
  20. #endif

  22. #ifdef ALIGNMENT_WORKAROUND
  23. static inline u64 U64(u32 low, u32 high)
  24. {
  25.     u64 self;

  27.     self.l = low;
  28.     self.h = high;

  30.     return self;
  31. }
  32. #endif

  34. #ifdef ALIGNMENT_WORKAROUND
  35. #    ifdef BIGENDIAN
  36. #    define WORD(x) (u16)((x)[1] + ((x)[0] << 8))
  37. #    define DWORD(x) (u32)((x)[3] + ((x)[2] << 8) + ((x)[1] << 16) + ((x)[0] << 24))
  38. #    define QWORD(x) (U64(DWORD(x + 4), DWORD(x)))
  39. #    else /* BIGENDIAN */
  40. #    define WORD(x) (u16)((x)[0] + ((x)[1] << 8))
  41. #    define DWORD(x) (u32)((x)[0] + ((x)[1] << 8) + ((x)[2] << 16) + ((x)[3] << 24))
  42. #    define QWORD(x) (U64(DWORD(x), DWORD(x + 4)))
  43. #    endif /* BIGENDIAN */
  44. #else /* ALIGNMENT_WORKAROUND */
  45. #define WORD(x) (u16)(*(const u16 *)(x))
  46. #define DWORD(x) (u32)(*(const u32 *)(x))
  47. #define QWORD(x) (*(const u64 *)(x))
  48. #endif /* ALIGNMENT_WORKAROUND */

  50. #endif


util.h

点击(此处)折叠或打开

  1. #include <sys/types.h>
  2. #include <sys/ioctl.h>
  3. #include <sys/socket.h>
  4. #include <netinet/in.h>
  5. #include <net/if.h>
  6. #include <string.h>

  8. #include "types.h"

  10. struct dmi_header
  11. {
  12.         u8 type;
  13.         u8 length;
  14.         u16 handle;
  15.         u8 *data;
  16. };

  18. int checksum(const u8 *buf, size_t len);
  19. void *mem_chunk(size_t base, size_t len, const char *devmem);
  20. u64 u64_range(u64 start, u64 end);
  21. void to_dmi_header(struct dmi_header *h, u8 *data);
  22. int get_cpuid(char *cpuid);
  23. int get_mac(char* mac);
util.c

点击(此处)折叠或打开

  1. #include <sys/types.h>
  2. #include <sys/stat.h>

  4. #ifdef USE_MMAP
  5. #include <sys/mman.h>
  6. #ifndef MAP_FAILED
  7. #define MAP_FAILED ((void *) -1)
  8. #endif /* !MAP_FAILED */
  9. #endif /* USE MMAP */

  11. #include <stdio.h>
  12. #include <stdlib.h>
  13. #include <unistd.h>
  14. #include <string.h>
  15. #include <fcntl.h>
  16. #include <errno.h>

  18. #include "types.h"
  19. #include "util.h"

  21. static int myread(int fd, u8 *buf, size_t count, const char *prefix)
  22. {
  23.     ssize_t r = 1;
  24.     size_t r2 = 0;

  26.     while (r2 != count && r != 0)
  27.     {
  28.         r = read(fd, buf + r2, count - r2);
  29.         if (r == -1)
  30.         {
  31.             if (errno != EINTR)
  32.             {
  33.                 close(fd);
  34.                 perror(prefix);
  35.                 return -1;
  36.             }
  37.         }
  38.         else
  39.             r2 += r;
  40.     }

  42.     if (r2 != count)
  43.     {
  44.         close(fd);
  45.         fprintf(stderr, "%s: Unexpected end of file\n", prefix);
  46.         return -1;
  47.     }

  49.     return 0;
  50. }

  52. int checksum(const u8 *buf, size_t len)
  53. {
  54.     u8 sum = 0;
  55.     size_t a;

  57.     for (a = 0; a < len; a++)
  58.         sum += buf[a];
  59.     return (sum == 0);
  60. }

  62. /*
  63.  * Copy a physical memory chunk into a memory buffer.
  64.  * This function allocates memory.
  65.  */
  66. void *mem_chunk(size_t base, size_t len, const char *devmem)
  67. {
  68.     void *p;
  69.     int fd;
  70. #ifdef USE_MMAP
  71.     size_t mmoffset;
  72.     void *mmp;
  73. #endif

  75.     if ((fd = open(devmem, O_RDONLY)) == -1)
  76.     {
  77.         perror(devmem);
  78.         return NULL;
  79.     }

  81.     if ((p = malloc(len)) == NULL)
  82.     {
  83.         perror("malloc");
  84.         return NULL;
  85.     }

  87. #ifdef USE_MMAP
  88. #ifdef _SC_PAGESIZE
  89.     mmoffset = base % sysconf(_SC_PAGESIZE);
  90. #else
  91.     mmoffset = base % getpagesize();
  92. #endif /* _SC_PAGESIZE */
  93.     /*
  94.      * Please note that we don't use mmap() for performance reasons here,
  95.      * but to workaround problems many people encountered when trying
  96.      * to read from /dev/mem using regular read() calls.
  97.      */
  98.     mmp = mmap(0, mmoffset + len, PROT_READ, MAP_SHARED, fd, base - mmoffset);
  99.     if (mmp == MAP_FAILED)
  100.         goto try_read;

  102.     memcpy(p, (u8 *)mmp + mmoffset, len);

  104.     if (munmap(mmp, mmoffset + len) == -1)
  105.     {
  106.         fprintf(stderr, "%s: ", devmem);
  107.         perror("munmap");
  108.     }

  110.     goto out;

  112. #endif /* USE_MMAP */

  114. try_read:
  115.     if (lseek(fd, base, SEEK_SET) == -1)
  116.     {
  117.         fprintf(stderr, "%s: ", devmem);
  118.         perror("lseek");
  119.         free(p);
  120.         return NULL;
  121.     }

  123.     if (myread(fd, p, len, devmem) == -1)
  124.     {
  125.         free(p);
  126.         return NULL;
  127.     }

  129. out:
  130.     if (close(fd) == -1)
  131.         perror(devmem);

  133.     return p;
  134. }

  136. void to_dmi_header(struct dmi_header *h, u8 *data)
  137. {
  138.     h->type = data[0];
  139.     h->length = data[1];
  140.     h->handle = WORD(data + 2);
  141.     h->data = data;
  142. }

  144. int get_cpuid(char *cpuid)
  145. {
  146.     char devmem[10] = "/dev/mem";
  147.     u16 len;
  148.     u16 num;
  149.     size_t fp;
  150.     u8 *buf=NULL, *nbuf=NULL, *data, *p;
  151.     int i = 0;
  152.     if ((buf = mem_chunk(0xF0000, 0x10000, devmem)) == NULL)
  153.     {
  154.         free(buf);
  155.         return 0;
  156.     }

  158.     for (fp = 0; fp <= 0xFFF0; fp += 16)
  159.     {
  160.         if(memcmp(buf + fp, "_SM_", 4) == 0 && fp <= 0xFFE0)
  161.         {
  162.             len = WORD(buf + fp + 0x16);
  163.             num = WORD(buf + fp + 0x1C);

  165.             if (!checksum(buf + fp, (buf + fp)[0x05]) || memcmp(buf + fp + 0x10, "_DMI_", 5) != 0 || !checksum(buf + fp + 0x10, 0x0F))
  166.             {
  167.                 free(buf);
  168.                 return 0;
  169.             }
  170.             if ((nbuf = mem_chunk(DWORD(buf + fp + 0x18), len, devmem)) == NULL)
  171.             {
  172.                 fprintf(stderr, "Table is unreachable, sorry.\n");
  173.                 free(buf);
  174.                 free(nbuf);
  175.                 return 0;
  176.             }
  177.             data = nbuf;
  178.             while (i < num && data+4 <= nbuf + len)
  179.             {
  180.                 u8 *next;
  181.                 struct dmi_header h;

  183.                 to_dmi_header(&h, data);

  185.                 if (h.length < 4)
  186.                 {
  187.                     printf("Invalid entry length (%u). DMI table is "
  188.                            "broken! Stop.\n\n", (unsigned int)h.length);
  189.                     free(buf);
  190.                     free(nbuf);
  191.                     return 0;
  192.                 }

  194.                 next = data + h.length;
  195.                 while (next - nbuf + 1 < len && (next[0] != 0 || next[1] != 0))
  196.                     next++;
  197.                 next += 2;
  198.                 if (h.type ==4)
  199.                 {
  200.                     p = h.data + 0x08;
  201.                     //printf("CPUID:%02X%02X%02X%02X%02X%02X%02X%02X\n", p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7]);
  202.                     sprintf(cpuid, "%02X%02X%02X%02X%02X%02X%02X%02X", p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7]);
  203.                     free(buf);
  204.                     free(nbuf);
  205.                     return 1;
  206.                 }
  207.                 data = next;
  208.                 i++;
  209.             }
  210.             fp += 16;
  211.         }
  212.     }
  213.     free(buf);
  214.     free(nbuf);
  215.     return 0;
  216. }

  218. int get_mac(char* mac)
  219. {
  220.     int sockfd;
  221.     struct ifreq tmp;
  222.     char mac_addr[30];

  224.     sockfd = socket(AF_INET, SOCK_STREAM, 0);
  225.     if( sockfd < 0)
  226.     {
  227.         perror("create socket fail\n");
  228.         return -1;
  229.     }

  231.     memset(&tmp,0,sizeof(struct ifreq));
  232.     strncpy(tmp.ifr_name,"eth0",sizeof(tmp.ifr_name)-1);
  233.     if( (ioctl(sockfd,SIOCGIFHWADDR,&tmp)) < 0 )
  234.     {
  235.         printf("mac ioctl error\n");
  236.         return -1;
  237.     }

  239.     sprintf(mac_addr, "%02x%02x%02x%02x%02x%02x",
  240.             (unsigned char)tmp.ifr_hwaddr.sa_data[0],
  241.             (unsigned char)tmp.ifr_hwaddr.sa_data[1],
  242.             (unsigned char)tmp.ifr_hwaddr.sa_data[2],
  243.             (unsigned char)tmp.ifr_hwaddr.sa_data[3],
  244.             (unsigned char)tmp.ifr_hwaddr.sa_data[4],
  245.             (unsigned char)tmp.ifr_hwaddr.sa_data[5]
  246.             );
  247.     close(sockfd);
  248.     memcpy(mac,mac_addr,strlen(mac_addr));

  250.     return 0;
  251. }
getsysinfo.c

点击(此处)折叠或打开

  1. #include <stdio.h>
  2. #include "util.h"

  4. int main()
  5. {
  6.     char cpuid[20], mac[20];
  7.     get_cpuid(cpuid);
  8.     printf("CPUID:%s\n", cpuid);
  9.     get_mac(mac);
  10.     printf("MAC:%s\n", mac);
  11.     return 0;
  12. }
Makefile

点击(此处)折叠或打开

  1. # Copyright (C) Zhou Zhongbai 46784314@qq.com
  3. CC = gcc
  4. CFLAGS = -W -Wall -Wshadow -Wstrict-prototypes -Wpointer-arith -Wcast-qual \
  5. -Wcast-align -Wwrite-strings -Wmissing-prototypes -Winline -Wundef
  6. #CFLAGS += -DBIGENDIAN
  7. #CFLAGS += -DALIGNMENT_WORKAROUND
  9. # When debugging, disable -O2 and enable -g.
  10. CFLAGS += -O2
  11. #CFLAGS += -g
  13. # Pass linker flags here
  14. LDFLAGS =
  16. DESTDIR =
  17. prefix = /usr/local
  18. sbindir = $(prefix)/sbin
  19. mandir = $(prefix)/share/man
  20. man8dir = $(mandir)/man8
  21. docdir = $(prefix)/share/doc/getcpuid
  23. INSTALL := install
  24. INSTALL_DATA := $(INSTALL) -m 644
  25. INSTALL_DIR := $(INSTALL) -m 755 -d
  26. INSTALL_PROGRAM := $(INSTALL) -m 755
  27. RM := rm -f
  29. # BSD make provides $MACHINE, but GNU make doesn't
  30. MACHINE ?= $(shell uname -m 2>/dev/null)
  32. PROGRAMS = getsysinfo
  34. all : $(PROGRAMS)
  36. #
  37. # Programs
  38. #
  40. getsysinfo : getsysinfo.o util.o
  41. $(CC) $(LDFLAGS) getsysinfo.o util.o -o $@
  43. #
  44. # Objects
  45. #
  47. getsysinfo.o : getsysinfo.c util.h
  48. $(CC) $(CFLAGS) -c $< -o $@
  49. util.o : util.c util.h
  50. $(CC) $(CFLAGS) -c $< -o $@
  52. #
  53. # Commands
  54. #
  56. strip : $(PROGRAMS)
  57. strip $(PROGRAMS)
  59. install : install-bin install-man install-doc
  61. uninstall : uninstall-bin uninstall-man uninstall-doc
  63. install-bin : $(PROGRAMS)
  64. $(INSTALL_DIR) $(DESTDIR)$(sbindir)
  65. for program in $(PROGRAMS) ; do \
  66. $(INSTALL_PROGRAM) $$program $(DESTDIR)$(sbindir) ; done
  68. uninstall-bin :
  69. for program in $(PROGRAMS) ; do \
  70. $(RM) $(DESTDIR)$(sbindir)/$$program ; done
  72. install-man :
  73. $(INSTALL_DIR) $(DESTDIR)$(man8dir)
  74. for program in $(PROGRAMS) ; do \
  75. $(INSTALL_DATA) man/$$program.8 $(DESTDIR)$(man8dir) ; done
  77. uninstall-man :
  78. for program in $(PROGRAMS) ; do \
  79. $(RM) $(DESTDIR)$(man8dir)/$$program.8 ; done
  81. install-doc :
  82. $(INSTALL_DIR) $(DESTDIR)$(docdir)
  83. $(INSTALL_DATA) README $(DESTDIR)$(docdir)
  84. $(INSTALL_DATA) CHANGELOG $(DESTDIR)$(docdir)
  85. $(INSTALL_DATA) AUTHORS $(DESTDIR)$(docdir)
  87. uninstall-doc :
  88. $(RM) -r $(DESTDIR)$(docdir)
  90. clean :
  91. $(RM) *.o $(PROGRAMS)

阅读(743) | 评论(0) | 转发(0) |
0

上一篇：深入理解Session与Cookie

下一篇：C语言中宏定义的#define中有空格（即# define）

给主人留下些什么吧！~~
关于我们 | 关于IT168 | 联系方式 | 广告合作 | 法律声明 | 免费注册

Copyright 2001-2010 ChinaUnix.net All Rights Reserved 北京皓辰网域网络信息技术有限公司. 版权所有

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6