C/C++:保存剪贴板中的截图为bmp文件(编译成dll供python调用)

获取更多文章和更新,请关注我的个人主页: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文件的数据构成为

  1. 写入事先手动设置好的文件头(占用14字节)
  2. 写入从剪贴板得到的图片数据

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;
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,233评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,013评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,030评论 0 241
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,827评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,221评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,542评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,814评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,513评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,225评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,497评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,998评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,342评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,986评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,055评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,812评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,560评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,461评论 2 266

推荐阅读更多精彩内容