From 9c300cd19102eb383dc233b442e00fbdfd6d4c46 Mon Sep 17 00:00:00 2001 From: MiaoWoo Date: Mon, 22 Jul 2019 16:08:35 +0800 Subject: [PATCH] feat: optimize code Signed-off-by: MiaoWoo --- src/core/buffer.ts | 771 ++++++++++++++++++++++----------------------- 1 file changed, 370 insertions(+), 401 deletions(-) diff --git a/src/core/buffer.ts b/src/core/buffer.ts index 2f3ff39..6ae2d67 100644 --- a/src/core/buffer.ts +++ b/src/core/buffer.ts @@ -16,217 +16,206 @@ function Write(buf, data, StringFormat?, ParamValue?, WorkStruct?) { global.ToLogTrace("ERRR StringFormat "); throw "ERR!!"; } - else { - var format = StringFormat; - if (format.substr(0, 6) === "buffer" && format.length > 6) { - ParamValue = parseInt(format.substr(6)); - format = "buffer"; - } - else - if (format.substr(0, 3) === "arr" && format.length > 3) { - ParamValue = parseInt(format.substr(3)); - format = "arr"; + var format = StringFormat; + if (format.substr(0, 6) === "buffer" && format.length > 6) { + ParamValue = parseInt(format.substr(6)); + format = "buffer"; + } else if (format.substr(0, 3) === "arr" && format.length > 3) { + ParamValue = parseInt(format.substr(3)); + format = "arr"; + } else if (format.substr(0, 3) === "str" && format.length > 3) { + var length = parseInt(format.substr(3)); + if (data) + buf.write(data, buf.len, length); + buf.len += length; + return; + } + switch (format) { + case "str": + { + var arr = global.toUTF8Array(data); + let length = arr.length; + if (length > 65535) + length = 0; + buf[buf.len] = length & 255; + buf[buf.len + 1] = (length >>> 8) & 255; + buf.len += 2; + for (var i = 0; i < length; i++) { + buf[buf.len + i] = arr[i]; + } + buf.len += length; + break; } - else - if (format.substr(0, 3) === "str" && format.length > 3) { - var length = parseInt(format.substr(3)); - if (data) - buf.write(data, buf.len, length); - buf.len += length; - return; + case "byte": + { + if (data < 0) + data = 0; + buf[buf.len] = data; + buf.len += 1; + break; + } + case "double": + { + buf.writeDoubleLE(data, buf.len, 8); + buf.len += 8; + break; + } + case "uint": + { + if (data < 0) + data = 0; + if (data >= 281474976710655) + data = 0; + buf.writeUIntLE(data, buf.len, 6); + buf.len += 6; + break; + } + case "uint16": + { + if (data < 0) + data = 0; + buf[buf.len] = data & 255; + buf[buf.len + 1] = (data >>> 8) & 255; + buf.len += 2; + break; + } + case "uint32": + { + if (data < 0) + data = 0; + buf.writeUInt32LE(data, buf.len, 4); + buf.len += 4; + break; + } + case "time": + { + var Time = data.valueOf(); + buf.writeUIntLE(Time, buf.len, 6); + buf.len += 6; + break; + } + case "addres": + case "hash": + { + let length; + if (data) + length = Math.min(32, data.length); + else + length = 0; + for (var i = 0; i < length; i++) { + buf[buf.len + i] = data[i]; } - switch (format) { - case "str": - { - var arr = global.toUTF8Array(data); - let length = arr.length; - if (length > 65535) - length = 0; - buf[buf.len] = length & 255; - buf[buf.len + 1] = (length >>> 8) & 255; - buf.len += 2; - for (var i = 0; i < length; i++) { - buf[buf.len + i] = arr[i]; - } - buf.len += length; - break; + buf.len += 32; + break; + } + case "buffer": + { + let length; + if (ParamValue === undefined) + length = data.length; + else + length = Math.min(ParamValue, data.length); + for (var i = 0; i < length; i++) { + buf[buf.len + i] = data[i]; } - case "byte": - { - if (data < 0) - data = 0; - buf[buf.len] = data; - buf.len += 1; - break; + buf.len += ParamValue; + break; + } + case "arr": + { + let length; + if (data) + length = Math.min(ParamValue, data.length); + else + length = 0; + for (var i = 0; i < length; i++) { + buf[buf.len + i] = data[i]; } - case "double": - { - buf.writeDoubleLE(data, buf.len, 8); - buf.len += 8; - break; + buf.len += ParamValue; + break; + } + case "tr": + { + let length = data.length; + if (global.MAX_TRANSACTION_SIZE > global.MAX_TRANSACTION_SIZE) + length = global.MAX_TRANSACTION_SIZE; + buf[buf.len] = length & 255; + buf[buf.len + 1] = (length >>> 8) & 255; + buf.len += 2; + for (var i = 0; i < length; i++) { + buf[buf.len + i] = data[i]; } - case "uint": - { - if (data < 0) - data = 0; - if (data >= 281474976710655) - data = 0; - buf.writeUIntLE(data, buf.len, 6); - buf.len += 6; - break; + buf.len += length; + break; + } + case "data": + { + let length = data.length; + buf.writeUInt32LE(length, buf.len, 4); + buf.len += 4; + for (var i = 0; i < length; i++) { + buf[buf.len + i] = data[i]; } - case "uint16": - { - if (data < 0) - data = 0; - buf[buf.len] = data & 255; - buf[buf.len + 1] = (data >>> 8) & 255; - buf.len += 2; - break; - } - case "uint32": - { - if (data < 0) - data = 0; - buf.writeUInt32LE(data, buf.len, 4); - buf.len += 4; - break; - } - case "time": - { - var Time = data.valueOf(); - buf.writeUIntLE(Time, buf.len, 6); - buf.len += 6; - break; - } - case "addres": - case "hash": - { + buf.len += length; + break; + } + case "hashSTR": + { + var Str = global.GetHexFromAddres(data); + buf.write(Str, buf.len, 64); + buf.len += 64; + break; + } + case "uintSTR": + { + var Str = data.toString(); + buf.write(Str, buf.len, 10); + buf.len += 10; + break; + } + default: + { + WorkStruct = WorkStruct || {}; + var CurFormat = StringFormat.substr(0, 1); + if (CurFormat === "[") { let length; if (data) - length = Math.min(32, data.length); - else - length = 0; - for (var i = 0; i < length; i++) { - buf[buf.len + i] = data[i]; - } - buf.len += 32; - break; - } - case "buffer": - { - let length; - if (ParamValue === undefined) length = data.length; - else - length = Math.min(ParamValue, data.length); + var formatNext = GetMiddleString(format); + Write(buf, length, "uint32"); for (var i = 0; i < length; i++) { - buf[buf.len + i] = data[i]; + Write(buf, data[i], formatNext, undefined, WorkStruct); } - buf.len += ParamValue; - break; - } - case "arr": - { + } else if (CurFormat === "<") { let length; if (data) - length = Math.min(ParamValue, data.length); - else - length = 0; - for (var i = 0; i < length; i++) { - buf[buf.len + i] = data[i]; - } - buf.len += ParamValue; - break; - } - case "tr": - { - let length = data.length; - if (global.MAX_TRANSACTION_SIZE > global.MAX_TRANSACTION_SIZE) - length = global.MAX_TRANSACTION_SIZE; - buf[buf.len] = length & 255; - buf[buf.len + 1] = (length >>> 8) & 255; - buf.len += 2; - for (var i = 0; i < length; i++) { - buf[buf.len + i] = data[i]; - } - buf.len += length; - break; - } - case "data": - { - let length = data.length; - buf.writeUInt32LE(length, buf.len, 4); + length = data.length; + var formatNext = GetMiddleString(format); + var IndexCount = 0; + var len = buf.len; buf.len += 4; for (var i = 0; i < length; i++) { - buf[buf.len + i] = data[i]; - } - buf.len += length; - break; - } - case "hashSTR": - { - var Str = global.GetHexFromAddres(data); - buf.write(Str, buf.len, 64); - buf.len += 64; - break; - } - case "uintSTR": - { - var Str = data.toString(); - buf.write(Str, buf.len, 10); - buf.len += 10; - break; - } - default: - { - WorkStruct = WorkStruct || {}; - var CurFormat = StringFormat.substr(0, 1); - if (CurFormat === "[") { - let length; - if (data) - length = data.length; - var formatNext = GetMiddleString(format); - Write(buf, length, "uint32"); - for (var i = 0; i < length; i++) { + if (data[i]) { + IndexCount++; + Write(buf, i, "uint32"); Write(buf, data[i], formatNext, undefined, WorkStruct); } } - else - if (CurFormat === "<") { - let length; - if (data) - length = data.length; - var formatNext = GetMiddleString(format); - var IndexCount = 0; - var len = buf.len; - buf.len += 4; - for (var i = 0; i < length; i++) { - if (data[i]) { - IndexCount++; - Write(buf, i, "uint32"); - Write(buf, data[i], formatNext, undefined, WorkStruct); - } - } - buf.writeUInt32LE(IndexCount, len, 4); - } - else - if (CurFormat === "{") { - var attrs = WorkStruct[format]; - if (!attrs) { - attrs = GetAttributes(GetMiddleString(format)); - WorkStruct[format] = attrs; - } - for (var i = 0; i < attrs.length; i++) { - var type = attrs[i]; - Write(buf, data[type.Key], type.Value, undefined, WorkStruct); - } - } - else { - throw "Bad write type params: " + format; - } + buf.writeUInt32LE(IndexCount, len, 4); + } else if (CurFormat === "{") { + var attrs = WorkStruct[format]; + if (!attrs) { + attrs = GetAttributes(GetMiddleString(format)); + WorkStruct[format] = attrs; + } + for (var i = 0; i < attrs.length; i++) { + var type = attrs[i]; + Write(buf, data[type.Key], type.Value, undefined, WorkStruct); + } + } else { + throw "Bad write type params: " + format; } - } + } } }; @@ -236,225 +225,205 @@ function Read(buf, StringFormat, ParamValue?, WorkStruct?, bDisableTime?) { global.ToLogTrace("ERR StringFormat"); throw "ERRR!"; } - else { - var format = StringFormat; - if (format.substr(0, 6) === "buffer") { - if (format.length > 6) { - ParamValue = parseInt(format.substr(6)); - format = "buffer"; - } - else { - ParamValue = 0; - } + var format = StringFormat; + if (format.substr(0, 6) === "buffer") { + if (format.length > 6) { + ParamValue = parseInt(format.substr(6)); + format = "buffer"; + } else { + ParamValue = 0; } - else - if (format.substr(0, 3) === "arr") { - if (format.length > 3) { - ParamValue = parseInt(format.substr(3)); - format = "arr"; - } - else { - ParamValue = 0; + } else if (format.substr(0, 3) === "arr") { + if (format.length > 3) { + ParamValue = parseInt(format.substr(3)); + format = "arr"; + } else { + ParamValue = 0; + } + } else if (format.substr(0, 3) === "str") { + if (format.length > 3) { + var length = parseInt(format.substr(3)); + ret = buf.toString('utf8', buf.len, buf.len + length); + buf.len += length; + var nEnd = - 1; + for (var i = ret.length - 1; i >= 0; i--) { + if (ret.charCodeAt(i) !== 0) { + nEnd = i; + break; } } - else - if (format.substr(0, 3) === "str") { - if (format.length > 3) { - var length = parseInt(format.substr(3)); - ret = buf.toString('utf8', buf.len, buf.len + length); - buf.len += length; - var nEnd = - 1; - for (var i = ret.length - 1; i >= 0; i--) { - if (ret.charCodeAt(i) !== 0) { - nEnd = i; - break; - } - } - if (nEnd >= 0) - ret = ret.substr(0, i + 1); - else - ret = ""; - return ret; - } - else { - ParamValue = 0; - } - } - switch (format) { - case "str": - { - let length; - if (buf.len + 2 <= buf.length) - length = buf[buf.len] + buf[buf.len + 1] * 256; + return nEnd >= 0 ? ret.substr(0, i + 1) : ""; + } else { + ParamValue = 0; + } + } + switch (format) { + case "str": + { + let length; + if (buf.len + 2 <= buf.length) + length = buf[buf.len] + buf[buf.len + 1] * 256; + else + length = 0; + buf.len += 2; + var arr = buf.slice(buf.len, buf.len + length); + ret = global.Utf8ArrayToStr(arr); + buf.len += length; + break; + } + case "byte": + { + if (buf.len + 1 <= buf.length) + ret = buf[buf.len]; + else + ret = 0; + buf.len += 1; + break; + } + case "double": + { + if (buf.len + 8 <= buf.length) + ret = buf.readDoubleLE(buf.len, 8); + else + ret = 0; + buf.len += 8; + break; + } + case "uint": + { + if (buf.len + 6 <= buf.length) + ret = buf.readUIntLE(buf.len, 6); + else + ret = 0; + buf.len += 6; + break; + } + case "uint16": + { + if (buf.len + 2 <= buf.length) + ret = buf[buf.len] + buf[buf.len + 1] * 256; + else + ret = 0; + buf.len += 2; + break; + } + case "uint32": + { + if (buf.len + 4 <= buf.length) + ret = buf.readUInt32LE(buf.len, 4); + else + ret = 0; + buf.len += 4; + break; + } + case "time": + { + if (bDisableTime) + throw "Bad read type params: time - DisableTime ON"; + if (buf.len + 6 <= buf.length) + ret = buf.readUIntLE(buf.len, 6); + else + ret = 0; + ret = new Date(ret); + buf.len += 6; + break; + } + case "addres": + case "hash": + { + ret = []; + for (var i = 0; i < 32; i++) { + if (buf.len + i <= buf.length) + ret[i] = buf[buf.len + i]; else - length = 0; - buf.len += 2; - var arr = buf.slice(buf.len, buf.len + length); - ret = global.Utf8ArrayToStr(arr); - buf.len += length; + ret[i] = 0; + } + buf.len += 32; + break; + } + case "buffer": + case "arr": + { + if (buf.len + ParamValue <= buf.length) + ret = buf.slice(buf.len, buf.len + ParamValue); + else + ret = Buffer.alloc(ParamValue); + buf.len += ParamValue; + break; + } + case "tr": + { + if (buf.len + 1 >= buf.length) { + ret = undefined; break; } - case "byte": - { - if (buf.len + 1 <= buf.length) - ret = buf[buf.len]; - else - ret = 0; - buf.len += 1; - break; - } - case "double": - { - if (buf.len + 8 <= buf.length) - ret = buf.readDoubleLE(buf.len, 8); - else - ret = 0; - buf.len += 8; - break; - } - case "uint": - { - if (buf.len + 6 <= buf.length) - ret = buf.readUIntLE(buf.len, 6); - else - ret = 0; - buf.len += 6; - break; - } - case "uint16": - { - if (buf.len + 2 <= buf.length) - ret = buf[buf.len] + buf[buf.len + 1] * 256; - else - ret = 0; - buf.len += 2; - break; - } - case "uint32": - { - if (buf.len + 4 <= buf.length) - ret = buf.readUInt32LE(buf.len, 4); - else - ret = 0; - buf.len += 4; - break; - } - case "time": - { - if (bDisableTime) - throw "Bad read type params: time - DisableTime ON"; - if (buf.len + 6 <= buf.length) - ret = buf.readUIntLE(buf.len, 6); - else - ret = 0; - ret = new Date(ret); - buf.len += 6; - break; - } - case "addres": - case "hash": - { + let length = buf[buf.len] + buf[buf.len + 1] * 256; + buf.len += 2; + ret = buf.slice(buf.len, buf.len + length); + buf.len += length; + break; + } + case "data": + { + let length; + if (buf.len + 4 <= buf.length) + length = buf.readUInt32LE(buf.len, 4); + else + length = 0; + if (length > buf.length - buf.len - 4) + length = 0; + buf.len += 4; + ret = buf.slice(buf.len, buf.len + length); + buf.len += length; + break; + } + case "hashSTR": + { + var Str = buf.toString('utf8', buf.len, buf.len + 64); + ret = global.GetAddresFromHex(Str); + buf.len += 64; + break; + } + case "uintSTR": + { + var Str = buf.toString('utf8', buf.len, buf.len + 10); + ret = parseInt(Str); + buf.len += 10; + break; + } + default: + { + WorkStruct = WorkStruct || {}; + var LStr = format.substr(0, 1); + if (LStr === "[" || LStr === "<") { + var bIndexArr = (LStr === "<"); ret = []; - for (var i = 0; i < 32; i++) { - if (buf.len + i <= buf.length) - ret[i] = buf[buf.len + i]; - else - ret[i] = 0; - } - buf.len += 32; - break; - } - case "buffer": - case "arr": - { - if (buf.len + ParamValue <= buf.length) - ret = buf.slice(buf.len, buf.len + ParamValue); - else - ret = Buffer.alloc(ParamValue); - buf.len += ParamValue; - break; - } - case "tr": - { - if (buf.len + 1 >= buf.length) { - ret = undefined; - break; - } - let length = buf[buf.len] + buf[buf.len + 1] * 256; - buf.len += 2; - ret = buf.slice(buf.len, buf.len + length); - buf.len += length; - break; - } - case "data": - { - let length; - if (buf.len + 4 <= buf.length) - length = buf.readUInt32LE(buf.len, 4); - else - length = 0; - if (length > buf.length - buf.len - 4) - length = 0; - buf.len += 4; - ret = buf.slice(buf.len, buf.len + length); - buf.len += length; - break; - } - case "hashSTR": - { - var Str = buf.toString('utf8', buf.len, buf.len + 64); - ret = global.GetAddresFromHex(Str); - buf.len += 64; - break; - } - case "uintSTR": - { - var Str = buf.toString('utf8', buf.len, buf.len + 10); - ret = parseInt(Str); - buf.len += 10; - break; - } - default: - { - WorkStruct = WorkStruct || {}; - var LStr = format.substr(0, 1); - if (LStr === "[" || LStr === "<") { - var bIndexArr = (LStr === "<"); - ret = []; - var formatNext = GetMiddleString(format); - let length = Read(buf, "uint32"); - for (var i = 0; i < length; i++) { - if (buf.len <= buf.length) { - if (bIndexArr) { - var index = Read(buf, "uint32"); - ret[index] = Read(buf, formatNext, undefined, WorkStruct, bDisableTime); - } - else { - ret[i] = Read(buf, formatNext, undefined, WorkStruct, bDisableTime); - } - } - else - break; + var formatNext = GetMiddleString(format); + let length = Read(buf, "uint32"); + for (var i = 0; i < length; i++) { + if (buf.len > buf.length) { break; } + if (bIndexArr) { + var index = Read(buf, "uint32"); + ret[index] = Read(buf, formatNext, undefined, WorkStruct, bDisableTime); + } else { + ret[i] = Read(buf, formatNext, undefined, WorkStruct, bDisableTime); } } - else - if (LStr === "{") { - var attrs = WorkStruct[format]; - if (!attrs) { - attrs = GetAttributes(GetMiddleString(format)); - WorkStruct[format] = attrs; - } - ret = {}; - for (var i = 0; i < attrs.length; i++) { - var type = attrs[i]; - ret[type.Key] = Read(buf, type.Value, undefined, WorkStruct, bDisableTime); - } - } - else { - throw "Bad read type params: " + format; - } + } else if (LStr === "{") { + var attrs = WorkStruct[format]; + if (!attrs) { + attrs = GetAttributes(GetMiddleString(format)); + WorkStruct[format] = attrs; + } + ret = {}; + for (var i = 0; i < attrs.length; i++) { + var type = attrs[i]; + ret[type.Key] = Read(buf, type.Value, undefined, WorkStruct, bDisableTime); + } + } else { + throw "Bad read type params: " + format; } - } + } } return ret; };