node-query/node_modules/mysql2/lib/commands/client_handshake.js

99 lines
3.4 KiB
JavaScript

var Command = require('./command');
var Packets = require('../packets/index.js');
var util = require('util');
var ClientConstants = require('../constants/client');
function ClientHandshake(clientFlags)
{
this.handshake = null;
this.clientFlags = clientFlags;
Command.call(this);
}
util.inherits(ClientHandshake, Command);
ClientHandshake.prototype.start = function() {
return ClientHandshake.prototype.handshakeInit;
};
ClientHandshake.prototype.sendSSLRequest = function(connection) {
var sslRequest = new Packets.SSLRequest(this.clientFlags);
connection.writePacket(sslRequest.toPacket());
};
function flagNames(flags) {
var res = [];
for (var c in ClientConstants) {
if (flags & ClientConstants[c])
res.push(c.replace(/_/g, ' ').toLowerCase());
}
return res;
}
ClientHandshake.prototype.sendCredentials = function(connection, packetIndex) {
if (connection.config.debug) {
console.log('Sending handshake packet: flags:%d=(%s)', this.clientFlags,
flagNames(this.clientFlags).join(', '));
}
var handshakeResponse = new Packets.HandshakeResponse({
flags : this.clientFlags,
password: connection.config.password,
user : connection.config.user,
database: connection.config.database,
charsetNumber: connection.config.charsetNumber,
authPluginData1: this.handshake.authPluginData1,
authPluginData2: this.handshake.authPluginData2,
compress: connection.config.compress
});
connection.writePacket(handshakeResponse.toPacket());
};
ClientHandshake.prototype.handshakeInit = function(helloPacket, connection) {
var command = this;
this.on('connect', function(connArgs) {
connection.emit('connect', connArgs);
});
this.on('error', function(err) {
connection._protocolError = err;
connection.emit('error', err);
});
this.handshake = Packets.Handshake.fromPacket(helloPacket);
if (connection.config.debug) {
console.log('Server hello packet: capability flags:%d=(%s)', this.handshake.capabilityFlags,
flagNames(this.handshake.capabilityFlags).join(', '));
}
connection.serverCapabilityFlags = this.handshake.capabilityFlags;
connection.connectionId = this.handshake.connectionId;
var serverSSLSupport = this.handshake.capabilityFlags & ClientConstants.SSL;
// use compression only if requested by client and supported by server
connection.config.compress = connection.config.compress && (this.handshake.capabilityFlags & ClientConstants.COMPRESS);
this.clientFlags = this.clientFlags | connection.config.compress;
if (connection.config.ssl) {
if (!serverSSLSupport)
throw new Error('Server does not support secure connnection');
// send ssl upgrade request and immediately upgrade connection to secure
this.clientFlags |= ClientConstants.SSL;
this.sendSSLRequest(connection);
connection.startTLS(function() {
// after connection is secure
command.sendCredentials(connection, 2);
});
} else {
this.sendCredentials(connection, 1);
}
return ClientHandshake.prototype.handshakeResult;
};
ClientHandshake.prototype.handshakeResult = function(okPacket, connection) {
// error is already checked in base class. Done auth.
connection.authorized = true;
if (connection.config.compress)
connection.packetParser.onPacket = connection.handleCompressedPacket.bind(connection);
// TODO any useful information in ok packet to pass as argument?
connection.emit('connect', true);
return null;
};
module.exports = ClientHandshake;