Andrew Eisenberg
2020-07-17 08:15:37 -07:00
parent 546ec2eb1c
commit 8093d9a529
2 changed files with 14 additions and 7 deletions

View File

@@ -31,10 +31,14 @@ export function Compare(_: {}): JSX.Element {
useEffect(() => {
window.addEventListener('message', (evt: MessageEvent) => {
const msg: ToCompareViewMessage = evt.data;
switch (msg.t) {
case 'setComparisons':
setComparison(msg);
if (evt.origin === window.origin) {
const msg: ToCompareViewMessage = evt.data;
switch (msg.t) {
case 'setComparisons':
setComparison(msg);
}
} else {
console.error(`Invalid event origin ${evt.origin}`);
}
});
});
@@ -60,8 +64,8 @@ export function Compare(_: {}): JSX.Element {
{hasRows ? (
<CompareTable comparison={comparison}></CompareTable>
) : (
<div className="vscode-codeql__compare-message">{message}</div>
)}
<div className="vscode-codeql__compare-message">{message}</div>
)}
</>
);
} catch (err) {

View File

@@ -275,7 +275,10 @@ class App extends React.Component<{}, ResultsViewState> {
componentDidMount(): void {
this.vscodeMessageHandler = (evt) =>
this.handleMessage(evt.data as IntoResultsViewMsg);
evt.origin === window.origin
? this.handleMessage(evt.data as IntoResultsViewMsg)
: console.error(`Invalid event origin ${evt.origin}`);
window.addEventListener('message', this.vscodeMessageHandler);
}