Update for reverse proxies

1 files changed, 2 insertions, 2 deletions

diff --git a/index.js b/index.js
index 737e3d9..cd3209a 100644
--- a/index.js
+++ b/index.js
@@ -72,7 +72,7 @@ const requestListener = function(req, res) {
             try {
                 ip = temp[uid].parts[4].replace(/[^a-z]/gm, "");
             } catch (e) {
-                ip = Buffer.from(req.socket.address().address).toString("base64");
+                ip = Buffer.from(userIp).toString("base64");
             }
             let id = crypto.createHash('sha512').update(crypto.randomBytes(128).toString("base64")).digest('hex');
             streams.push({
@@ -96,7 +96,7 @@ const requestListener = function(req, res) {
 
                 if (selected.length > 0) {
                     if (streams.map((i) => { return i.id; }).includes(stream)) {
-                        if (streams.filter(i => i.id === stream)[0].user === req.headers["user-agent"] && streams.filter(i => i.id === stream)[0].ip === req.socket.address().address && streams.filter(i => i.id === stream)[0].song === song && streams.filter(i => i.id === stream)[0].quality === quality) {
+                        if (streams.filter(i => i.id === stream)[0].user === req.headers["user-agent"] && streams.filter(i => i.id === stream)[0].ip === userIp && streams.filter(i => i.id === stream)[0].song === song && streams.filter(i => i.id === stream)[0].quality === quality) {
                             streams.filter(i => i.id === stream)[0].created = new Date();
                             res.writeHead(200, {
                                 'Content-Type': mime.lookup(selected[0]) || 'application/octet-stream',