Skip to content

Commit

Permalink
Fix UA string
Browse files Browse the repository at this point in the history
Co-authored-by: Fedor Indutny <79877362+indutny-signal@users.noreply.github.com>
  • Loading branch information
automated-signal and indutny-signal committed Jun 21, 2022
1 parent 60d23da commit 9816ed8
Show file tree
Hide file tree
Showing 3 changed files with 28 additions and 10 deletions.
6 changes: 3 additions & 3 deletions ts/test-node/logging/uploadDebugLogs_test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ describe('upload', () => {
beforeEach(function beforeEach() {
this.sandbox = sinon.createSandbox();

this.sandbox.stub(process, 'platform').get(() => 'linux');
this.sandbox.stub(process, 'platform').get(() => 'freebsd');

this.fakeGet = this.sandbox.stub(got, 'get');
this.fakePost = this.sandbox.stub(got, 'post');
Expand Down Expand Up @@ -48,15 +48,15 @@ describe('upload', () => {
sinon.assert.calledOnce(this.fakeGet);
sinon.assert.calledWith(this.fakeGet, 'https://debuglogs.org/', {
responseType: 'json',
headers: { 'User-Agent': 'Signal-Desktop/1.2.3 Linux' },
headers: { 'User-Agent': 'Signal-Desktop/1.2.3' },
timeout: { request: durations.MINUTE },
});

const compressedContent = await gzip('hello world');

sinon.assert.calledOnce(this.fakePost);
sinon.assert.calledWith(this.fakePost, 'https://example.com/fake-upload', {
headers: { 'User-Agent': 'Signal-Desktop/1.2.3 Linux' },
headers: { 'User-Agent': 'Signal-Desktop/1.2.3' },
timeout: { request: durations.MINUTE },
body: sinon.match((value: unknown) => {
if (!(value instanceof FormData)) {
Expand Down
17 changes: 13 additions & 4 deletions ts/test-node/util/getUserAgent_test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -17,21 +17,30 @@ describe('getUserAgent', () => {

it('returns the right User-Agent on Windows', function test() {
this.sandbox.stub(process, 'platform').get(() => 'win32');
assert.strictEqual(getUserAgent('1.2.3'), 'Signal-Desktop/1.2.3 Windows');
assert.strictEqual(
getUserAgent('1.2.3', '10.0.22000'),
'Signal-Desktop/1.2.3 Windows 10.0.22000'
);
});

it('returns the right User-Agent on macOS', function test() {
this.sandbox.stub(process, 'platform').get(() => 'darwin');
assert.strictEqual(getUserAgent('1.2.3'), 'Signal-Desktop/1.2.3 macOS');
assert.strictEqual(
getUserAgent('1.2.3', '21.5.0'),
'Signal-Desktop/1.2.3 macOS 21.5.0'
);
});

it('returns the right User-Agent on Linux', function test() {
this.sandbox.stub(process, 'platform').get(() => 'linux');
assert.strictEqual(getUserAgent('1.2.3'), 'Signal-Desktop/1.2.3 Linux');
assert.strictEqual(
getUserAgent('1.2.3', '20.04'),
'Signal-Desktop/1.2.3 Linux 20.04'
);
});

it('omits the platform on unsupported platforms', function test() {
this.sandbox.stub(process, 'platform').get(() => 'freebsd');
assert.strictEqual(getUserAgent('1.2.3'), 'Signal-Desktop/1.2.3');
assert.strictEqual(getUserAgent('1.2.3', '13.1'), 'Signal-Desktop/1.2.3');
});
});
15 changes: 12 additions & 3 deletions ts/util/getUserAgent.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
// Copyright 2020 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only

import os from 'os';

import { getOwn } from './getOwn';

const PLATFORM_STRINGS: { [platform: string]: string } = {
Expand All @@ -9,12 +11,19 @@ const PLATFORM_STRINGS: { [platform: string]: string } = {
linux: 'Linux',
};

export function getUserAgent(appVersion: string): string {
export function getUserAgent(
appVersion: string,
release = os.release()
): string {
// `process.platform` could be missing if someone figures out how to compile Signal on
// an unsupported OS and forgets to update this file. We'd rather send nothing than
// crash.
const platformString = getOwn(PLATFORM_STRINGS, process.platform);
const platformStringWithSpace = platformString ? ` ${platformString}` : '';

return `Signal-Desktop/${appVersion}${platformStringWithSpace}`;
let result = `Signal-Desktop/${appVersion}`;
if (platformString) {
result += ` ${platformString} ${release}`;
}

return result;
}

0 comments on commit 9816ed8

Please sign in to comment.