fix packet

tediousjs · Oct 29, 2024 · 8db7c47 · 8db7c47
1 parent 809227a
commit 8db7c47
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 9 deletions.
diff --git a/src/connection.ts b/src/connection.ts
@@ -873,6 +873,7 @@ export interface ConnectionOptions {
 interface RoutingData {
   server: string;
   port: number;
+  login7server: string;
 }
 
 /**
@@ -2484,7 +2485,7 @@ class Connection extends EventEmitter {
     }
 
     payload.hostname = this.config.options.workstationId || os.hostname();
-    payload.serverName = this.routingData ? this.routingData.server : this.config.server;
+    payload.serverName = this.routingData ? this.routingData.login7server : this.config.server;
     payload.appName = this.config.options.appName || 'Tedious';
     payload.libraryName = libraryName;
     payload.language = this.config.options.language;

diff --git a/src/login7-payload.ts b/src/login7-payload.ts
@@ -251,15 +251,12 @@ class Login7Payload {
     }
 
     // (ibUnused / ibExtension): 2-byte
-    offset = fixedData.writeUInt16LE(dataOffset, offset);
+    const extensionOffsetHeaderOffset = offset;
+    offset = fixedData.writeUInt16LE(0, offset);
+
 
     // (cchUnused / cbExtension): 2-byte
-    const extensions = this.buildFeatureExt();
     offset = fixedData.writeUInt16LE(4, offset);
-    const extensionOffset = Buffer.alloc(4);
-    extensionOffset.writeUInt32LE(dataOffset += 4, 0);
-    dataOffset += extensions.length;
-    buffers.push(extensionOffset, extensions);
 
     // ibCltIntName: 2-byte
     offset = fixedData.writeUInt16LE(dataOffset, offset);
@@ -365,6 +362,13 @@ class Login7Payload {
       fixedData.writeUInt32LE(0, offset);
     }
 
+    fixedData.writeUInt16LE(dataOffset, extensionOffsetHeaderOffset);
+
+    const extensions = this.buildFeatureExt();
+    const extensionOffset = Buffer.alloc(4);
+    extensionOffset.writeUInt32LE(dataOffset + 4, 0);
+    buffers.push(extensionOffset, extensions);
+
     const data = Buffer.concat(buffers);
     data.writeUInt32LE(data.length, 0);
     return data;

diff --git a/src/token/handler.ts b/src/token/handler.ts
@@ -246,7 +246,7 @@ export class Login7TokenHandler extends TokenHandler {
   declare connection: Connection;
 
   declare fedAuthInfoToken: FedAuthInfoToken | undefined;
-  declare routingData: { server: string, port: number } | undefined;
+  declare routingData: { server: string, port: number, login7server: string } | undefined;
 
   declare loginAckReceived: boolean;
 
@@ -340,7 +340,7 @@ export class Login7TokenHandler extends TokenHandler {
     const [ server ] = token.newValue.server.split('\\');
 
     this.routingData = {
-      server, port: token.newValue.port
+      server, port: token.newValue.port, login7server: token.newValue.server
     };
   }