Ops/WireguardConfigGen
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge branch 'master' of https://git.stamm.me/Ops/WireguardConfigGen

Browse Source
This commit is contained in:
Fabian Stamm 2021-04-07 21:14:27 +02:00
commit 346ab97d6a
11 changed files with 1499 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
.editorconfig Normal file
View File

@ -0,0 +1,11 @@
root=true
[*]
charset = utf-8
end_of_line = lf
indent_size = 3
indent_style = space
insert_final_newline = true
[*.yml]
indent_size = 2
[*.md]
indent_size = 2

190
gen.js Normal file
View File

@ -0,0 +1,190 @@
#!node
const fs = require("fs");
const p = require("path");
const qrimage = require("./qr/qr");
const crypto = require("crypto");
const deleteRec = (path) => {
path = p.resolve(path);
if (fs.existsSync(path)) {
if (fs.statSync(path).isDirectory()) {
fs.readdirSync(path).forEach(entry => deleteRec(p.join(path, entry)));
fs.rmdirSync(path);
} else {
fs.unlinkSync(path);
}
}
}
const config = fs.existsSync("config.json") ? JSON.parse(fs.readFileSync("config.json").toString()) : {};
const child = require("child_process");
const getPrivate = () => {
return child.execSync("wg genkey").toString().replace("\n", "").replace("\r", "").trim();
}
const getPublic = (private) => {
return child.execSync(`echo ${private} | wg pubkey`).toString().replace("\n", "").replace("\r", "").trim();
}
const keyPair = () => {
const private = getPrivate();
const public = getPublic(private);
return { private, public };
}
const clientArray = (clients) => Object.keys(clients).map(key => ({ ...clients[key], name: key }))
const ipTablesPostUP = [
"iptables -A FORWARD -i %i -j ACCEPT",
"iptables -A FORWARD -o %i -j ACCEPT",
"iptables -t nat -A POSTROUTING -o %i -j MASQUERADE",
"iptables -t nat -A POSTROUTING -o eth0 -j MASQUERADE",
"iptables -t nat -A POSTROUTING -o wg0 -j MASQUERADE"
].join("; ");
const ipTablesPostDOWN = [
"iptables -D FORWARD -i %i -j ACCEPT",
"iptables -D FORWARD -o %i -j ACCEPT",
"iptables -t nat -D POSTROUTING -o %i -j MASQUERADE",
"iptables -t nat -D POSTROUTING -o eth0 -j MASQUERADE",
"iptables -t nat -D POSTROUTING -o wg0 -j MASQUERADE"
].join("; ");
function makeServer({ port, subnet, subnet6, server, clients }) {
const base = `[Interface] #Server
ListenPort = ${port}
Address = ${subnet}1/32,${subnet6}1/48
PrivateKey = ${server.private}
PostUp = ${ipTablesPostUP}
PostDown = ${ipTablesPostDOWN}
# ---------------------------
`
let peers = clientArray(clients).map(client => {
return `[Peer] #${client.name}
PublicKey = ${client.public}
AllowedIPs = ${subnet}${client.ip}/32,${subnet6}${client.ip}/128
` + (client.endpoint ? "\nEndpoint = " + client.endpoint + ":" + port : "");
}).join("\n\n");
fs.writeFileSync("out/server/server.conf", base + peers);
fs.writeFileSync("out/server/server.public", server.public);
}
function makeClients({ port, subnet, subnet6, server, clients }) {
clientArray(clients).map(client => {
const config = `[Interface]
ListenPort = ${port}
Address = ${subnet}${client.ip}/32, ${subnet6}${client.ip}/128
PrivateKey = ${client.private}
DNS = 1.1.1.1
[Peer] #Server
PublicKey = ${server.public}
AllowedIPs = 0.0.0.0/0, ::0/0
Endpoint = ${server.endpoint}:${port}
`
fs.writeFileSync(`out/clients/${client.name}.conf`, config);
fs.writeFileSync(`out/clients/${client.name}.public`, client.public);
var qr_svg = qrimage.image(config, { type: 'png' });
qr_svg.pipe(require('fs').createWriteStream(`out/clients/${client.name}.png`));
})
}
try {
if (!config.subnet6) {
let sn = crypto.randomBytes(5).toString("hex");
let s1 = sn.substr(0, 2);
let s23 = sn.substr(2, 4);
let s45 = sn.substr(6, 4);
config.subnet6 = `fd${s1}:${s23}:${s45}::`;
}
switch (process.argv[2]) {
case "init": {
const hostname = process.argv[3];
let subnet = process.argv[4];
const port = Number(process.argv[5]) | 51823;
if (!hostname)
throw new Error("Hostname required");
if (!subnet) {
let randomIP1 = Math.floor(Math.random() * 256);
let randomIP2 = Math.floor(Math.random() * 256);
subnet = `10.${randomIP1}.${randomIP2}.`;
} else {
if (!subnet.endsWith("."))
subnet += ".";
}
config.subnet = subnet;
config.port = port;
config.server = {
endpoint: hostname,
...keyPair()
}
config.clients = {};
console.log("Created new configuration");
console.log("Server Hostname:", hostname);
console.log(" Port:", port);
console.log(" Subnet:", subnet + "0/24");
break;
}
case "generate": {
deleteRec("./out");
fs.mkdirSync("./out")
fs.mkdirSync("./out/server")
fs.mkdirSync("./out/clients")
makeServer(config)
makeClients(config)
break;
}
case "add": {
const name = process.argv[3];
const endpoint = process.argv[4];
if (!name)
throw new Error("No name!");
if (config.clients[name]) {
throw new Error("A device with this name exists already!\n Remove with 'remove <name>'");
}
let freeIP = 2;
clientArray(config.clients).forEach(c => c.ip >= freeIP ? freeIP = c.ip + 1 : undefined);
config.clients[name] = {
ip: freeIP,
endpoint,
...keyPair()
}
console.log("Peer added:")
console.log(`IP: ${config.subnet}${freeIP}`)
break;
}
case "remove": {
const name = process.argv[3];
if (!name)
throw new Error("No name!");
delete config.clients[name];
break;
}
default: {
console.log("Wireguard VPN Config generator v0.1");
console.log("Usage:")
console.log("generate, add, remove")
}
}
} catch (err) {
console.error(err);
console.error(err.message);
}
fs.writeFileSync("config.json", JSON.stringify(config, undefined, " "));

