Skip to content

Commit

Permalink
[misc] restrict capabilities to server capabilities
Browse files Browse the repository at this point in the history
  • Loading branch information
rusher committed Oct 12, 2023
1 parent f2fff31 commit e8b4b56
Showing 1 changed file with 6 additions and 10 deletions.
16 changes: 6 additions & 10 deletions lib/cmd/handshake/client-capabilities.js
Original file line number Diff line number Diff line change
Expand Up @@ -24,11 +24,8 @@ module.exports.init = function (opts, info) {
Capabilities.SESSION_TRACK |
Capabilities.CONNECT_ATTRS |
Capabilities.PLUGIN_AUTH_LENENC_CLIENT_DATA |
Capabilities.MARIADB_CLIENT_EXTENDED_TYPE_INFO;

if (info.serverCapabilities & Capabilities.PLUGIN_AUTH) {
capabilities |= Capabilities.PLUGIN_AUTH;
}
Capabilities.MARIADB_CLIENT_EXTENDED_TYPE_INFO |
Capabilities.PLUGIN_AUTH;

if (opts.foundRows) {
capabilities |= Capabilities.FOUND_ROWS;
Expand Down Expand Up @@ -65,14 +62,13 @@ module.exports.init = function (opts, info) {
}
}

if (opts.bulk) {
if (info.serverCapabilities & Capabilities.MARIADB_CLIENT_STMT_BULK_OPERATIONS) {
capabilities |= Capabilities.MARIADB_CLIENT_STMT_BULK_OPERATIONS;
}
if (opts.bulk && info.serverCapabilities & Capabilities.MARIADB_CLIENT_STMT_BULK_OPERATIONS) {
capabilities |= Capabilities.MARIADB_CLIENT_STMT_BULK_OPERATIONS;
}

if (opts.permitConnectionWhenExpired) {
capabilities |= Capabilities.CAN_HANDLE_EXPIRED_PASSWORDS;
}
info.clientCapabilities = capabilities;

info.clientCapabilities = capabilities & info.serverCapabilities;
};

0 comments on commit e8b4b56

Please sign in to comment.