LittleChest/dns-packet-polyfill
7
0
Fork 0
forked from LittleChest/dns-packet
Code Pull Requests Activity

check for .. also in encodingLength

Browse Source
This commit is contained in:
Mathias Buus 2021-05-25 13:24:03 +02:00
parent bb04450fe4
commit bd84615252
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
index.js
View File

@ -77,7 +77,7 @@ name.decode = function (buf, offset) {
name.decode.bytes = 0 name.decode.bytes = 0
name.encodingLength = function (n) { name.encodingLength = function (n) {
if (n === '.') return 1 if (n === '.' || n === '..') return 1
return Buffer.byteLength(n.replace(/^\.|\.$/gm, '')) + 2 return Buffer.byteLength(n.replace(/^\.|\.$/gm, '')) + 2
} }
@ -1442,7 +1442,9 @@ exports.CHECKING_DISABLED = 1 << 4
exports.DNSSEC_OK = 1 << 15 exports.DNSSEC_OK = 1 << 15
exports.encode = function (result, buf, offset) { exports.encode = function (result, buf, offset) {
if (!buf) buf = Buffer.allocUnsafe(exports.encodingLength(result)) const allocing = !buf
if (allocing) buf = Buffer.allocUnsafe(exports.encodingLength(result))
if (!offset) offset = 0 if (!offset) offset = 0
const oldOffset = offset const oldOffset = offset
@ -1462,6 +1464,11 @@ exports.encode = function (result, buf, offset) {
exports.encode.bytes = offset - oldOffset exports.encode.bytes = offset - oldOffset
// just a quick sanity check
if (allocing && exports.encode.bytes !== buf.length) {
return buf.slice(0, exports.encode.bytes)
}
return buf return buf
} }