19
qr/LICENSE Normal file
View File

@ -0,0 +1,19 @@
Copyright (c) 2013 Yandex LLC
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.

50
qr/crc32buffer.js Normal file
View File

@ -0,0 +1,50 @@
'use strict';
var crc_table = [];
for (var n = 0; n < 256; n++) {
var c = crc_table[n] = Buffer.alloc(4);
c.writeUInt32BE(n, 0);
for (var k = 0; k < 8; k++) {
var b0 = c[0] & 1;
var b1 = c[1] & 1;
var b2 = c[2] & 1;
var b3 = c[3] & 1;
c[0] = (c[0] >> 1) ^ (b3 ? 0xed : 0);
c[1] = (c[1] >> 1) ^ (b3 ? 0xb8 : 0) ^ (b0 ? 0x80 : 0);
c[2] = (c[2] >> 1) ^ (b3 ? 0x83 : 0) ^ (b1 ? 0x80 : 0);
c[3] = (c[3] >> 1) ^ (b3 ? 0x20 : 0) ^ (b2 ? 0x80 : 0);
}
}
function update(c, buf) {
var l = buf.length;
for (var n = 0; n < l; n++) {
var e = crc_table[c[3] ^ buf[n]];
c[3] = e[3] ^ c[2];
c[2] = e[2] ^ c[1];
c[1] = e[1] ^ c[0];
c[0] = e[0];
}
}
function crc32(/* arguments */) {
var l = arguments.length;
var c = Buffer.alloc(4);
c.fill(0xff);
for (var i = 0; i < l; i++) {
update(c, Buffer.from(arguments[i]));
}
c[0] = c[0] ^ 0xff;
c[1] = c[1] ^ 0xff;
c[2] = c[2] ^ 0xff;
c[3] = c[3] ^ 0xff;
return c.readUInt32BE(0);
}
module.exports = crc32;

175
qr/encode.js Normal file
View File

@ -0,0 +1,175 @@
"use strict";
function pushBits(arr, n, value) {
for (var bit = 1 << (n - 1); bit; bit = bit >>> 1) {
arr.push(bit & value ? 1 : 0);
}
}
// {{{1 8bit encode
function encode_8bit(data) {
var len = data.length;
var bits = [];
for (var i = 0; i < len; i++) {
pushBits(bits, 8, data[i]);
}
var res = {};
var d = [0, 1, 0, 0];
pushBits(d, 16, len);
res.data10 = res.data27 = d.concat(bits);
if (len < 256) {
var d = [0, 1, 0, 0];
pushBits(d, 8, len);
res.data1 = d.concat(bits);
}
return res;
}
// {{{1 alphanumeric encode
var ALPHANUM = (function (s) {
var res = {};
for (var i = 0; i < s.length; i++) {
res[s[i]] = i;
}
return res;
})('0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ $%*+-./:');
function encode_alphanum(str) {
var len = str.length;
var bits = [];
for (var i = 0; i < len; i += 2) {
var b = 6;
var n = ALPHANUM[str[i]];
if (str[i + 1]) {
b = 11;
n = n * 45 + ALPHANUM[str[i + 1]];
}
pushBits(bits, b, n);
}
var res = {};
var d = [0, 0, 1, 0];
pushBits(d, 13, len);
res.data27 = d.concat(bits);
if (len < 2048) {
var d = [0, 0, 1, 0];
pushBits(d, 11, len);
res.data10 = d.concat(bits);
}
if (len < 512) {
var d = [0, 0, 1, 0];
pushBits(d, 9, len);
res.data1 = d.concat(bits);
}
return res;
}
// {{{1 numeric encode
function encode_numeric(str) {
var len = str.length;
var bits = [];
for (var i = 0; i < len; i += 3) {
var s = str.substr(i, 3);
var b = Math.ceil(s.length * 10 / 3);
pushBits(bits, b, parseInt(s, 10));
}
var res = {};
var d = [0, 0, 0, 1];
pushBits(d, 14, len);
res.data27 = d.concat(bits);
if (len < 4096) {
var d = [0, 0, 0, 1];
pushBits(d, 12, len);
res.data10 = d.concat(bits);
}
if (len < 1024) {
var d = [0, 0, 0, 1];
pushBits(d, 10, len);
res.data1 = d.concat(bits);
}
return res;
}
// {{{1 url encode
function encode_url(str) {
var slash = str.indexOf('/', 8) + 1 || str.length;
var res = encode(str.slice(0, slash).toUpperCase(), false);
if (slash >= str.length) {
return res;
}
var path_res = encode(str.slice(slash), false);
res.data27 = res.data27.concat(path_res.data27);
if (res.data10 && path_res.data10) {
res.data10 = res.data10.concat(path_res.data10);
}
if (res.data1 && path_res.data1) {
res.data1 = res.data1.concat(path_res.data1);
}
return res;
}
// {{{1 Choose encode mode and generates struct with data for different version
function encode(data, parse_url) {
var str;
var t = typeof data;
if (t == 'string' || t == 'number') {
str = '' + data;
data = Buffer.from(str);
} else if (Buffer.isBuffer(data)) {
str = data.toString();
} else if (Array.isArray(data)) {
data = Buffer.from(data);
str = data.toString();
} else {
throw new Error("Bad data");
}
if (/^[0-9]+$/.test(str)) {
if (data.length > 7089) {
throw new Error("Too much data");
}
return encode_numeric(str);
}
if (/^[0-9A-Z \$%\*\+\.\/\:\-]+$/.test(str)) {
if (data.length > 4296) {
throw new Error("Too much data");
}
return encode_alphanum(str);
}
if (parse_url && /^https?:/i.test(str)) {
return encode_url(str);
}
if (data.length > 2953) {
throw new Error("Too much data");
}
return encode_8bit(data);
}
// {{{1 export functions
module.exports = encode;

