Skip to content

Commit

Permalink
Merge pull request #400 from mattreim/formatting
Browse files Browse the repository at this point in the history 
Formatting main.js and lib
  • Loading branch information
@Jey-Cee
Jey-Cee authored Oct 6, 2024
2 parents 0c279f1 + 1f184fd commit c1aec32
Show file tree
Hide file tree
Showing 3 changed files with 195 additions and 213 deletions.
146 changes: 73 additions & 73 deletions lib/discovery.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,11 @@
Thanks to Jonas Hermsmeier who has written the original code, which can be found here: https://github.com/jhermsmeier/node-net-ssdp
*/

var os = require( 'os' )
var dgram = require( 'dgram' )
var net = require( 'net' )
var Emitter = require( 'events' )
var SSDP = require( './ssdp' )
var os = require('os')
var dgram = require('dgram')
var net = require('net')
var Emitter = require('events')
var SSDP = require('./ssdp')

class Discovery extends Emitter {

Expand All @@ -18,11 +18,11 @@ class Discovery extends Emitter {
* @param {Number} [options.multicastTTL]
* @returns {Discovery}
*/
constructor( options ) {
constructor(options) {

super()

options = Object.assign( {}, options )
options = Object.assign({}, options)

this.reuseAddr = options.reuseAddr !== false

Expand All @@ -31,198 +31,198 @@ class Discovery extends Emitter {

}

_createSockets( interfaces ) {
_createSockets(interfaces) {

interfaces = interfaces || os.networkInterfaces()

Object.keys( interfaces ).forEach(( name ) => {
interfaces[ name ].forEach(( info ) => {
Object.keys(interfaces).forEach((name) => {
interfaces[name].forEach((info) => {

// Skip internal interfaces
if( info.internal ) { return }
if (info.internal) { return }

var iface = new Interface( name, info )
var iface = new Interface(name, info)

iface.socket = dgram.createSocket({
type: iface.family === 'IPv6' ? 'udp6' : 'udp4',
ipv6Only: iface.family === 'IPv6',
reuseAddr: this.reuseAddr,
})

this.interfaces.set( iface.address, iface )
this.interfaces.set(iface.address, iface)

})
})

}

_handleMessage( msg, rinfo, iface ) {
_handleMessage(msg, rinfo, iface) {

var message = null

if( msg.indexOf( 'HTTP' ) === 0 ) {
message = SSDP.parseResponse( msg )
if (msg.indexOf('HTTP') === 0) {
message = SSDP.parseResponse(msg)
//console.log(message);
this.emit( 'message', message, rinfo, iface )
this.emit('message', message, rinfo, iface)
} else {
message = SSDP.parseRequest( msg )
if( message.method === 'M-SEARCH' ) {
this.emit( 'search', message, rinfo, iface )
message = SSDP.parseRequest(msg)
if (message.method === 'M-SEARCH') {
this.emit('search', message, rinfo, iface)
// TODO/FIXME: Determine if the search came from ourselves
this._handleSearch( message, rinfo, iface )
} else if( message.method === 'NOTIFY' ) {
this.emit( 'notify', message, rinfo, iface )
this._handleSearch(message, rinfo, iface)
} else if (message.method === 'NOTIFY') {
this.emit('notify', message, rinfo, iface)
}
}

}

_handleSearch( msg, rinfo, iface ) {
_handleSearch(msg, rinfo, iface) {

var services = /^ssdp:all$/i.test( msg.headers.st ) ?
var services = /^ssdp:all$/i.test(msg.headers.st) ?
this.services.slice() :
this.services.filter(( service ) => {
this.services.filter((service) => {
return service.type === msg.headers.st
})

// Calculate a random timeout up to MX until transmitting response
var mx = parseInt( msg.headers.mx, 10 ) || SSDP.DEFAULT_MX
var mx = parseInt(msg.headers.mx, 10) || SSDP.DEFAULT_MX
var delay = mx * Math.random() * 1000

setTimeout(() => {
var date = new Date().toGMTString()
services.forEach(( service ) => {
services.forEach((service) => {
this.respond({
'date': date,
'cache-control': `max-age=${service.maxAge}`,
// 'ext': '',
'location': iface.family === 'IPv6' ? service.location.ipv6 : service.location.ipv4,
'st': service.type,
'usn': service.usn,
}, rinfo, iface )
}, rinfo, iface)
})
}, delay )
}, delay)

}

respond( options, rinfo, iface ) {
respond(options, rinfo, iface) {

var message = `HTTP/1.1 200 OK\r\n`

Object.keys( options ).forEach(( field ) => {
message += `${ field.toUpperCase() }: ${ options[field] }\r\n`
Object.keys(options).forEach((field) => {
message += `${field.toUpperCase()}: ${options[field]}\r\n`
})

message += '\r\n'

var buffer = Buffer.from( message )
var buffer = Buffer.from(message)
var length = buffer.length
var offset = 0

iface.socket.send( buffer, offset, length, rinfo.port, rinfo.address )
iface.socket.send(buffer, offset, length, rinfo.port, rinfo.address)

}

search( options, callback ) {
search(options, callback) {

var headers = Object.assign({
'host': '',
'man': '"ssdp:discover"',
'mx': '1',
'st': 'ssdp:all'
}, options )
}, options)

for( var [ address, iface ] of this.interfaces ) {
for (var [address, iface] of this.interfaces) {

let message = `M-SEARCH * HTTP/1.1\r\n`
let host = iface.family === 'IPv6' ? SSDP.IPv6 : SSDP.IPv4

headers.host = iface.family === 'IPv6' ?
`[${host}]:${SSDP.PORT}` : `${host}:${SSDP.PORT}`

for( var field in headers ) {
for (var field in headers) {
message += field.toUpperCase() + ': ' + headers[field] + '\r\n'
}

message += '\r\n'

var buffer = Buffer.from( message )
var buffer = Buffer.from(message)
var length = buffer.length
var offset = 0

iface.socket.send( buffer, offset, length, SSDP.PORT, host )
iface.socket.send(buffer, offset, length, SSDP.PORT, host)

}

if( typeof callback === 'function' ) {
if (typeof callback === 'function') {
setTimeout(() => {
callback.call( this, null, [] )
}, 5000 )
callback.call(this, null, [])
}, 5000)
}

}

announce( options, callback ) {
if( typeof callback === 'function' ) {
announce(options, callback) {
if (typeof callback === 'function') {
process.nextTick(() => {
callback.call( this, new Error( 'Not implemented' ) )
callback.call(this, new Error('Not implemented'))
})
}
}

publish( service, callback ) {
this.services.push( service )
this.announce( service, callback )
publish(service, callback) {
this.services.push(service)
this.announce(service, callback)
}

listen( interfaces, callback ) {
listen(interfaces, callback) {

if( typeof interfaces === 'function' ) {
if (typeof interfaces === 'function') {
callback = interfaces
interfaces = null
}

this._createSockets( interfaces )
this._createSockets(interfaces)

var interfaceCount = this.interfaces.size
var socketsBound = 0

this.interfaces.forEach(( iface, address ) => {
this.interfaces.forEach((iface, address) => {

var socket = iface.socket

socket.on( 'error', ( error ) => { this.emit( 'warning', error, socket ) })
socket.on( 'message', ( msg, rinfo ) => { this._handleMessage( msg, rinfo, iface ) })
socket.on( 'listening', () => { this.emit( 'listening', socket ) })
socket.on('error', (error) => { this.emit('warning', error, socket) })
socket.on('message', (msg, rinfo) => { this._handleMessage(msg, rinfo, iface) })
socket.on('listening', () => { this.emit('listening', socket) })

var address = iface.family === 'IPv6' ?
SSDP.IPv6 : SSDP.IPv4

socket.bind( SSDP.PORT, () => {
socket.bind(SSDP.PORT, () => {

var addr = socket.address()

try {
socket.setMulticastTTL( SSDP.MULTICAST_TTL )
socket.setMulticastLoopback( true )
socket.setBroadcast( true )
if( iface.family === 'IPv6' ) {
socket.addMembership( SSDP.IPv6, iface.address )
socket.setMulticastTTL(SSDP.MULTICAST_TTL)
socket.setMulticastLoopback(true)
socket.setBroadcast(true)
if (iface.family === 'IPv6') {
socket.addMembership(SSDP.IPv6, iface.address)
} else {
socket.addMembership( SSDP.IPv4, iface.address )
socket.addMembership(SSDP.IPv4, iface.address)
}
} catch( error ) {
this.emit( 'warning', error, iface )
this.interfaces.delete( iface.address )
} catch (error) {
this.emit('warning', error, iface)
this.interfaces.delete(iface.address)
iface.socket.removeAllListeners()
iface.socket.close()
iface.socket = null
}

if( ( socketsBound += 1 ) === interfaceCount ) {
if ((socketsBound += 1) === interfaceCount) {
var error = !this.interfaces.size ?
new Error( 'No interfaces bound' ) : null
callback.call( this, error )
new Error('No interfaces bound') : null
callback.call(this, error)
}

})
Expand All @@ -232,8 +232,8 @@ class Discovery extends Emitter {
}

close() {
this.interfaces.forEach(( iface, address ) => {
if( iface.socket ) {
this.interfaces.forEach((iface, address) => {
if (iface.socket) {
iface.socket.close(() => {
iface.socket.removeAllListeners()
iface.socket = null
Expand All @@ -245,7 +245,7 @@ class Discovery extends Emitter {
}

class Interface {
constructor( name, options ) {
constructor(name, options) {
this.name = name || ''
this.address = options.address || ''
this.netmask = options.netmask || ''
Expand Down
Loading

0 comments on commit c1aec32

Please sign in to comment.