153 lines
4.5 KiB
JavaScript
153 lines
4.5 KiB
JavaScript
// This file was modified by Oracle on June 1, 2021.
|
|
// A utility method was introduced to generate an Error instance from a
|
|
// binary server packet.
|
|
// Modifications copyright (c) 2021, Oracle and/or its affiliates.
|
|
|
|
// This file was modified by Oracle on September 21, 2021.
|
|
// The new AuthNextFactor packet is now available.
|
|
// Modifications copyright (c) 2021, Oracle and/or its affiliates.
|
|
|
|
'use strict';
|
|
|
|
const process = require('process');
|
|
|
|
const AuthNextFactor = require('./auth_next_factor');
|
|
const AuthSwitchRequest = require('./auth_switch_request');
|
|
const AuthSwitchRequestMoreData = require('./auth_switch_request_more_data');
|
|
const AuthSwitchResponse = require('./auth_switch_response');
|
|
const BinaryRow = require('./binary_row');
|
|
const BinlogDump = require('./binlog_dump');
|
|
const ChangeUser = require('./change_user');
|
|
const CloseStatement = require('./close_statement');
|
|
const ColumnDefinition = require('./column_definition');
|
|
const Execute = require('./execute');
|
|
const Handshake = require('./handshake');
|
|
const HandshakeResponse = require('./handshake_response');
|
|
const PrepareStatement = require('./prepare_statement');
|
|
const PreparedStatementHeader = require('./prepared_statement_header');
|
|
const Query = require('./query');
|
|
const RegisterSlave = require('./register_slave');
|
|
const ResultSetHeader = require('./resultset_header');
|
|
const SSLRequest = require('./ssl_request');
|
|
const TextRow = require('./text_row');
|
|
|
|
const ctorMap = {
|
|
AuthNextFactor,
|
|
AuthSwitchRequest,
|
|
AuthSwitchRequestMoreData,
|
|
AuthSwitchResponse,
|
|
BinaryRow,
|
|
BinlogDump,
|
|
ChangeUser,
|
|
CloseStatement,
|
|
ColumnDefinition,
|
|
Execute,
|
|
Handshake,
|
|
HandshakeResponse,
|
|
PrepareStatement,
|
|
PreparedStatementHeader,
|
|
Query,
|
|
RegisterSlave,
|
|
ResultSetHeader,
|
|
SSLRequest,
|
|
TextRow,
|
|
};
|
|
Object.entries(ctorMap).forEach(([name, ctor]) => {
|
|
module.exports[name] = ctor;
|
|
// monkey-patch it to include name if debug is on
|
|
if (process.env.NODE_DEBUG) {
|
|
if (ctor.prototype.toPacket) {
|
|
const old = ctor.prototype.toPacket;
|
|
ctor.prototype.toPacket = function () {
|
|
const p = old.call(this);
|
|
p._name = name;
|
|
return p;
|
|
};
|
|
}
|
|
}
|
|
});
|
|
|
|
// simple packets:
|
|
const Packet = require('./packet');
|
|
exports.Packet = Packet;
|
|
|
|
class OK {
|
|
static toPacket(args, encoding) {
|
|
args = args || {};
|
|
const affectedRows = args.affectedRows || 0;
|
|
const insertId = args.insertId || 0;
|
|
const serverStatus = args.serverStatus || 0;
|
|
const warningCount = args.warningCount || 0;
|
|
const message = args.message || '';
|
|
|
|
let length = 9 + Packet.lengthCodedNumberLength(affectedRows);
|
|
length += Packet.lengthCodedNumberLength(insertId);
|
|
|
|
const buffer = Buffer.allocUnsafe(length);
|
|
const packet = new Packet(0, buffer, 0, length);
|
|
packet.offset = 4;
|
|
packet.writeInt8(0);
|
|
packet.writeLengthCodedNumber(affectedRows);
|
|
packet.writeLengthCodedNumber(insertId);
|
|
packet.writeInt16(serverStatus);
|
|
packet.writeInt16(warningCount);
|
|
packet.writeString(message, encoding);
|
|
packet._name = 'OK';
|
|
return packet;
|
|
}
|
|
}
|
|
|
|
exports.OK = OK;
|
|
|
|
// warnings, statusFlags
|
|
class EOF {
|
|
static toPacket(warnings, statusFlags) {
|
|
if (typeof warnings === 'undefined') {
|
|
warnings = 0;
|
|
}
|
|
if (typeof statusFlags === 'undefined') {
|
|
statusFlags = 0;
|
|
}
|
|
const packet = new Packet(0, Buffer.allocUnsafe(9), 0, 9);
|
|
packet.offset = 4;
|
|
packet.writeInt8(0xfe);
|
|
packet.writeInt16(warnings);
|
|
packet.writeInt16(statusFlags);
|
|
packet._name = 'EOF';
|
|
return packet;
|
|
}
|
|
}
|
|
|
|
exports.EOF = EOF;
|
|
|
|
class Error {
|
|
static toPacket(args, encoding) {
|
|
const length = 13 + Buffer.byteLength(args.message, 'utf8');
|
|
const packet = new Packet(0, Buffer.allocUnsafe(length), 0, length);
|
|
packet.offset = 4;
|
|
packet.writeInt8(0xff);
|
|
packet.writeInt16(args.code);
|
|
// TODO: sql state parameter
|
|
packet.writeString('#_____', encoding);
|
|
packet.writeString(args.message, encoding);
|
|
packet._name = 'Error';
|
|
return packet;
|
|
}
|
|
|
|
static fromPacket(packet) {
|
|
packet.readInt8(); // marker
|
|
const code = packet.readInt16();
|
|
packet.readString(1, 'ascii'); // sql state marker
|
|
// The SQL state of the ERR_Packet which is always 5 bytes long.
|
|
// https://dev.mysql.com/doc/dev/mysql-server/8.0.11/page_protocol_basic_dt_strings.html#sect_protocol_basic_dt_string_fix
|
|
packet.readString(5, 'ascii'); // sql state (ignore for now)
|
|
const message = packet.readNullTerminatedString('utf8');
|
|
const error = new Error();
|
|
error.message = message;
|
|
error.code = code;
|
|
return error;
|
|
}
|
|
}
|
|
|
|
exports.Error = Error;
|