77
qr/errorcode.js Normal file
View File

@ -0,0 +1,77 @@
"use strict";
// {{{1 Galois Field Math
var GF256_BASE = 285;
var EXP_TABLE = [1];
var LOG_TABLE = [];
for (var i = 1; i < 256; i++) {
var n = EXP_TABLE[i - 1] << 1;
if (n > 255) n = n ^ GF256_BASE;
EXP_TABLE[i] = n;
}
for (var i = 0; i < 255; i++) {
LOG_TABLE[EXP_TABLE[i]] = i;
}
function exp(k) {
while (k < 0) k += 255;
while (k > 255) k -= 255;
return EXP_TABLE[k];
}
function log(k) {
if (k < 1 || k > 255) {
throw Error('Bad log(' + k + ')');
}
return LOG_TABLE[k];
}
// {{{1 Generator Polynomials
var POLYNOMIALS = [
[0], // a^0 x^0
[0, 0], // a^0 x^1 + a^0 x^0
[0, 25, 1], // a^0 x^2 + a^25 x^1 + a^1 x^0
// and so on...
];
function generatorPolynomial(num) {
if (POLYNOMIALS[num]) {
return POLYNOMIALS[num];
}
var prev = generatorPolynomial(num - 1);
var res = [];
res[0] = prev[0];
for (var i = 1; i <= num; i++) {
res[i] = log(exp(prev[i]) ^ exp(prev[i - 1] + num - 1));
}
POLYNOMIALS[num] = res;
return res;
}
// {{{1 export functions
module.exports = function calculate_ec(msg, ec_len) {
// `msg` could be array or buffer
// convert `msg` to array
msg = [].slice.call(msg);
// Generator Polynomial
var poly = generatorPolynomial(ec_len);
for (var i = 0; i < ec_len; i++) msg.push(0);
while (msg.length > ec_len) {
if (!msg[0]) {
msg.shift();
continue;
}
var log_k = log(msg[0]);
for (var i = 0; i <= ec_len; i++) {
msg[i] = msg[i] ^ exp(poly[i] + log_k);
}
msg.shift();
}
return Buffer.from(msg);
}

352
qr/matrix.js Normal file
View File

