Fix origin check

Author: Ives van Hoorne

packages/app/src/sandbox/index.js

@@ -7,7 +7,7 @@ import { getModulePath } from 'common/sandbox/modules';
 import { generateFileFromSandbox } from 'common/templates/configuration/package-json';
 
 import setupHistoryListeners from './url-listeners';
-import compile, { getCurrentManager } from './compile';
+import compile from './compile';
 import setupConsole from './console';
 import transformJSON from './console/transform-json';
 

packages/codesandbox-api/src/dispatcher/index.ts

@@ -68,10 +68,13 @@ function notifyFrames(message: Object) {
 function eventListener(e: MessageEvent) {
   const { data } = e;
 
+  const testedHost = host || 'https://codesandbox.io';
+  const origin = testedHost.replace('https://', '').replace('http://', '');
+
   if (
     data.codesandbox &&
-    (e.origin.endsWith('.codesandbox.io') ||
-      e.origin.endsWith('/codesandbox.io') ||
+    (e.origin.endsWith(`.${origin}`) ||
+      e.origin.endsWith(`/${origin}`) ||
       process.env.NODE_ENV !== 'production')
   ) {
     notifyListeners(data, e.source);