intval3/src/wifi/index.ts

'use strict'

const networkPattern = /network[\s\S]*?=[\s\S]*?{([\s\S]*?)}/gi
const quoteRe = new RegExp('"', 'g')

const filePath = '/etc/wpa_supplicant/wpa_supplicant.conf'
const reconfigure = '/sbin/wpa_cli reconfigure'
const refresh = 'ip link set wlan0 down && ip link set wlan0 up'
const iwlist = '/sbin/iwlist wlan0 scanning | grep "ESSID:"'
const iwgetid = '/sbin/iwgetid'

const log = require('../log')('wifi')
const exec = require('child_process').exec
const fs = require('fs')

let _entry = null
let _ssid = null
let _cb = null

/** Class representing the wifi features */
class Wifi {
	constructor () {

	}
	/**
	* List available wifi access points
	*
	* @param {function} 	callback 	Function which gets invoked after list is returned
	*/
	list (callback) {
		exec(iwlist, (err, stdout, stderr) => {
			if (err) {
				console.error(err)
				return callback(err)
			}
			const limit = 20;
			const lines = stdout.split('\n')
			let output = []
			let line
			let i = 0
			for (let l of lines) {
				line = l.replace('ESSID:',  '').trim()
				if (line !== '""' && i < limit) {
					line = line.replace(quoteRe, '')
					output.push(line)
				}
				i++
			}
			output = output.filter(ap => {
				if (ap !== '') return ap
			})
			return callback(null, output)
		})
	}
	/**
	* (internal function) Invoked after config file is read, 
	* then invokes file write on the config file
	*
	* @param {object} 	err 		(optional) Error object only present if problem reading config file
	* @param {string} 	data 		Contents of the config file
	*/
	_readConfigCb (err, data) {
		let parsed
		let current
		if (err) {
			console.error(err)
			return _cb(err)
		}
		parsed = this._parseConfig(data)
		current = parsed.find(network => {
			return network.ssid === _ssid
		})
		if (typeof current !== 'undefined') {
			data = data.replace(current.raw, _entry)
		} else {
			data += '\n\n' + _entry
		}
		_entry = null
		fs.writeFile(filePath, data, 'utf8', this._writeConfigCb.bind(this))
	}
	/**
	* (internal function) Invoked after config file is written, 
	* then executes reconfiguration command
	*
	* @param {object} 	err 		(optional) Error object only present if problem writing config file
	*/
	_writeConfigCb (err) {
		if (err) {
			console.error(err)
			return _cb(err)
		}
		exec(reconfigure, this._reconfigureCb.bind(this))
	}
	/**
	* (internal function) Invoked after reconfiguration command is complete
	*
	* @param {object} 	err 		(optional) Error object only present if configuration command fails
	* @param {string} 	stdout 		Standard output from reconfiguration command
	* @param {string} 	stderr 		Error output from command if fails
	*/
	_reconfigureCb (err, stdout, stderr) {
		if (err) {
			console.error(err)
			return _cb(err)
		}
		console.log('Wifi reconfigured')
		exec(refresh, this._refreshCb.bind(this))
	}
	/**
	* (internal function) Invoked after wifi refresh command is complete
	*
	* @param {object} 	err 		(optional) Error object only present if refresh command fails
	* @param {string} 	stdout 		Standard output from refresh command
	* @param {string} 	stderr 		Error output from command if fails
	*/
	_refreshCb (err, stdout, stderr) {
		if (err) {
			console.error(err)
			return _cb(err)
		}
		console.log('Wifi refreshed')
		_cb(null, { ssid : _ssid })
		_cb = () => {}
	}
	_parseConfig (str) {
		const networks = []
		const lines = str.split('\n')
		let network = {}
		for (let line of lines) {
			if (line.substring(0, 9) === 'network={') {
				network = {}
				network.raw = line
			} else if (network.raw && line.indexOf('ssid=') !== -1) {
				network.ssid = line.replace('ssid=', '').trim().replace(quoteRe, '')
				if (network.raw) {
					network.raw += '\n' + line
				}
			} else if (network.raw && line.substring(0, 1) === '}') {
				network.raw += '\n' + line
				networks.push(network)
				network = {}
			} else if (network.raw) {
				network.raw += '\n' + line
			}
		}
		return networks
	}
	/**
	 * Create sanitized wpa_supplicant.conf stanza for
	 * configuring wifi without storing plaintext passwords
	 * @example
	 * network={
	 *	 	ssid="YOUR_SSID"
	 *		#psk="YOUR_PASSWORD"
	 *		psk=6a24edf1592aec4465271b7dcd204601b6e78df3186ce1a62a31f40ae9630702
	 *	}
	 *
	 * @param {string} 	ssid 		SSID of wifi network
	 * @param {string}  pwd 		Plaintext passphrase of wifi network
	 * @param {function} callback 	Function called after psk hash is generated
	 */
	createPSK (ssid, pwd, callback) {
		const cmd = `wpa_passphrase '${ssid.replace(/'/g, `'\\''`)}' '${pwd.replace(/'/g, `'\\''`)}' | grep "psk="`
		let lines
		let hash
		let plaintext
		exec(cmd, (err, stdout, stderr) => {
			if (err) {
				return callback(err)
			}
			lines = stdout.replace('#psk=', '').split('psk=')
			hash = lines[1]
			plaintext = lines[0]
			callback(null, hash.trim(), plaintext.trim())
		})
	}
	/**
	* Function which initializes the processes for adding a wifi access point authentication
	*
	* @param {string} 	ssid 		SSID of network to configure
	* @param {string} 	pwd 		Password of access point, plaintext to be masked
	* @param {string} 	hash 		Password/SSID of access point, securely hashed
	* @param {function} callback 	Function invoked after process is complete, or fails
	*/
	setNetwork (ssid, pwd, hash, callback) {
		let masked = pwd.split('').map(char => { return char !== '"' ? '*' : '"' }).join('')
		_entry = `network={\n\tssid="${ssid}"\n\t#psk=${masked}\n\tpsk=${hash}\n}\n`
		_cb = callback
		_ssid = ssid
		fs.readFile(filePath, 'utf8', this._readConfigCb.bind(this))
	}
	/**
	* Executes command which gets the currently connected network
	*
	* @param {function} 	callback 	Function which is invoked after command is completed
	*/
	getNetwork (callback) {
		let output
		exec(iwgetid, (err, stdout, stderr) => {
			if (err) {
				return callback(err)
			}
			output = stdout.split('ESSID:')[1].replace(quoteRe, '').trim()
			callback(null, output)
		})
	}
}

module.exports = new Wifi()