@ -0,0 +1,352 @@
"use strict";
// {{{1 Initialize matrix with zeros
function init(version) {
var N = version * 4 + 17;
var matrix = [];
var zeros = new Buffer(N);
zeros.fill(0);
zeros = [].slice.call(zeros);
for (var i = 0; i < N; i++) {
matrix[i] = zeros.slice();
}
return matrix;
}
// {{{1 Put finders into matrix
function fillFinders(matrix) {
var N = matrix.length;
for (var i = -3; i <= 3; i++) {
for (var j = -3; j <= 3; j++) {
var max = Math.max(i, j);
var min = Math.min(i, j);
var pixel = (max == 2 && min >= -2) || (min == -2 && max <= 2) ? 0x80 : 0x81;
matrix[3 + i][3 + j] = pixel;
matrix[3 + i][N - 4 + j] = pixel;
matrix[N - 4 + i][3 + j] = pixel;
}
}
for (var i = 0; i < 8; i++) {
matrix[7][i] = matrix[i][7] =
matrix[7][N - i - 1] = matrix[i][N - 8] =
matrix[N - 8][i] = matrix[N - 1 - i][7] = 0x80;
}
}
// {{{1 Put align and timinig
function fillAlignAndTiming(matrix) {
var N = matrix.length;
if (N > 21) {
var len = N - 13;
var delta = Math.round(len / Math.ceil(len / 28));
if (delta % 2) delta++;
var res = [];
for (var p = len + 6; p > 10; p -= delta) {
res.unshift(p);
}
res.unshift(6);
for (var i = 0; i < res.length; i++) {
for (var j = 0; j < res.length; j++) {
var x = res[i], y = res[j];
if (matrix[x][y]) continue;
for (var r = -2; r <=2 ; r++) {
for (var c = -2; c <=2 ; c++) {
var max = Math.max(r, c);
var min = Math.min(r, c);
var pixel = (max == 1 && min >= -1) || (min == -1 && max <= 1) ? 0x80 : 0x81;
matrix[x + r][y + c] = pixel;
}
}
}
}
}
for (var i = 8; i < N - 8; i++) {
matrix[6][i] = matrix[i][6] = i % 2 ? 0x80 : 0x81;
}
}
// {{{1 Fill reserved areas with zeroes
function fillStub(matrix) {
var N = matrix.length;
for (var i = 0; i < 8; i++) {
if (i != 6) {
matrix[8][i] = matrix[i][8] = 0x80;
}
matrix[8][N - 1 - i] = 0x80;
matrix[N - 1 - i][8] = 0x80;
}
matrix[8][8] = 0x80;
matrix[N - 8][8] = 0x81;
if (N < 45) return;
for (var i = N - 11; i < N - 8; i++) {
for (var j = 0; j < 6; j++) {
matrix[i][j] = matrix[j][i] = 0x80;
}
}
}
// {{{1 Fill reserved areas
var fillReserved = (function() {
var FORMATS = Array(32);
var VERSIONS = Array(40);
var gf15 = 0x0537;
var gf18 = 0x1f25;
var formats_mask = 0x5412;
for (var format = 0; format < 32; format++) {
var res = format << 10;
for (var i = 5; i > 0; i--) {
if (res >>> (9 + i)) {
res = res ^ (gf15 << (i - 1));
}
}
FORMATS[format] = (res | (format << 10)) ^ formats_mask;
}
for (var version = 7; version <= 40; version++) {
var res = version << 12;
for (var i = 6; i > 0; i--) {
if (res >>> (11 + i)) {
res = res ^ (gf18 << (i - 1));
}
}
VERSIONS[version] = (res | (version << 12));
}
var EC_LEVELS = { L: 1, M: 0, Q: 3, H: 2 };
return function fillReserved(matrix, ec_level, mask) {
var N = matrix.length;
var format = FORMATS[EC_LEVELS[ec_level] << 3 | mask];
function F(k) { return format >> k & 1 ? 0x81 : 0x80 };
for (var i = 0; i < 8; i++) {
matrix[8][N - 1 - i] = F(i);
if (i < 6) matrix[i][8] = F(i);
}
for (var i = 8; i < 15; i++) {
matrix[N - 15 + i][8] = F(i);
if (i > 8) matrix[8][14 - i] = F(i);
}
matrix[7][8] = F(6);
matrix[8][8] = F(7);
matrix[8][7] = F(8);
var version = VERSIONS[(N - 17)/4];
if (!version) return;
function V(k) { return version >> k & 1 ? 0x81 : 0x80 };
for (var i = 0; i < 6; i++) {
for (var j = 0; j < 3; j++) {
matrix[N - 11 + j][i] = matrix[i][N - 11 + j] = V(i * 3 + j);
}
}
}
})();
// {{{1 Fill data
var fillData = (function() {
var MASK_FUNCTIONS = [
function(i, j) { return (i + j) % 2 == 0 },
function(i, j) { return i % 2 == 0 },
function(i, j) { return j % 3 == 0 },
function(i, j) { return (i + j) % 3 == 0 },
function(i, j) { return (Math.floor(i / 2) + Math.floor(j / 3) ) % 2 == 0 },
function(i, j) { return (i * j) % 2 + (i * j) % 3 == 0 },
function(i, j) { return ( (i * j) % 2 + (i * j) % 3) % 2 == 0 },
function(i, j) { return ( (i * j) % 3 + (i + j) % 2) % 2 == 0 }
];
return function fillData(matrix, data, mask) {
var N = matrix.length;
var row, col, dir = -1;
row = col = N - 1;
var mask_fn = MASK_FUNCTIONS[mask];
var len = data.blocks[data.blocks.length - 1].length;
for (var i = 0; i < len; i++) {
for (var b = 0; b < data.blocks.length; b++) {
if (data.blocks[b].length <= i) continue;
put(data.blocks[b][i]);
}
}
len = data.ec_len;
for (var i = 0; i < len; i++) {
for (var b = 0; b < data.ec.length; b++) {
put(data.ec[b][i]);
}
}
if (col > -1) {
do {
matrix[row][col] = mask_fn(row, col) ? 1 : 0;
} while (next());
}
function put(byte) {
for (var mask = 0x80; mask; mask = mask >> 1) {
var pixel = !!(mask & byte);
if (mask_fn(row, col)) pixel = !pixel;
matrix[row][col] = pixel ? 1 : 0;
next();
}
}
function next() {
do {
if ((col % 2) ^ (col < 6)) {
if (dir < 0 && row == 0 || dir > 0 && row == N - 1) {
col--;
dir = -dir;
} else {
col++;
row += dir;
}
} else {
col--;
}
if (col == 6) {
col--;
}
if (col < 0) {
return false;
}
} while (matrix[row][col] & 0xf0);
return true;
}
}
})();
// {{{1 Calculate penalty
function calculatePenalty(matrix) {
var N = matrix.length;
var penalty = 0;
// Rule 1
for (var i = 0; i < N; i++) {
var pixel = matrix[i][0] & 1;
var len = 1;
for (var j = 1; j < N; j++) {
var p = matrix[i][j] & 1;
if (p == pixel) {
len++;
continue;
}
if (len >= 5) {
penalty += len - 2;
}
pixel = p;
len = 1;
}
if (len >= 5) {
penalty += len - 2;
}
}
for (var j = 0; j < N; j++) {
var pixel = matrix[0][j] & 1;
var len = 1;
for (var i = 1; i < N; i++) {
var p = matrix[i][j] & 1;
if (p == pixel) {
len++;
continue;
}
if (len >= 5) {
penalty += len - 2;
}
pixel = p;
len = 1;
}
if (len >= 5) {
penalty += len - 2;
}
}
// Rule 2
for (var i = 0; i < N - 1; i++) {
for (var j = 0; j < N - 1; j++) {
var s = matrix[i][j] + matrix[i][j + 1] + matrix[i + 1][j] + matrix[i + 1][j + 1] & 7;
if (s == 0 || s == 4) {
penalty += 3;
}
}
}
// Rule 3
function I(k) { return matrix[i][j + k] & 1 };
function J(k) { return matrix[i + k][j] & 1 };
for (var i = 0; i < N; i++) {
for (var j = 0; j < N; j++) {
if (j < N - 6 && I(0) && !I(1) && I(2) && I(3) && I(4) && !I(5) && I(6)) {
if (j >= 4 && !(I(-4) || I(-3) || I(-2) || I(-1))) {
penalty += 40;
}
if (j < N - 10 && !(I(7) || I(8) || I(9) || I(10))) {
penalty += 40;
}
}
if (i < N - 6 && J(0) && !J(1) && J(2) && J(3) && J(4) && !J(5) && J(6)) {
if (i >= 4 && !(J(-4) || J(-3) || J(-2) || J(-1))) {
penalty += 40;
}
if (i < N - 10 && !(J(7) || J(8) || J(9) || J(10))) {
penalty += 40;
}
}
}
}
// Rule 4
var numDark = 0;
for (var i = 0; i < N; i++) {
for (var j = 0; j < N; j++) {
if (matrix[i][j] & 1) numDark++;
}
}
penalty += 10 * Math.floor(Math.abs(10 - 20 * numDark/(N * N)));
return penalty;
}
// {{{1 All-in-one function
function getMatrix(data) {
var matrix = init(data.version);
fillFinders(matrix);
fillAlignAndTiming(matrix);
fillStub(matrix);
var penalty = Infinity;
var bestMask = 0;
for (var mask = 0; mask < 8; mask++) {
fillData(matrix, data, mask);
fillReserved(matrix, data.ec_level, mask);
var p = calculatePenalty(matrix);
if (p < penalty) {
penalty = p;
bestMask = mask;
}
}
fillData(matrix, data, bestMask);
fillReserved(matrix, data.ec_level, bestMask);
return matrix.map(function(row) {
return row.map(function(cell) {
return cell & 1;
});
});
}
// {{{1 export functions
module.exports = {
getMatrix: getMatrix,
init: init,
fillFinders: fillFinders,
fillAlignAndTiming: fillAlignAndTiming,
fillStub: fillStub,
fillReserved: fillReserved,
fillData: fillData,
calculatePenalty: calculatePenalty,
}

