获取更多文章和更新，请关注我的个人主页：https://leiting6.cn
最近在研究windows API，仿佛发现新大陆一般 --- 想在windows系统下面玩的更溜一些，还是得了解系统本身的API函数才行。其实python下通过安装pywin32也可以调用win32API，但是总觉得是瞎折腾；而且，毕竟C语言是学生时期的启蒙语言，直接利用VC编写动态链接库（dll）也不是难事。刚好这些天想写一个自动保存截图的小工具，经过一番搜索和自己的修改完善，目前已经完成了核心部分。
2018/10/21更新
除了生成dll，也稍微修改一下代码生成了exe版本，方便更多语言调用，比如最近一直在研究的autohotkey，毕竟在ahk中调用exe更简单直接。
dll和exe下载链接：点击跳转
C/C++部分
这部分只有一个函数，详情如下：
- 函数名：SaveBmpFileFromClipboard()
- 参数: char *file_name （BMP文件的文件名，可以带上完整路径）
- 功能：从剪贴板保存图片为BMP格式到本地
- 返回值： 成功保存则返回0， 遇到错误则返回1
需要注意的是，想要保存一个BMP文件，还是要先了解BMP文件的构成的，点击查看。
简单的说：
BMP文件=文件头（14字节）+信息头（40字节，全部默认，实测可不写入）+调色板（可选，不写入）+图片数据
而我这里生成的：
BMP文件=文件头（14字节）+图片数据
因此BMP文件的数据构成为
- 写入事先手动设置好的文件头（占用14字节）
- 写入从剪贴板得到的图片数据
return值说明
0：保存bmp文成功
1：剪贴板中内容不是DIB格式
2：打开剪贴板失败
3：路径无法访问/没有操作权限
// SaveBmpFileFromClipboard.cpp : 定义 DLL 应用程序的导出函数。
//
//返回值：
//0 - 存储bmp文件成功
//1 - 剪贴板内容非DIB格式
//2 - 打开剪贴板失败
//3 - 存储bmp文件失败
#include "stdafx.h"
#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <io.h>
#define EXPORT extern "C" _declspec(dllexport)
EXPORT int SaveBmpFileFromClipboard(const char *file_name = "capture.bmp")
{
if (!OpenClipboard(NULL)) //尝试打开剪贴板
{
printf("%s", "Open clipboard fail!\n");
CloseClipboard(); //关闭剪贴板
return 2;
}
HANDLE hDib;
LPVOID lpDib;
BITMAPFILEHEADER bfBuf;
if (IsClipboardFormatAvailable(CF_DIB)) //检查剪贴板内容是否为DIB类型的数据
{
hDib = GetClipboardData(CF_DIB); //获取剪贴板内容并存入hDib句柄
lpDib = GlobalLock(hDib); //利用GlobalLock锁住hDib占用的内存，并返回内存块的首字节指针
printf("%s", "Getting data...\n");
int iData = GlobalSize(lpDib); //利用GlobalSize获取上述锁住的内存块的大小
//bfBuf是BITMAPFILEHEADER结构体类型，下面手动指定BMP文件头参数
bfBuf.bfType = 0x4d42; //"BM"的ASCII码的hex值
bfBuf.bfSize = iData + sizeof(BITMAPFILEHEADER); //BMP文件大小=数据大小+文件头大小
bfBuf.bfReserved1 = 0; //固定值
bfBuf.bfReserved2 = 0; //固定值
bfBuf.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER); //文件数据写入时的偏移量=文件头大小+信息头大小
printf("%s", "Writing files...\n");
//下面是创建文件并写入文件头和写入文件数据，即生成了BMP文件
DWORD dwWriten;
HANDLE hf = CreateFileA(file_name, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
WriteFile(hf, &bfBuf, sizeof(BITMAPFILEHEADER), &dwWriten, NULL); //写入文件头
WriteFile(hf, lpDib, iData, &dwWriten, NULL); //写入图片数据
CloseHandle(hf);
if (!_access(file_name, 0))
{
printf("%s", "Bmp file saved!\n");
}
else
{
printf("%s", "Bmp file not saved!\n");
CloseClipboard(); //关闭剪贴板
return 3;
}
GlobalUnlock(hDib); //解锁被锁住的内存块
GlobalFree(hDib); //彻底释放hDib
}
else
{
printf("%s", "Not DIB format!\n");
CloseClipboard(); //关闭剪贴板
return 1;
}
CloseClipboard(); //关闭剪贴板
return 0;
}
Python部分
接下来python部分的代码就很简单了，导入ctypes模块，利用ctypes.cdll加载上述dll，传入代表文件路径和文件名的字符串：
import ctypes
file_name = r'D:\\clip.bmp'
dll = ctypes.cdll.LoadLibrary('SaveBmpFileFromClipboard.dll')
dll.SaveBmpFileFromClipboard(file_name.encode('gbk'))
exe版本
在dll上小修了一下，用main函数把代码包裹起来，这样就可以生成一个更方便调用和使用的exe格式可执行文件。需要注意的是，报错时的return值重新编排了，是为了适应autohotkey中RunWait函数调用exe文件的返回值，觉得不习惯的可以自行修改重新编译，问题不大。
return值说明
0：保存bmp文成功
11：剪贴板中内容不是DIB格式
12：打开剪贴板失败
13：路径无法访问/没有操作权限
14：没有传入参数（至少得有一个参数，即截图文件的完整路径）
完整代码如下：
// SaveBmpFile.cpp: 定义控制台应用程序的入口点。
//
//返回值：
//0 - 存储bmp文件成功
//11 - 剪贴板内容非DIB格式
//12 - 打开剪贴板失败
//13 - 存储bmp文件失败
//14 - 没有传入参数
#include "stdafx.h"
#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <io.h>
int main(int argc, char* argv[])
{
char* file_name;
if (argc < 2)
{
printf("No file name given!");
return 14;
}
else file_name = argv[1];
if (!OpenClipboard(NULL)) //尝试打开剪贴板
{
printf("%s", "Open clipboard fail!\n");
CloseClipboard(); //关闭剪贴板
return 12;
}
HANDLE hDib;
LPVOID lpDib;
BITMAPFILEHEADER bfBuf;
if (IsClipboardFormatAvailable(CF_DIB)) //检查剪贴板内容是否为DIB类型的数据
{
hDib = GetClipboardData(CF_DIB); //获取剪贴板内容并存入hDib句柄
lpDib = GlobalLock(hDib); //利用GlobalLock锁住hDib占用的内存，并返回内存块的首字节指针
printf("%s", "Getting data...\n");
int iData = GlobalSize(lpDib); //利用GlobalSize获取上述锁住的内存块的大小
//bfBuf是BITMAPFILEHEADER结构体类型，下面手动指定BMP文件头参数
bfBuf.bfType = 0x4d42; //"BM"的ASCII码的hex值
bfBuf.bfSize = iData + sizeof(BITMAPFILEHEADER); //BMP文件大小=数据大小+文件头大小
bfBuf.bfReserved1 = 0; //固定值
bfBuf.bfReserved2 = 0; //固定值
bfBuf.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER); //文件数据写入时的偏移量=文件头大小+信息头大小
printf("%s", "Writing files...\n");
//下面是创建文件并写入文件头和写入文件数据，即生成了BMP文件
DWORD dwWriten;
HANDLE hf;
hf = CreateFileA(file_name, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
printf("%s", "Start writing...\n");
WriteFile(hf, &bfBuf, sizeof(BITMAPFILEHEADER), &dwWriten, NULL); //写入文件头
printf("%s", "File head writen!\n");
WriteFile(hf, lpDib, iData, &dwWriten, NULL); //写入图片数据
printf("%s", "Pic data writen!\n");
CloseHandle(hf);
printf("%s", "Close handle!\n");
if (!_access(file_name, 0))
{
printf("%s", "Bmp file saved!\n");
}
else
{
printf("%s", "Bmp file not saved!\n");
CloseClipboard(); //关闭剪贴板
return 13;
}
GlobalUnlock(hDib); //解锁被锁住的内存块
GlobalFree(hDib); //彻底释放hDib
}
else
{
printf("%s", "Not DIB format!\n");
CloseClipboard(); //关闭剪贴板
return 11;
}
CloseClipboard(); //关闭剪贴板
return 0;
}
网友评论