12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628 |
- #if HAVE_CONFIG_H
- # include "config.h"
- #endif
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <errno.h>
- #include "qrencode.h"
- #include "qrspec.h"
- #include "mqrspec.h"
- #include "bitstream.h"
- #include "qrinput.h"
- int QRinput_isSplittableMode(QRencodeMode mode)
- {
- return (mode >= QR_MODE_NUM && mode <= QR_MODE_KANJI);
- }
- static QRinput_List *QRinput_List_newEntry(QRencodeMode mode, int size, const unsigned char *data)
- {
- QRinput_List *entry;
- if(QRinput_check(mode, size, data)) {
- errno = EINVAL;
- return NULL;
- }
- entry = (QRinput_List *)malloc(sizeof(QRinput_List));
- if(entry == NULL) return NULL;
- entry->mode = mode;
- entry->size = size;
- entry->data = NULL;
- if(size > 0) {
- entry->data = (unsigned char *)malloc((size_t)size);
- if(entry->data == NULL) {
- free(entry);
- return NULL;
- }
- memcpy(entry->data, data, (size_t)size);
- }
- entry->bstream = NULL;
- entry->next = NULL;
- return entry;
- }
- static void QRinput_List_freeEntry(QRinput_List *entry)
- {
- if(entry != NULL) {
- free(entry->data);
- BitStream_free(entry->bstream);
- free(entry);
- }
- }
- static QRinput_List *QRinput_List_dup(QRinput_List *entry)
- {
- QRinput_List *n;
- n = (QRinput_List *)malloc(sizeof(QRinput_List));
- if(n == NULL) return NULL;
- n->mode = entry->mode;
- n->size = entry->size;
- n->data = (unsigned char *)malloc((size_t)n->size);
- if(n->data == NULL) {
- free(n);
- return NULL;
- }
- memcpy(n->data, entry->data, (size_t)entry->size);
- n->bstream = NULL;
- n->next = NULL;
- return n;
- }
- QRinput *QRinput_new(void)
- {
- return QRinput_new2(0, QR_ECLEVEL_L);
- }
- QRinput *QRinput_new2(int version, QRecLevel level)
- {
- QRinput *input;
- if(version < 0 || version > QRSPEC_VERSION_MAX || level > QR_ECLEVEL_H) {
- errno = EINVAL;
- return NULL;
- }
- input = (QRinput *)malloc(sizeof(QRinput));
- if(input == NULL) return NULL;
- input->head = NULL;
- input->tail = NULL;
- input->version = version;
- input->level = level;
- input->mqr = 0;
- input->fnc1 = 0;
- return input;
- }
- QRinput *QRinput_newMQR(int version, QRecLevel level)
- {
- QRinput *input;
- if(version <= 0 || version > MQRSPEC_VERSION_MAX) goto INVALID;
- if((MQRspec_getECCLength(version, level) == 0)) goto INVALID;
- input = QRinput_new2(version, level);
- if(input == NULL) return NULL;
- input->mqr = 1;
- return input;
- INVALID:
- errno = EINVAL;
- return NULL;
- }
- int QRinput_getVersion(QRinput *input)
- {
- return input->version;
- }
- int QRinput_setVersion(QRinput *input, int version)
- {
- if(input->mqr || version < 0 || version > QRSPEC_VERSION_MAX) {
- errno = EINVAL;
- return -1;
- }
- input->version = version;
- return 0;
- }
- QRecLevel QRinput_getErrorCorrectionLevel(QRinput *input)
- {
- return input->level;
- }
- int QRinput_setErrorCorrectionLevel(QRinput *input, QRecLevel level)
- {
- if(input->mqr || level > QR_ECLEVEL_H) {
- errno = EINVAL;
- return -1;
- }
- input->level = level;
- return 0;
- }
- int QRinput_setVersionAndErrorCorrectionLevel(QRinput *input, int version, QRecLevel level)
- {
- if(input->mqr) {
- if(version <= 0 || version > MQRSPEC_VERSION_MAX) goto INVALID;
- if((MQRspec_getECCLength(version, level) == 0)) goto INVALID;
- } else {
- if(version < 0 || version > QRSPEC_VERSION_MAX) goto INVALID;
- if(level > QR_ECLEVEL_H) goto INVALID;
- }
- input->version = version;
- input->level = level;
- return 0;
- INVALID:
- errno = EINVAL;
- return -1;
- }
- static void QRinput_appendEntry(QRinput *input, QRinput_List *entry)
- {
- if(input->tail == NULL) {
- input->head = entry;
- input->tail = entry;
- } else {
- input->tail->next = entry;
- input->tail = entry;
- }
- entry->next = NULL;
- }
- int QRinput_append(QRinput *input, QRencodeMode mode, int size, const unsigned char *data)
- {
- QRinput_List *entry;
- entry = QRinput_List_newEntry(mode, size, data);
- if(entry == NULL) {
- return -1;
- }
- QRinput_appendEntry(input, entry);
- return 0;
- }
- STATIC_IN_RELEASE int QRinput_insertStructuredAppendHeader(QRinput *input, int size, int number, unsigned char parity)
- {
- QRinput_List *entry;
- unsigned char buf[3];
- if(size > MAX_STRUCTURED_SYMBOLS) {
- errno = EINVAL;
- return -1;
- }
- if(number <= 0 || number > size) {
- errno = EINVAL;
- return -1;
- }
- buf[0] = (unsigned char)size;
- buf[1] = (unsigned char)number;
- buf[2] = parity;
- entry = QRinput_List_newEntry(QR_MODE_STRUCTURE, 3, buf);
- if(entry == NULL) {
- return -1;
- }
- entry->next = input->head;
- input->head = entry;
- return 0;
- }
- int QRinput_appendECIheader(QRinput *input, unsigned int ecinum)
- {
- unsigned char data[4];
- if(ecinum > 999999) {
- errno = EINVAL;
- return -1;
- }
-
- data[0] = ecinum & 0xff;
- data[1] = (ecinum >> 8) & 0xff;
- data[2] = (ecinum >> 16) & 0xff;
- data[3] = (ecinum >> 24) & 0xff;
- return QRinput_append(input, QR_MODE_ECI, 4, data);
- }
- void QRinput_free(QRinput *input)
- {
- QRinput_List *list, *next;
- if(input != NULL) {
- list = input->head;
- while(list != NULL) {
- next = list->next;
- QRinput_List_freeEntry(list);
- list = next;
- }
- free(input);
- }
- }
- static unsigned char QRinput_calcParity(QRinput *input)
- {
- unsigned char parity = 0;
- QRinput_List *list;
- int i;
- list = input->head;
- while(list != NULL) {
- if(list->mode != QR_MODE_STRUCTURE) {
- for(i = list->size-1; i >= 0; i--) {
- parity ^= list->data[i];
- }
- }
- list = list->next;
- }
- return parity;
- }
- QRinput *QRinput_dup(QRinput *input)
- {
- QRinput *n;
- QRinput_List *list, *e;
- if(input->mqr) {
- n = QRinput_newMQR(input->version, input->level);
- } else {
- n = QRinput_new2(input->version, input->level);
- }
- if(n == NULL) return NULL;
- list = input->head;
- while(list != NULL) {
- e = QRinput_List_dup(list);
- if(e == NULL) {
- QRinput_free(n);
- return NULL;
- }
- QRinput_appendEntry(n, e);
- list = list->next;
- }
- return n;
- }
- static int QRinput_checkModeNum(int size, const char *data)
- {
- int i;
- for(i = 0; i < size; i++) {
- if(data[i] < '0' || data[i] > '9')
- return -1;
- }
- return 0;
- }
- int QRinput_estimateBitsModeNum(int size)
- {
- int w;
- int bits;
- w = size / 3;
- bits = w * 10;
- switch(size - w * 3) {
- case 1:
- bits += 4;
- break;
- case 2:
- bits += 7;
- break;
- default:
- break;
- }
- return bits;
- }
- static int QRinput_encodeModeNum(QRinput_List *entry, BitStream *bstream, int version, int mqr)
- {
- int words, i, ret;
- unsigned int val;
- if(mqr) {
- if(version > 1) {
- ret = BitStream_appendNum(bstream, (size_t)(version - 1), MQRSPEC_MODEID_NUM);
- if(ret < 0) return -1;
- }
- ret = BitStream_appendNum(bstream, (size_t)MQRspec_lengthIndicator(QR_MODE_NUM, version), (unsigned int)entry->size);
- if(ret < 0) return -1;
- } else {
- ret = BitStream_appendNum(bstream, 4, QRSPEC_MODEID_NUM);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, (size_t)QRspec_lengthIndicator(QR_MODE_NUM, version), (unsigned int)entry->size);
- if(ret < 0) return -1;
- }
- words = entry->size / 3;
- for(i = 0; i < words; i++) {
- val = (unsigned int)(entry->data[i*3 ] - '0') * 100;
- val += (unsigned int)(entry->data[i*3+1] - '0') * 10;
- val += (unsigned int)(entry->data[i*3+2] - '0');
- ret = BitStream_appendNum(bstream, 10, val);
- if(ret < 0) return -1;
- }
- if(entry->size - words * 3 == 1) {
- val = (unsigned int)(entry->data[words*3] - '0');
- ret = BitStream_appendNum(bstream, 4, val);
- if(ret < 0) return -1;
- } else if(entry->size - words * 3 == 2) {
- val = (unsigned int)(entry->data[words*3 ] - '0') * 10;
- val += (unsigned int)(entry->data[words*3+1] - '0');
- BitStream_appendNum(bstream, 7, val);
- if(ret < 0) return -1;
- }
- return 0;
- }
- const signed char QRinput_anTable[128] = {
- -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
- -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
- 36, -1, -1, -1, 37, 38, -1, -1, -1, -1, 39, 40, -1, 41, 42, 43,
- 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 44, -1, -1, -1, -1, -1,
- -1, 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, -1, -1, -1, -1, -1,
- -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
- -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1
- };
- static int QRinput_checkModeAn(int size, const char *data)
- {
- int i;
- for(i = 0; i < size; i++) {
- if(QRinput_lookAnTable(data[i]) < 0)
- return -1;
- }
- return 0;
- }
- int QRinput_estimateBitsModeAn(int size)
- {
- int w;
- int bits;
- w = size / 2;
- bits = w * 11;
- if(size & 1) {
- bits += 6;
- }
- return bits;
- }
- static int QRinput_encodeModeAn(QRinput_List *entry, BitStream *bstream, int version, int mqr)
- {
- int words, i, ret;
- unsigned int val;
- if(mqr) {
- if(version < 2) {
- errno = EINVAL;
- return -1;
- }
- ret = BitStream_appendNum(bstream, (size_t)(version - 1), MQRSPEC_MODEID_AN);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, (size_t)MQRspec_lengthIndicator(QR_MODE_AN, version), (unsigned int)entry->size);
- if(ret < 0) return -1;
- } else {
- ret = BitStream_appendNum(bstream, 4, QRSPEC_MODEID_AN);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, (size_t)QRspec_lengthIndicator(QR_MODE_AN, version), (unsigned int)entry->size);
- if(ret < 0) return -1;
- }
- words = entry->size / 2;
- for(i = 0; i < words; i++) {
- val = (unsigned int)QRinput_lookAnTable(entry->data[i*2 ]) * 45;
- val += (unsigned int)QRinput_lookAnTable(entry->data[i*2+1]);
- ret = BitStream_appendNum(bstream, 11, val);
- if(ret < 0) return -1;
- }
- if(entry->size & 1) {
- val = (unsigned int)QRinput_lookAnTable(entry->data[words * 2]);
- ret = BitStream_appendNum(bstream, 6, val);
- if(ret < 0) return -1;
- }
- return 0;
- }
- int QRinput_estimateBitsMode8(int size)
- {
- return size * 8;
- }
- static int QRinput_encodeMode8(QRinput_List *entry, BitStream *bstream, int version, int mqr)
- {
- int ret;
- if(mqr) {
- if(version < 3) {
- errno = EINVAL;
- return -1;
- }
- ret = BitStream_appendNum(bstream, (size_t)(version - 1), MQRSPEC_MODEID_8);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, (size_t)MQRspec_lengthIndicator(QR_MODE_8, version), (unsigned int)entry->size);
- if(ret < 0) return -1;
- } else {
- ret = BitStream_appendNum(bstream, 4, QRSPEC_MODEID_8);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, (size_t)QRspec_lengthIndicator(QR_MODE_8, version), (unsigned int)entry->size);
- if(ret < 0) return -1;
- }
- ret = BitStream_appendBytes(bstream, (size_t)entry->size, entry->data);
- if(ret < 0) return -1;
- return 0;
- }
- int QRinput_estimateBitsModeKanji(int size)
- {
- return (size / 2) * 13;
- }
- static int QRinput_checkModeKanji(int size, const unsigned char *data)
- {
- int i;
- unsigned int val;
- if(size & 1)
- return -1;
- for(i = 0; i < size; i+=2) {
- val = ((unsigned int)data[i] << 8) | data[i+1];
- if(val < 0x8140 || (val > 0x9ffc && val < 0xe040) || val > 0xebbf) {
- return -1;
- }
- }
- return 0;
- }
- static int QRinput_encodeModeKanji(QRinput_List *entry, BitStream *bstream, int version, int mqr)
- {
- int ret, i;
- unsigned int val, h;
- if(mqr) {
- if(version < 2) {
- errno = EINVAL;
- return -1;
- }
- ret = BitStream_appendNum(bstream, (size_t)(version - 1), MQRSPEC_MODEID_KANJI);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, (size_t)MQRspec_lengthIndicator(QR_MODE_KANJI, version), (unsigned int)entry->size/2);
- if(ret < 0) return -1;
- } else {
- ret = BitStream_appendNum(bstream, 4, QRSPEC_MODEID_KANJI);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, (size_t)QRspec_lengthIndicator(QR_MODE_KANJI, version), (unsigned int)entry->size/2);
- if(ret < 0) return -1;
- }
- for(i = 0; i < entry->size; i+=2) {
- val = ((unsigned int)entry->data[i] << 8) | entry->data[i+1];
- if(val <= 0x9ffc) {
- val -= 0x8140;
- } else {
- val -= 0xc140;
- }
- h = (val >> 8) * 0xc0;
- val = (val & 0xff) + h;
- ret = BitStream_appendNum(bstream, 13, val);
- if(ret < 0) return -1;
- }
- return 0;
- }
- static int QRinput_encodeModeStructure(QRinput_List *entry, BitStream *bstream, int mqr)
- {
- int ret;
- if(mqr) {
- errno = EINVAL;
- return -1;
- }
- ret = BitStream_appendNum(bstream, 4, QRSPEC_MODEID_STRUCTURE);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, 4, entry->data[1] - 1U);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, 4, entry->data[0] - 1U);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, 8, entry->data[2]);
- if(ret < 0) return -1;
- return 0;
- }
- static int QRinput_checkModeFNC1Second(int size)
- {
- if(size != 1) return -1;
-
- return 0;
- }
- static int QRinput_encodeModeFNC1Second(QRinput_List *entry, BitStream *bstream)
- {
- int ret;
- ret = BitStream_appendNum(bstream, 4, QRSPEC_MODEID_FNC1SECOND);
- if(ret < 0) return -1;
- ret = BitStream_appendBytes(bstream, 1, entry->data);
- if(ret < 0) return -1;
- return 0;
- }
- static unsigned int QRinput_decodeECIfromByteArray(unsigned char *data)
- {
- int i;
- unsigned int ecinum;
- ecinum = 0;
- for(i = 0; i < 4; i++) {
- ecinum = ecinum << 8;
- ecinum |= data[3-i];
- }
- return ecinum;
- }
- static int QRinput_estimateBitsModeECI(unsigned char *data)
- {
- unsigned int ecinum;
- ecinum = QRinput_decodeECIfromByteArray(data);
-
- if(ecinum < 128) {
- return MODE_INDICATOR_SIZE + 8;
- } else if(ecinum < 16384) {
- return MODE_INDICATOR_SIZE + 16;
- } else {
- return MODE_INDICATOR_SIZE + 24;
- }
- }
- static int QRinput_encodeModeECI(QRinput_List *entry, BitStream *bstream)
- {
- int ret, words;
- unsigned int ecinum, code;
- ecinum = QRinput_decodeECIfromByteArray(entry->data);
-
- if(ecinum < 128) {
- words = 1;
- code = ecinum;
- } else if(ecinum < 16384) {
- words = 2;
- code = 0x8000 + ecinum;
- } else {
- words = 3;
- code = 0xc0000 + ecinum;
- }
- ret = BitStream_appendNum(bstream, 4, QRSPEC_MODEID_ECI);
- if(ret < 0) return -1;
- ret = BitStream_appendNum(bstream, (size_t)words * 8, code);
- if(ret < 0) return -1;
- return 0;
- }
- int QRinput_check(QRencodeMode mode, int size, const unsigned char *data)
- {
- if((mode == QR_MODE_FNC1FIRST && size < 0) || size <= 0) return -1;
- switch(mode) {
- case QR_MODE_NUM:
- return QRinput_checkModeNum(size, (const char *)data);
- case QR_MODE_AN:
- return QRinput_checkModeAn(size, (const char *)data);
- case QR_MODE_KANJI:
- return QRinput_checkModeKanji(size, data);
- case QR_MODE_8:
- return 0;
- case QR_MODE_STRUCTURE:
- return 0;
- case QR_MODE_ECI:
- return 0;
- case QR_MODE_FNC1FIRST:
- return 0;
- case QR_MODE_FNC1SECOND:
- return QRinput_checkModeFNC1Second(size);
- case QR_MODE_NUL:
- break;
- }
- return -1;
- }
- static int QRinput_estimateBitStreamSizeOfEntry(QRinput_List *entry, int version, int mqr)
- {
- int bits = 0;
- int l, m;
- int num;
- if(version == 0) version = 1;
- switch(entry->mode) {
- case QR_MODE_NUM:
- bits = QRinput_estimateBitsModeNum(entry->size);
- break;
- case QR_MODE_AN:
- bits = QRinput_estimateBitsModeAn(entry->size);
- break;
- case QR_MODE_8:
- bits = QRinput_estimateBitsMode8(entry->size);
- break;
- case QR_MODE_KANJI:
- bits = QRinput_estimateBitsModeKanji(entry->size);
- break;
- case QR_MODE_STRUCTURE:
- return STRUCTURE_HEADER_SIZE;
- case QR_MODE_ECI:
- bits = QRinput_estimateBitsModeECI(entry->data);
- break;
- case QR_MODE_FNC1FIRST:
- return MODE_INDICATOR_SIZE;
- case QR_MODE_FNC1SECOND:
- return MODE_INDICATOR_SIZE + 8;
- default:
- return 0;
- }
- if(mqr) {
- l = QRspec_lengthIndicator(entry->mode, version);
- m = version - 1;
- bits += l + m;
- } else {
- l = QRspec_lengthIndicator(entry->mode, version);
- m = 1 << l;
- num = (entry->size + m - 1) / m;
- bits += num * (MODE_INDICATOR_SIZE + l);
- }
- return bits;
- }
- STATIC_IN_RELEASE int QRinput_estimateBitStreamSize(QRinput *input, int version)
- {
- QRinput_List *list;
- int bits = 0;
- list = input->head;
- while(list != NULL) {
- bits += QRinput_estimateBitStreamSizeOfEntry(list, version, input->mqr);
- list = list->next;
- }
- return bits;
- }
- static int QRinput_estimateVersion(QRinput *input)
- {
- int bits;
- int version, prev;
- version = 0;
- do {
- prev = version;
- bits = QRinput_estimateBitStreamSize(input, prev);
- version = QRspec_getMinimumVersion((bits + 7) / 8, input->level);
- } while (version > prev);
- return version;
- }
- STATIC_IN_RELEASE int QRinput_lengthOfCode(QRencodeMode mode, int version, int bits)
- {
- int payload, size, chunks, remain, maxsize;
- payload = bits - 4 - QRspec_lengthIndicator(mode, version);
- switch(mode) {
- case QR_MODE_NUM:
- chunks = payload / 10;
- remain = payload - chunks * 10;
- size = chunks * 3;
- if(remain >= 7) {
- size += 2;
- } else if(remain >= 4) {
- size += 1;
- }
- break;
- case QR_MODE_AN:
- chunks = payload / 11;
- remain = payload - chunks * 11;
- size = chunks * 2;
- if(remain >= 6) size++;
- break;
- case QR_MODE_8:
- size = payload / 8;
- break;
- case QR_MODE_KANJI:
- size = (payload / 13) * 2;
- break;
- case QR_MODE_STRUCTURE:
- size = payload / 8;
- break;
- default:
- size = 0;
- break;
- }
- maxsize = QRspec_maximumWords(mode, version);
- if(size < 0) size = 0;
- if(maxsize > 0 && size > maxsize) size = maxsize;
- return size;
- }
- static int QRinput_encodeBitStream(QRinput_List *entry, BitStream *bstream, int version, int mqr)
- {
- int words, ret;
- QRinput_List *st1 = NULL, *st2 = NULL;
- int prevsize;
- prevsize = (int)BitStream_size(bstream);
- words = QRspec_maximumWords(entry->mode, version);
- if(words != 0 && entry->size > words) {
- st1 = QRinput_List_newEntry(entry->mode, words, entry->data);
- if(st1 == NULL) goto ABORT;
- st2 = QRinput_List_newEntry(entry->mode, entry->size - words, &entry->data[words]);
- if(st2 == NULL) goto ABORT;
- ret = QRinput_encodeBitStream(st1, bstream, version, mqr);
- if(ret < 0) goto ABORT;
- ret = QRinput_encodeBitStream(st2, bstream, version, mqr);
- if(ret < 0) goto ABORT;
- QRinput_List_freeEntry(st1);
- QRinput_List_freeEntry(st2);
- } else {
- ret = 0;
- switch(entry->mode) {
- case QR_MODE_NUM:
- ret = QRinput_encodeModeNum(entry, bstream, version, mqr);
- break;
- case QR_MODE_AN:
- ret = QRinput_encodeModeAn(entry, bstream, version, mqr);
- break;
- case QR_MODE_8:
- ret = QRinput_encodeMode8(entry, bstream, version, mqr);
- break;
- case QR_MODE_KANJI:
- ret = QRinput_encodeModeKanji(entry, bstream, version, mqr);
- break;
- case QR_MODE_STRUCTURE:
- ret = QRinput_encodeModeStructure(entry, bstream, mqr);
- break;
- case QR_MODE_ECI:
- ret = QRinput_encodeModeECI(entry, bstream);
- break;
- case QR_MODE_FNC1SECOND:
- ret = QRinput_encodeModeFNC1Second(entry, bstream);
- break;
- default:
- break;
- }
- if(ret < 0) return -1;
- }
- return (int)BitStream_size(bstream) - prevsize;
- ABORT:
- QRinput_List_freeEntry(st1);
- QRinput_List_freeEntry(st2);
- return -1;
- }
- static int QRinput_createBitStream(QRinput *input, BitStream *bstream)
- {
- QRinput_List *list;
- int bits, total = 0;
- list = input->head;
- while(list != NULL) {
- bits = QRinput_encodeBitStream(list, bstream, input->version, input->mqr);
- if(bits < 0) return -1;
- total += bits;
- list = list->next;
- }
- return total;
- }
- static int QRinput_convertData(QRinput *input, BitStream *bstream)
- {
- int bits;
- int ver;
- ver = QRinput_estimateVersion(input);
- if(ver > QRinput_getVersion(input)) {
- QRinput_setVersion(input, ver);
- }
- for(;;) {
- BitStream_reset(bstream);
- bits = QRinput_createBitStream(input, bstream);
- if(bits < 0) return -1;
- ver = QRspec_getMinimumVersion((bits + 7) / 8, input->level);
- if(ver > QRinput_getVersion(input)) {
- QRinput_setVersion(input, ver);
- } else {
- break;
- }
- }
- return 0;
- }
- static int QRinput_appendPaddingBit(BitStream *bstream, QRinput *input)
- {
- int bits, maxbits, words, maxwords, i, ret;
- int padlen;
- bits = (int)BitStream_size(bstream);
- maxwords = QRspec_getDataLength(input->version, input->level);
- maxbits = maxwords * 8;
- if(maxbits < bits) {
- errno = ERANGE;
- return -1;
- }
- if(maxbits == bits) {
- return 0;
- }
- if(maxbits - bits <= 4) {
- return (int)BitStream_appendNum(bstream, (size_t)(maxbits - bits), 0);
- }
- words = (bits + 4 + 7) / 8;
- ret = (int)BitStream_appendNum(bstream, (size_t)(words * 8 - bits), 0);
- if(ret < 0) return ret;
- padlen = maxwords - words;
- if(padlen > 0) {
- for(i = 0; i < padlen; i++) {
- ret = (int)BitStream_appendNum(bstream, 8, (i&1)?0x11:0xec);
- if(ret < 0) {
- return ret;
- }
- }
- }
- return 0;
- }
- static int QRinput_appendPaddingBitMQR(BitStream *bstream, QRinput *input)
- {
- int bits, maxbits, words, maxwords, i, ret, termbits;
- int padlen;
- bits = (int)BitStream_size(bstream);
- maxbits = MQRspec_getDataLengthBit(input->version, input->level);
- maxwords = maxbits / 8;
- if(maxbits < bits) {
- errno = ERANGE;
- return -1;
- }
- if(maxbits == bits) {
- return 0;
- }
- termbits = input->version * 2 + 1;
- if(maxbits - bits <= termbits) {
- return (int)BitStream_appendNum(bstream, (size_t)(maxbits - bits), 0);
- }
- bits += termbits;
- words = (bits + 7) / 8;
- if(maxbits - words * 8 > 0) {
- termbits += words * 8 - bits;
- if(words == maxwords) termbits += maxbits - words * 8;
- } else {
- termbits += words * 8 - bits;
- }
- ret = (int)BitStream_appendNum(bstream, (size_t)termbits, 0);
- if(ret < 0) return ret;
- padlen = maxwords - words;
- if(padlen > 0) {
- for(i = 0; i < padlen; i++) {
- ret = (int)BitStream_appendNum(bstream, 8, (i&1)?0x11:0xec);
- if(ret < 0) return ret;
- }
- termbits = maxbits - maxwords * 8;
- if(termbits > 0) {
- ret = (int)BitStream_appendNum(bstream, (size_t)termbits, 0);
- if(ret < 0) return ret;
- }
- }
- return 0;
- }
- static int QRinput_insertFNC1Header(QRinput *input)
- {
- QRinput_List *entry = NULL;
- if(input->fnc1 == 1) {
- entry = QRinput_List_newEntry(QR_MODE_FNC1FIRST, 0, NULL);
- } else if(input->fnc1 == 2) {
- entry = QRinput_List_newEntry(QR_MODE_FNC1SECOND, 1, &(input->appid));
- }
- if(entry == NULL) {
- return -1;
- }
- if(input->head->mode != QR_MODE_STRUCTURE && input->head->mode != QR_MODE_ECI) {
- entry->next = input->head;
- input->head = entry;
- } else {
- entry->next = input->head->next;
- input->head->next = entry;
- }
- return 0;
- }
- STATIC_IN_RELEASE int QRinput_mergeBitStream(QRinput *input, BitStream *bstream)
- {
- if(input->mqr) {
- if(QRinput_createBitStream(input, bstream) < 0) {
- return -1;
- }
- } else {
- if(input->fnc1) {
- if(QRinput_insertFNC1Header(input) < 0) {
- return -1;
- }
- }
- if(QRinput_convertData(input, bstream) < 0) {
- return -1;
- }
- }
- return 0;
- }
- STATIC_IN_RELEASE int QRinput_getBitStream(QRinput *input, BitStream *bstream)
- {
- int ret;
- ret = QRinput_mergeBitStream(input, bstream);
- if(ret < 0) return -1;
- if(input->mqr) {
- ret = QRinput_appendPaddingBitMQR(bstream, input);
- } else {
- ret = QRinput_appendPaddingBit(bstream, input);
- }
- if(ret < 0) return -1;
- return 0;
- }
- unsigned char *QRinput_getByteStream(QRinput *input)
- {
- BitStream *bstream;
- unsigned char *array;
- int ret;
- bstream = BitStream_new();
- if(bstream == NULL) {
- return NULL;
- }
- ret = QRinput_getBitStream(input, bstream);
- if(ret < 0) {
- BitStream_free(bstream);
- return NULL;
- }
- array = BitStream_toByte(bstream);
- BitStream_free(bstream);
- return array;
- }
- static QRinput_InputList *QRinput_InputList_newEntry(QRinput *input)
- {
- QRinput_InputList *entry;
- entry = (QRinput_InputList *)malloc(sizeof(QRinput_InputList));
- if(entry == NULL) return NULL;
- entry->input = input;
- entry->next = NULL;
- return entry;
- }
- static void QRinput_InputList_freeEntry(QRinput_InputList *entry)
- {
- if(entry != NULL) {
- QRinput_free(entry->input);
- free(entry);
- }
- }
- QRinput_Struct *QRinput_Struct_new(void)
- {
- QRinput_Struct *s;
- s = (QRinput_Struct *)malloc(sizeof(QRinput_Struct));
- if(s == NULL) return NULL;
- s->size = 0;
- s->parity = -1;
- s->head = NULL;
- s->tail = NULL;
- return s;
- }
- void QRinput_Struct_setParity(QRinput_Struct *s, unsigned char parity)
- {
- s->parity = (int)parity;
- }
- int QRinput_Struct_appendInput(QRinput_Struct *s, QRinput *input)
- {
- QRinput_InputList *e;
- if(input->mqr) {
- errno = EINVAL;
- return -1;
- }
- e = QRinput_InputList_newEntry(input);
- if(e == NULL) return -1;
- s->size++;
- if(s->tail == NULL) {
- s->head = e;
- s->tail = e;
- } else {
- s->tail->next = e;
- s->tail = e;
- }
- return s->size;
- }
- void QRinput_Struct_free(QRinput_Struct *s)
- {
- QRinput_InputList *list, *next;
- if(s != NULL) {
- list = s->head;
- while(list != NULL) {
- next = list->next;
- QRinput_InputList_freeEntry(list);
- list = next;
- }
- free(s);
- }
- }
- static unsigned char QRinput_Struct_calcParity(QRinput_Struct *s)
- {
- QRinput_InputList *list;
- unsigned char parity = 0;
- list = s->head;
- while(list != NULL) {
- parity ^= QRinput_calcParity(list->input);
- list = list->next;
- }
- QRinput_Struct_setParity(s, parity);
- return parity;
- }
- static int QRinput_List_shrinkEntry(QRinput_List *entry, int bytes)
- {
- unsigned char *data;
- data = (unsigned char *)malloc((size_t)bytes);
- if(data == NULL) return -1;
- memcpy(data, entry->data, (size_t)bytes);
- free(entry->data);
- entry->data = data;
- entry->size = bytes;
- return 0;
- }
- STATIC_IN_RELEASE int QRinput_splitEntry(QRinput_List *entry, int bytes)
- {
- QRinput_List *e;
- int ret;
- e = QRinput_List_newEntry(entry->mode, entry->size - bytes, entry->data + bytes);
- if(e == NULL) {
- return -1;
- }
- ret = QRinput_List_shrinkEntry(entry, bytes);
- if(ret < 0) {
- QRinput_List_freeEntry(e);
- return -1;
- }
- e->next = entry->next;
- entry->next = e;
- return 0;
- }
- QRinput_Struct *QRinput_splitQRinputToStruct(QRinput *input)
- {
- QRinput *p = NULL;
- QRinput_Struct *s = NULL;
- int bits, maxbits, nextbits, bytes, ret;
- QRinput_List *list, *next, *prev;
- BitStream *bstream = NULL;
- if(input->mqr) {
- errno = EINVAL;
- return NULL;
- }
- s = QRinput_Struct_new();
- if(s == NULL) return NULL;
- input = QRinput_dup(input);
- if(input == NULL) {
- QRinput_Struct_free(s);
- return NULL;
- }
- QRinput_Struct_setParity(s, QRinput_calcParity(input));
- maxbits = QRspec_getDataLength(input->version, input->level) * 8 - STRUCTURE_HEADER_SIZE;
- if(maxbits <= 0) goto ABORT;
- bstream = BitStream_new();
- if(bstream == NULL) goto ABORT;
- bits = 0;
- list = input->head;
- prev = NULL;
- while(list != NULL) {
- nextbits = QRinput_estimateBitStreamSizeOfEntry(list, input->version, input->mqr);
- if(bits + nextbits <= maxbits) {
- BitStream_reset(bstream);
- ret = QRinput_encodeBitStream(list, bstream, input->version, input->mqr);
- if(ret < 0) goto ABORT;
- bits += ret;
- prev = list;
- list = list->next;
- } else {
- bytes = QRinput_lengthOfCode(list->mode, input->version, maxbits - bits);
- p = QRinput_new2(input->version, input->level);
- if(p == NULL) goto ABORT;
- if(bytes > 0) {
-
- ret = QRinput_splitEntry(list, bytes);
- if(ret < 0) {
- QRinput_free(p);
- goto ABORT;
- }
-
- next = list->next;
- list->next = NULL;
-
- p->head = next;
-
- p->tail = input->tail;
- input->tail = list;
-
- prev = list;
- list = next;
- } else {
-
- prev->next = NULL;
- p->head = list;
- p->tail = input->tail;
- input->tail = prev;
- }
- ret = QRinput_Struct_appendInput(s, input);
- if(ret < 0) {
- QRinput_free(p);
- goto ABORT;
- }
- input = p;
- bits = 0;
- }
- }
- ret = QRinput_Struct_appendInput(s, input);
- if(ret < 0) goto ABORT;
- if(s->size > MAX_STRUCTURED_SYMBOLS) {
- errno = ERANGE;
- QRinput_Struct_free(s);
- BitStream_free(bstream);
- return NULL;
- }
- ret = QRinput_Struct_insertStructuredAppendHeaders(s);
- if(ret < 0) {
- QRinput_Struct_free(s);
- BitStream_free(bstream);
- return NULL;
- }
- BitStream_free(bstream);
- return s;
- ABORT:
- BitStream_free(bstream);
- QRinput_free(input);
- QRinput_Struct_free(s);
- return NULL;
- }
- int QRinput_Struct_insertStructuredAppendHeaders(QRinput_Struct *s)
- {
- int i;
- QRinput_InputList *list;
- if(s->size == 1) {
- return 0;
- }
- if(s->parity < 0) {
- QRinput_Struct_calcParity(s);
- }
- i = 1;
- list = s->head;
- while(list != NULL) {
- if(QRinput_insertStructuredAppendHeader(list->input, s->size, i, s->parity))
- return -1;
- i++;
- list = list->next;
- }
- return 0;
- }
- int QRinput_setFNC1First(QRinput *input)
- {
- if(input->mqr) {
- errno = EINVAL;
- return -1;
- }
- input->fnc1 = 1;
- return 0;
- }
- int QRinput_setFNC1Second(QRinput *input, unsigned char appid)
- {
- if(input->mqr) {
- errno = EINVAL;
- return -1;
- }
- input->fnc1 = 2;
- input->appid = appid;
- return 0;
- }
|