64
qr/png.js Normal file
View File

@ -0,0 +1,64 @@
"use strict";
var zlib = require('zlib');
var crc32 = require('./crc32buffer');
var PNG_HEAD = new Buffer([137, 80, 78, 71, 13, 10, 26, 10]);
var PNG_IHDR = new Buffer([0, 0, 0, 13, 73, 72, 68, 82, 0, 0, 0, 0, 0, 0, 0, 0, 8, 0, 0, 0, 0, 0, 0, 0, 0]);
var PNG_IDAT = new Buffer([0, 0, 0, 0, 73, 68, 65, 84]);
var PNG_IEND = new Buffer([0, 0, 0, 0, 73, 69, 78, 68, 174, 66, 96, 130]);
function png(bitmap, stream) {
stream.push(PNG_HEAD);
var IHDR = Buffer.concat([PNG_IHDR]);
IHDR.writeUInt32BE(bitmap.size, 8);
IHDR.writeUInt32BE(bitmap.size, 12);
IHDR.writeUInt32BE(crc32(IHDR.slice(4, -4)), 21);
stream.push(IHDR);
var IDAT = Buffer.concat([
PNG_IDAT,
zlib.deflateSync(bitmap.data, { level: 9 }),
new Buffer(4)
]);
IDAT.writeUInt32BE(IDAT.length - 12, 0);
IDAT.writeUInt32BE(crc32(IDAT.slice(4, -4)), IDAT.length - 4);
stream.push(IDAT);
stream.push(PNG_IEND);
stream.push(null);
}
function bitmap(matrix, size, margin) {
var N = matrix.length;
var X = (N + 2 * margin) * size;
var data = new Buffer((X + 1) * X);
data.fill(255);
for (var i = 0; i < X; i++) {
data[i * (X + 1)] = 0;
}
for (var i = 0; i < N; i++) {
for (var j = 0; j < N; j++) {
if (matrix[i][j]) {
var offset = ((margin + i) * (X + 1) + (margin + j)) * size + 1;
data.fill(0, offset, offset + size);
for (var c = 1; c < size; c++) {
data.copy(data, offset + c * (X + 1), offset, offset + size);
}
}
}
}
return {
data: data,
size: X
}
}
module.exports = {
bitmap: bitmap,
png: png
}

