qrcode/qrcode.c at master · selfcustody/qrcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
// Include required definitions first.
#include "py/obj.h"
#include "py/runtime.h"
#include "py/builtin.h"
#include "qrcodegen.h"
#include "string.h"

STATIC mp_obj_t qrcode_encode(mp_obj_t text_obj) {
    mp_buffer_info_t bufinfo;
    mp_get_buffer_raise(text_obj, &bufinfo, MP_BUFFER_READ);

    enum qrcodegen_Ecc errCorLvl = qrcodegen_Ecc_LOW;  // Error correction level

    uint8_t qrcode[qrcodegen_BUFFER_LEN_MAX];
    uint8_t tempBuffer[qrcodegen_BUFFER_LEN_MAX];
    memcpy(tempBuffer, bufinfo.buf, bufinfo.len);
    bool ok = false;
    bool is_bytes = false;
    for (int i = 0; i < bufinfo.len; i++) {
        if (tempBuffer[i] == 0) { // \x00 will cut the byte array
            is_bytes = true;
        }
    }
    if (is_bytes) {
        ok = qrcodegen_encodeBinary(tempBuffer, bufinfo.len, qrcode, errCorLvl,
                                     qrcodegen_VERSION_MIN, qrcodegen_VERSION_MAX, qrcodegen_Mask_AUTO, true);
    } else {
        ok = qrcodegen_encodeText(bufinfo.buf, tempBuffer, qrcode, errCorLvl,
                                  qrcodegen_VERSION_MIN, qrcodegen_VERSION_MAX, qrcodegen_Mask_AUTO, true);
    }
    if (!ok) {
        mp_raise_ValueError("Failed to encode");
    }
    int size = qrcodegen_getSize(qrcode);

    size_t bufsize = (size * size + 7) / 8;
    vstr_t vstr;
    vstr_init_len(&vstr, bufsize);
    for (size_t i = 0; i < bufsize - 1; ++i) {
        vstr.buf[i] = qrcode[(i + 1)];
    }
    return mp_obj_new_str_from_vstr(&mp_type_bytes, &vstr);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(qrcode_encode_obj, qrcode_encode);


STATIC mp_obj_t qrcode_encode_to_string(mp_obj_t text_obj){
    mp_buffer_info_t bufinfo;
    mp_get_buffer_raise(text_obj, &bufinfo, MP_BUFFER_READ);

    enum qrcodegen_Ecc errCorLvl = qrcodegen_Ecc_LOW;  // Error correction level

    uint8_t qrcode[qrcodegen_BUFFER_LEN_MAX];
    uint8_t tempBuffer[qrcodegen_BUFFER_LEN_MAX];
    memcpy(tempBuffer, bufinfo.buf, bufinfo.len);
    bool ok = false;
    bool is_bytes = false;
    for(int i = 0; i < bufinfo.len; i++){
        if(tempBuffer[i] == 0){ // \x00 will cut the byte array
            is_bytes = true;
        }
    }
    if(is_bytes){
        ok = qrcodegen_encodeBinary(tempBuffer, bufinfo.len, qrcode, errCorLvl,
            qrcodegen_VERSION_MIN, qrcodegen_VERSION_MAX, qrcodegen_Mask_AUTO, true);
    }else{
        ok = qrcodegen_encodeText(bufinfo.buf, tempBuffer, qrcode, errCorLvl,
            qrcodegen_VERSION_MIN, qrcodegen_VERSION_MAX, qrcodegen_Mask_AUTO, true);
    }
    if(!ok){
        mp_raise_ValueError("Failed to encode");
    }
    int size = qrcodegen_getSize(qrcode);
    size_t bufsize = (size+1)*size;
    vstr_t vstr;
    vstr_init_len(&vstr, bufsize);
    memset((byte*)vstr.buf, '0', bufsize);
    for(int y=0; y<size; y++){
        for(int x=0; x<size; x++){
            if(qrcodegen_getModule(qrcode, x, y)){
                vstr.buf[y*(size+1)+x]='1';
            }
        }
        vstr.buf[y*(size+1)+size]='\n';
    }
    return mp_obj_new_str_from_vstr(&mp_type_str, &vstr);
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(qrcode_encode_to_string_obj, qrcode_encode_to_string);
/****************************** MODULE ******************************/

STATIC const mp_rom_map_elem_t qrcode_module_globals_table[] = {
    { MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_qrcode) },
    { MP_ROM_QSTR(MP_QSTR_encode), MP_ROM_PTR(&qrcode_encode_obj) },
    { MP_ROM_QSTR(MP_QSTR_encode_to_string), MP_ROM_PTR(&qrcode_encode_to_string_obj) },
};
STATIC MP_DEFINE_CONST_DICT(qrcode_module_globals, qrcode_module_globals_table);

// Define module object.
const mp_obj_module_t qrcode_user_cmodule = {
    .base = { &mp_type_module },
    .globals = (mp_obj_dict_t*)&qrcode_module_globals,
};

// Register the module to make it available in Python
MP_REGISTER_MODULE(MP_QSTR_qrcode, qrcode_user_cmodule, MODULE_QRCODE_ENABLED);