diff --git a/src/HMRServer.js b/src/HMRServer.js index 45176385e90..286d7bb5014 100644 --- a/src/HMRServer.js +++ b/src/HMRServer.js @@ -17,7 +17,17 @@ class HMRServer { this.server = https.createServer(await getCertificate(options.https)); } - this.wss = new WebSocket.Server({server: this.server}); + let websocketOptions = { + server: this.server + }; + + if (options.hmrHostname) { + websocketOptions.origin = `${options.https ? 'https' : 'http'}://${ + options.hmrHostname + }`; + } + + this.wss = new WebSocket.Server(websocketOptions); this.server.listen(options.hmrPort, resolve); });