180
qr/qr-base.js Normal file
View File

@ -0,0 +1,180 @@
"use strict";
var encode = require('./encode');
var calculateEC = require('./errorcode');
var matrix = require('./matrix');
function _deepCopy(obj) {
return JSON.parse(JSON.stringify(obj));
}
var EC_LEVELS = ['L', 'M', 'Q', 'H'];
// {{{1 Versions
var versions = [
[], // there is no version 0
// total number of codewords, (number of ec codewords, number of blocks) * ( L, M, Q, H )
[26, 7, 1, 10, 1, 13, 1, 17, 1],
[44, 10, 1, 16, 1, 22, 1, 28, 1],
[70, 15, 1, 26, 1, 36, 2, 44, 2],
[100, 20, 1, 36, 2, 52, 2, 64, 4],
[134, 26, 1, 48, 2, 72, 4, 88, 4], // 5
[172, 36, 2, 64, 4, 96, 4, 112, 4],
[196, 40, 2, 72, 4, 108, 6, 130, 5],
[242, 48, 2, 88, 4, 132, 6, 156, 6],
[292, 60, 2, 110, 5, 160, 8, 192, 8],
[346, 72, 4, 130, 5, 192, 8, 224, 8], // 10
[404, 80, 4, 150, 5, 224, 8, 264, 11],
[466, 96, 4, 176, 8, 260, 10, 308, 11],
[532, 104, 4, 198, 9, 288, 12, 352, 16],
[581, 120, 4, 216, 9, 320, 16, 384, 16],
[655, 132, 6, 240, 10, 360, 12, 432, 18], // 15
[733, 144, 6, 280, 10, 408, 17, 480, 16],
[815, 168, 6, 308, 11, 448, 16, 532, 19],
[901, 180, 6, 338, 13, 504, 18, 588, 21],
[991, 196, 7, 364, 14, 546, 21, 650, 25],
[1085, 224, 8, 416, 16, 600, 20, 700, 25], // 20
[1156, 224, 8, 442, 17, 644, 23, 750, 25],
[1258, 252, 9, 476, 17, 690, 23, 816, 34],
[1364, 270, 9, 504, 18, 750, 25, 900, 30],
[1474, 300, 10, 560, 20, 810, 27, 960, 32],
[1588, 312, 12, 588, 21, 870, 29, 1050, 35], // 25
[1706, 336, 12, 644, 23, 952, 34, 1110, 37],
[1828, 360, 12, 700, 25, 1020, 34, 1200, 40],
[1921, 390, 13, 728, 26, 1050, 35, 1260, 42],
[2051, 420, 14, 784, 28, 1140, 38, 1350, 45],
[2185, 450, 15, 812, 29, 1200, 40, 1440, 48], // 30
[2323, 480, 16, 868, 31, 1290, 43, 1530, 51],
[2465, 510, 17, 924, 33, 1350, 45, 1620, 54],
[2611, 540, 18, 980, 35, 1440, 48, 1710, 57],
[2761, 570, 19, 1036, 37, 1530, 51, 1800, 60],
[2876, 570, 19, 1064, 38, 1590, 53, 1890, 63], // 35
[3034, 600, 20, 1120, 40, 1680, 56, 1980, 66],
[3196, 630, 21, 1204, 43, 1770, 59, 2100, 70],
[3362, 660, 22, 1260, 45, 1860, 62, 2220, 74],
[3532, 720, 24, 1316, 47, 1950, 65, 2310, 77],
[3706, 750, 25, 1372, 49, 2040, 68, 2430, 81] // 40
];
versions = versions.map(function(v, index) {
if (!index) return {};
var res = {
}
for (var i = 1; i < 8; i += 2) {
var length = v[0] - v[i];
var num_template = v[i+1];
var ec_level = EC_LEVELS[(i/2)|0];
var level = {
version: index,
ec_level: ec_level,
data_len: length,
ec_len: v[i] / num_template,
blocks: [],
ec: []
}
for (var k = num_template, n = length; k > 0; k--) {
var block = (n / k)|0;
level.blocks.push(block);
n -= block;
}
res[ec_level] = level;
}
return res;
});
// {{{1 Get version template
function getTemplate(message, ec_level) {
var i = 1;
var len;
if (message.data1) {
len = Math.ceil(message.data1.length / 8);
} else {
i = 10;
}
for (/* i */; i < 10; i++) {
var version = versions[i][ec_level];
if (version.data_len >= len) {
return _deepCopy(version);
}
}
if (message.data10) {
len = Math.ceil(message.data10.length / 8);
} else {
i = 27;
}
for (/* i */; i < 27; i++) {
var version = versions[i][ec_level];
if (version.data_len >= len) {
return _deepCopy(version);
}
}
len = Math.ceil(message.data27.length / 8);
for (/* i */; i < 41; i++) {
var version = versions[i][ec_level];
if (version.data_len >= len) {
return _deepCopy(version);
}
}
throw new Error("Too much data");
}
// {{{1 Fill template
function fillTemplate(message, template) {
var blocks = new Buffer(template.data_len);
blocks.fill(0);
if (template.version < 10) {
message = message.data1;
} else if (template.version < 27) {
message = message.data10;
} else {
message = message.data27;
}
var len = message.length;
for (var i = 0; i < len; i += 8) {
var b = 0;
for (var j = 0; j < 8; j++) {
b = (b << 1) | (message[i + j] ? 1 : 0);
}
blocks[i / 8] = b;
}
var pad = 236;
for (var i = Math.ceil((len + 4) / 8); i < blocks.length; i++) {
blocks[i] = pad;
pad = (pad == 236) ? 17 : 236;
}
var offset = 0;
template.blocks = template.blocks.map(function(n) {
var b = blocks.slice(offset, offset + n);
offset += n;
template.ec.push(calculateEC(b, template.ec_len));
return b;
});
return template;
}
// {{{1 All-in-one
function QR(text, ec_level, parse_url) {
ec_level = EC_LEVELS.indexOf(ec_level) > -1 ? ec_level : 'M';
var message = encode(text, parse_url);
var data = fillTemplate(message, getTemplate(message, ec_level));
return matrix.getMatrix(data);
}
// {{{1 export functions
module.exports = {
QR: QR,
getTemplate: getTemplate,
fillTemplate: fillTemplate,
}

