• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * QR Code generator test worker (C)
3  *
4  * This program reads data and encoding parameters from standard input and writes
5  * QR Code bitmaps to standard output. The I/O format is one integer per line.
6  * Run with no command line arguments. The program is intended for automated
7  * batch testing of end-to-end functionality of this QR Code generator library.
8  *
9  * Copyright (c) Project Nayuki. (MIT License)
10  * https://www.nayuki.io/page/qr-code-generator-library
11  *
12  * Permission is hereby granted, free of charge, to any person obtaining a copy of
13  * this software and associated documentation files (the "Software"), to deal in
14  * the Software without restriction, including without limitation the rights to
15  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
16  * the Software, and to permit persons to whom the Software is furnished to do so,
17  * subject to the following conditions:
18  * - The above copyright notice and this permission notice shall be included in
19  *   all copies or substantial portions of the Software.
20  * - The Software is provided "as is", without warranty of any kind, express or
21  *   implied, including but not limited to the warranties of merchantability,
22  *   fitness for a particular purpose and noninfringement. In no event shall the
23  *   authors or copyright holders be liable for any claim, damages or other
24  *   liability, whether in an action of contract, tort or otherwise, arising from,
25  *   out of or in connection with the Software or the use or other dealings in the
26  *   Software.
27  */
28 
29 #include <stdbool.h>
30 #include <stddef.h>
31 #include <stdint.h>
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <string.h>
35 #include "qrcodegen.h"
36 
37 
main(void)38 int main(void) {
39 	while (true) {
40 
41 		// Read data length or exit
42 		size_t length;
43 		{
44 			int temp;
45 			if (scanf("%d", &temp) != 1)
46 				return EXIT_FAILURE;
47 			if (temp == -1)
48 				break;
49 			length = (size_t)temp;
50 		}
51 
52 		// Read data bytes
53 		bool isAscii = true;
54 		uint8_t *data = malloc(length * sizeof(uint8_t));
55 		if (data == NULL) {
56 			perror("malloc");
57 			return EXIT_FAILURE;
58 		}
59 		for (size_t i = 0; i < length; i++) {
60 			int b;
61 			if (scanf("%d", &b) != 1)
62 				return EXIT_FAILURE;
63 			data[i] = (uint8_t)b;
64 			isAscii &= 0 < b && b < 128;
65 		}
66 
67 		// Read encoding parameters
68 		int errCorLvl, minVersion, maxVersion, mask, boostEcl;
69 		if (scanf("%d %d %d %d %d", &errCorLvl, &minVersion, &maxVersion, &mask, &boostEcl) != 5)
70 			return EXIT_FAILURE;
71 
72 		// Allocate memory for QR Code
73 		size_t bufferLen = (size_t)qrcodegen_BUFFER_LEN_FOR_VERSION(maxVersion);
74 		uint8_t *qrcode     = malloc(bufferLen * sizeof(uint8_t));
75 		uint8_t *tempBuffer = malloc(bufferLen * sizeof(uint8_t));
76 		if (qrcode == NULL || tempBuffer == NULL) {
77 			perror("malloc");
78 			return EXIT_FAILURE;
79 		}
80 
81 		// Try to make QR Code symbol
82 		bool ok;
83 		if (isAscii) {
84 			char *text = malloc((length + 1) * sizeof(char));
85 			if (text == NULL) {
86 				perror("malloc");
87 				return EXIT_FAILURE;
88 			}
89 			for (size_t i = 0; i < length; i++)
90 				text[i] = (char)data[i];
91 			text[length] = '\0';
92 			ok = qrcodegen_encodeText(text, tempBuffer, qrcode, (enum qrcodegen_Ecc)errCorLvl,
93 				minVersion, maxVersion, (enum qrcodegen_Mask)mask, boostEcl == 1);
94 			free(text);
95 		} else if (length <= bufferLen) {
96 			memcpy(tempBuffer, data, length * sizeof(data[0]));
97 			ok = qrcodegen_encodeBinary(tempBuffer, length, qrcode, (enum qrcodegen_Ecc)errCorLvl,
98 				minVersion, maxVersion, (enum qrcodegen_Mask)mask, boostEcl == 1);
99 		} else
100 			ok = false;
101 		free(data);
102 		free(tempBuffer);
103 
104 		if (ok) {
105 			// Print grid of modules
106 			int size = qrcodegen_getSize(qrcode);
107 			printf("%d\n", (size - 17) / 4);
108 			for (int y = 0; y < size; y++) {
109 				for (int x = 0; x < size; x++)
110 					printf("%d\n", qrcodegen_getModule(qrcode, x, y) ? 1 : 0);
111 			}
112 		} else
113 			printf("-1\n");
114 		free(qrcode);
115 		fflush(stdout);
116 	}
117 	return EXIT_SUCCESS;
118 }
119