void test_newframe(void) { unsigned char buf[QRSPEC_WIDTH_MAX * QRSPEC_WIDTH_MAX]; int i, width; size_t len; FILE *fp; unsigned char *frame; QRcode *qrcode; unsigned int version; testStart("Checking newly created frame."); fp = fopen("frame", "rb"); if(fp == NULL) { perror("Failed to open \"frame\":"); abort(); } for(i=1; i<=QRSPEC_VERSION_MAX; i++) { frame = QRspec_newFrame(i); width = QRspec_getWidth(i); len = fread(buf, 1, width * width, fp); if((int)len != width * width) { perror("Failed to read the pattern file:"); abort(); } assert_zero(memcmp(frame, buf, len), "frame pattern mismatch (version %d)\n", i); qrcode = QRcode_new(i, width, frame); version = QRcode_decodeVersion(qrcode); assert_equal(version, i, "Decoded version number is wrong: %d, expected %d.\n", version, i); QRcode_free(qrcode); } testFinish(); fclose(fp); }
void test_format(void) { unsigned char *frame, *masked; int version, mask, width, dmask; QRecLevel level, dlevel; QRcode *code; int ret; testStart("Checking format info."); for(version=1; version<=QRSPEC_VERSION_MAX; version++) { frame = QRspec_newFrame(version); width = QRspec_getWidth(version); for(level=0; level<4; level++) { for(mask=0; mask<8; mask++) { masked = Mask_makeMask(width, frame, mask, level); code = QRcode_new(version, width, masked); ret = QRcode_decodeFormat(code, &dlevel, &dmask); assert_zero(ret, "Something wrong in format info.\n"); assert_equal(dlevel, level, "Decoded level is wrong: %d, expected %d", dlevel, level); assert_equal(dmask, mask, "Decoded mask is wrong: %d, expected %d", dlevel, level); QRcode_free(code); } } free(frame); } testFinish(); }
__STATIC QRcode *QRcode_encodeMask(QRinput *input, int mask) { int width, version; QRRawCode *raw; unsigned char *frame, *masked, *p, code, bit; FrameFiller *filler; int i, j; QRcode *qrcode = NULL; if(input->mqr) { errno = EINVAL; return NULL; } if(input->version < 0 || input->version > QRSPEC_VERSION_MAX) { errno = EINVAL; return NULL; } if(input->level > QR_ECLEVEL_H) { errno = EINVAL; return NULL; } raw = QRraw_new(input); if(raw == NULL) return NULL; version = raw->version; width = QRspec_getWidth(version); frame = QRspec_newFrame(version); if(frame == NULL) { QRraw_free(raw); return NULL; } filler = FrameFiller_new(width, frame, 0); if(filler == NULL) { QRraw_free(raw); free(frame); return NULL; } /* inteleaved data and ecc codes */ for(i=0; i<raw->dataLength + raw->eccLength; i++) { code = QRraw_getCode(raw); bit = 0x80; for(j=0; j<8; j++) { p = FrameFiller_next(filler); if(p == NULL) goto EXIT; *p = 0x02 | ((bit & code) != 0); bit = bit >> 1; } } QRraw_free(raw); raw = NULL; /* remainder bits */ j = QRspec_getRemainder(version); for(i=0; i<j; i++) { p = FrameFiller_next(filler); if(p == NULL) goto EXIT; *p = 0x02; } /* masking */ if(mask == -2) { // just for debug purpose masked = (unsigned char *)malloc(width * width); memcpy(masked, frame, width * width); } else if(mask < 0) { masked = Mask_mask(width, frame, input->level); } else { masked = Mask_makeMask(width, frame, mask, input->level); } if(masked == NULL) { goto EXIT; } qrcode = QRcode_new(version, width, masked); EXIT: QRraw_free(raw); free(filler); free(frame); return qrcode; }
STATIC_IN_RELEASE QRcode *QRcode_encodeMaskMQR(QRinput *input, int mask) { int width, version; MQRRawCode *raw; unsigned char *frame, *masked, *p, code, bit; int i, j, length; QRcode *qrcode = NULL; FrameFiller filler; if(!input->mqr) { errno = EINVAL; return NULL; } if(input->version <= 0 || input->version > MQRSPEC_VERSION_MAX) { errno = EINVAL; return NULL; } if(input->level > QR_ECLEVEL_Q) { errno = EINVAL; return NULL; } raw = MQRraw_new(input); if(raw == NULL) return NULL; version = raw->version; width = MQRspec_getWidth(version); frame = MQRspec_newFrame(version); if(frame == NULL) { MQRraw_free(raw); return NULL; } FrameFiller_set(&filler, width, frame, 1); /* interleaved data and ecc codes */ for(i = 0; i < raw->dataLength; i++) { code = MQRraw_getCode(raw); bit = 0x80; if(raw->oddbits && i == raw->dataLength - 1) { length = raw->oddbits; } else { length = 8; } for(j = 0; j < length; j++) { p = FrameFiller_next(&filler); if(p == NULL) goto EXIT; *p = ((bit & code) != 0); bit = bit >> 1; } } for(i = 0; i < raw->eccLength; i++) { code = MQRraw_getCode(raw); bit = 0x80; length = 8; for(j = 0; j < length; j++) { p = FrameFiller_next(&filler); if(p == NULL) goto EXIT; *p = 0x02 | ((bit & code) != 0); bit = bit >> 1; } } MQRraw_free(raw); raw = NULL; /* masking */ if(mask == -2) { // just for debug purpose masked = (unsigned char *)malloc(width * width); memcpy(masked, frame, width * width); } else if(mask < 0) { masked = MMask_mask(version, frame, input->level); } else { masked = MMask_makeMask(version, frame, mask, input->level); } if(masked == NULL) { goto EXIT; } qrcode = QRcode_new(version, width, masked); if(qrcode == NULL) { free(masked); } EXIT: MQRraw_free(raw); free(frame); return qrcode; }