121
qr/qr.js Normal file
View File

@ -0,0 +1,121 @@
"use strict";
var Readable = require('stream').Readable;
var QR = require('./qr-base').QR;
var png = require('./png');
var vector = require('./vector');
var fn_noop = function() {};
var BITMAP_OPTIONS = {
parse_url: false,
ec_level: 'M',
size: 5,
margin: 4,
customize: null
};
var VECTOR_OPTIONS = {
parse_url: false,
ec_level: 'M',
margin: 1,
size: 0
};
function get_options(options, force_type) {
if (typeof options === 'string') {
options = { 'ec_level': options }
} else {
options = options || {};
}
var _options = {
type: String(force_type || options.type || 'png').toLowerCase()
};
var defaults = _options.type == 'png' ? BITMAP_OPTIONS : VECTOR_OPTIONS;
for (var k in defaults) {
_options[k] = k in options ? options[k] : defaults[k];
}
return _options;
}
function qr_image(text, options) {
options = get_options(options);
var matrix = QR(text, options.ec_level, options.parse_url);
var stream = new Readable();
stream._read = fn_noop;
switch (options.type) {
case 'svg':
case 'pdf':
case 'eps':
process.nextTick(function() {
vector[options.type](matrix, stream, options.margin, options.size);
});
break;
case 'svgpath':
// deprecated, use svg_object method
process.nextTick(function() {
var obj = vector.svg_object(matrix, options.margin, options.size);
stream.push(obj.path);
stream.push(null);
});
break;
case 'png':
default:
process.nextTick(function() {
var bitmap = png.bitmap(matrix, options.size, options.margin);
if (options.customize) {
options.customize(bitmap);
}
png.png(bitmap, stream);
});
}
return stream;
}
function qr_image_sync(text, options) {
options = get_options(options);
var matrix = QR(text, options.ec_level, options.parse_url);
var stream = [];
var result;
switch (options.type) {
case 'svg':
case 'pdf':
case 'eps':
vector[options.type](matrix, stream, options.margin, options.size);
result = stream.filter(Boolean).join('');
break;
case 'png':
default:
var bitmap = png.bitmap(matrix, options.size, options.margin);
if (options.customize) {
options.customize(bitmap);
}
png.png(bitmap, stream);
result = Buffer.concat(stream.filter(Boolean));
}
return result;
}
function svg_object(text, options) {
options = get_options(options, 'svg');
var matrix = QR(text, options.ec_level);
return vector.svg_object(matrix, options.margin);
}
module.exports = {
matrix: QR,
image: qr_image,
imageSync: qr_image_sync,
svgObject: svg_object
};

260
qr/vector.js Normal file
View File

@ -0,0 +1,260 @@
"use strict";
function matrix2path(matrix) {
var N = matrix.length;
var filled = [];
for (var row = -1; row <= N; row++) {
filled[row] = [];
}
var path = [];
for (var row = 0; row < N; row++) {
for (var col = 0; col < N; col++) {
if (filled[row][col]) continue;
filled[row][col] = 1;
if (isDark(row, col)) {
if (!isDark(row - 1, col)) {
path.push(plot(row, col, 'right'));
}
} else {
if (isDark(row, col - 1)) {
path.push(plot(row, col, 'down'));
}
}
}
}
return path;
function isDark(row, col) {
if (row < 0 || col < 0 || row >= N || col >= N) return false;
return !!matrix[row][col];
}
function plot(row0, col0, dir) {
filled[row0][col0] = 1;
var res = [];
res.push(['M', col0, row0 ]);
var row = row0;
var col = col0;
var len = 0;
do {
switch (dir) {
case 'right':
filled[row][col] = 1;
if (isDark(row, col)) {
filled[row - 1][col] = 1;
if (isDark(row - 1, col)) {
res.push(['h', len]);
len = 0;
dir = 'up';
} else {
len++;
col++;
}
} else {
res.push(['h', len]);
len = 0;
dir = 'down';
}
break;
case 'left':
filled[row - 1][col - 1] = 1;
if (isDark(row - 1, col - 1)) {
filled[row][col - 1] = 1;
if (isDark(row, col - 1)) {
res.push(['h', -len]);
len = 0;
dir = 'down';
} else {
len++;
col--;
}
} else {
res.push(['h', -len]);
len = 0;
dir = 'up';
}
break;
case 'down':
filled[row][col - 1] = 1;
if (isDark(row, col - 1)) {
filled[row][col] = 1;
if (isDark(row, col)) {
res.push(['v', len]);
len = 0;
dir = 'right';
} else {
len++;
row++;
}
} else {
res.push(['v', len]);
len = 0;
dir = 'left';
}
break;
case 'up':
filled[row - 1][col] = 1;
if (isDark(row - 1, col)) {
filled[row - 1][col - 1] = 1;
if (isDark(row - 1, col - 1)) {
res.push(['v', -len]);
len = 0;
dir = 'left';
} else {
len++;
row--;
}
} else {
res.push(['v', -len]);
len = 0;
dir = 'right';
}
break;
}
} while (row != row0 || col != col0);
return res;
}
}
function pushSVGPath(matrix, stream, margin) {
matrix2path(matrix).forEach(function(subpath) {
var res = '';
for (var k = 0; k < subpath.length; k++) {
var item = subpath[k];
switch (item[0]) {
case 'M':
res += 'M' + (item[1] + margin) + ' ' + (item[2] + margin);
break;
default:
res += item.join('');
}
}
res += 'z';
stream.push(res);
});
}
function SVG_object(matrix, margin) {
var stream = [];
pushSVGPath(matrix, stream, margin);
var result = {
size: matrix.length + 2 * margin,
path: stream.filter(Boolean).join('')
}
return result;
}
function SVG(matrix, stream, margin, size) {
var X = matrix.length + 2 * margin;
stream.push('<svg xmlns="http://www.w3.org/2000/svg" ');
if (size > 0) {
var XY = X * size;
stream.push('width="' + XY + '" height="' + XY + '" ');
}
stream.push('viewBox="0 0 ' + X + ' ' + X + '">');
stream.push('<path d="');
pushSVGPath(matrix, stream, margin);
stream.push('"/></svg>');
stream.push(null);
}
function EPS(matrix, stream, margin) {
var N = matrix.length;
var scale = 9;
var X = (N + 2 * margin) * scale;
stream.push([
'%!PS-Adobe-3.0 EPSF-3.0',
'%%BoundingBox: 0 0 ' + X + ' ' + X,
'/h { 0 rlineto } bind def',
'/v { 0 exch neg rlineto } bind def',
'/M { neg ' + (N + margin) + ' add moveto } bind def',
'/z { closepath } bind def',
scale + ' ' + scale + ' scale',
''
].join('\n'));
matrix2path(matrix).forEach(function(subpath) {
var res = '';
for (var k = 0; k < subpath.length; k++) {
var item = subpath[k];
switch (item[0]) {
case 'M':
res += (item[1] + margin) + ' ' + item[2] + ' M ';
break;
default:
res += item[1] + ' ' + item[0] + ' ';
}
}
res += 'z\n';
stream.push(res);
});
stream.push('fill\n%%EOF\n');
stream.push(null);
}
function PDF(matrix, stream, margin) {
// TODO deflate
var N = matrix.length;
var scale = 9;
var X = (N + 2 * margin) * scale;
var data = [
'%PDF-1.0\n\n',
'1 0 obj << /Type /Catalog /Pages 2 0 R >> endobj\n',
'2 0 obj << /Type /Pages /Count 1 /Kids [ 3 0 R ] >> endobj\n',
];
data.push('3 0 obj << /Type /Page /Parent 2 0 R /Resources <<>> ' +
'/Contents 4 0 R /MediaBox [ 0 0 ' + X + ' ' + X + ' ] >> endobj\n');
var path = scale + ' 0 0 ' + scale + ' 0 0 cm\n';
path += matrix2path(matrix).map(function(subpath) {
var res = '';
var x, y;
for (var k = 0; k < subpath.length; k++) {
var item = subpath[k];
switch (item[0]) {
case 'M':
x = item[1] + margin;
y = N - item[2] + margin;
res += x + ' ' + y + ' m ';
break;
case 'h':
x += item[1];
res += x + ' ' + y + ' l ';
break;
case 'v':
y -= item[1];
res += x + ' ' + y + ' l ';
break;
}
}
res += 'h';
return res;
}).join('\n');
path += '\nf\n';
data.push('4 0 obj << /Length ' + path.length + ' >> stream\n' +
path + 'endstream\nendobj\n');
var xref = 'xref\n0 5\n0000000000 65535 f \n';
for (var i = 1, l = data[0].length; i < 5; i++) {
xref += ('0000000000' + l).substr(-10) + ' 00000 n \n';
l += data[i].length;
}
data.push(
xref,
'trailer << /Root 1 0 R /Size 5 >>\n',
'startxref\n' + l + '\n%%EOF\n'
);
stream.push(data.join(''));
stream.push(null);
}
module.exports = {
svg: SVG,
eps: EPS,
pdf: PDF,
svg_object: SVG_object
}