Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Replace @pkg/nv with https.request call #27

Merged
merged 3 commits into from
Nov 4, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -123,5 +123,7 @@ out
tags
tags.*

.etag
core.json
schedule.etag
schedule.json
security.etag
security.json
97 changes: 60 additions & 37 deletions is-vulnerable.js
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,21 @@ const { request } = require('https')
const fs = require('fs')
const path = require('path')
const satisfies = require('semver/functions/satisfies')
const nv = require('@pkgjs/nv')

const CORE_RAW_URL = 'https://raw.githubusercontent.com/nodejs/security-wg/main/vuln/core/index.json'

let lastETagValue

const coreLocalFile = path.join(__dirname, 'core.json')
const ETagFile = path.join(__dirname, '.etag')
const STORE = {
security: {
url: 'https://raw.githubusercontent.com/nodejs/security-wg/main/vuln/core/index.json',
jsonFile: path.join(__dirname, 'security.json'),
etagFile: path.join(__dirname, 'security.etag'),
etagValue: ''
},
schedule: {
url: 'https://raw.githubusercontent.com/nodejs/Release/main/schedule.json',
jsonFile: path.join(__dirname, 'schedule.json'),
etagFile: path.join(__dirname, 'schedule.etag'),
etagValue: ''
}
}

async function readLocal (file) {
return require(file)
Expand All @@ -23,26 +30,23 @@ function debug (msg) {
}

function loadETag () {
if (fs.existsSync(ETagFile)) {
debug('Loading local ETag')
lastETagValue = fs.readFileSync(ETagFile).toString()
for (const [key, obj] of Object.entries(STORE)) {
if (fs.existsSync(obj.etagFile)) {
debug(`Loading local ETag for '${key}'`)
obj.etagValue = fs.readFileSync(obj.etagFile).toString()
}
}
}

function updateLastETag (etag) {
lastETagValue = etag
fs.writeFileSync(ETagFile, lastETagValue)
}

async function fetchCoreIndex () {
async function fetchJson (obj) {
await new Promise((resolve) => {
request(CORE_RAW_URL, (res) => {
request(obj.url, (res) => {
if (res.statusCode !== 200) {
console.error(`Request to Github returned http status ${res.statusCode}. Aborting...`)
process.nextTick(() => { process.exit(1) })
}

const fileStream = fs.createWriteStream(coreLocalFile)
const fileStream = fs.createWriteStream(obj.jsonFile)
res.pipe(fileStream)

fileStream.on('finish', () => {
Expand All @@ -51,20 +55,20 @@ async function fetchCoreIndex () {
})

fileStream.on('error', (err) => {
console.error(`Error ${err.message} while writing to '${coreLocalFile}'. Aborting...`)
console.error(`Error ${err.message} while writing to '${obj.jsonFile}'. Aborting...`)
process.nextTick(() => { process.exit(1) })
})
}).on('error', (err) => {
console.error(`Request to Github returned error ${err.message}. Aborting...`)
process.nextTick(() => { process.exit(1) })
}).end()
})
return readLocal(coreLocalFile)
return readLocal(obj.jsonFile)
}

async function getCoreIndex () {
async function getJson (obj) {
return new Promise((resolve) => {
request(CORE_RAW_URL, { method: 'HEAD' }, (res) => {
request(obj.url, { method: 'HEAD' }, (res) => {
if (res.statusCode !== 200) {
console.error(`Request to Github returned http status ${res.statusCode}. Aborting...`)
process.nextTick(() => { process.exit(1) })
Expand All @@ -73,13 +77,14 @@ async function getCoreIndex () {
res.on('data', () => {})

const { etag } = res.headers
if (!lastETagValue || lastETagValue !== etag || !fs.existsSync(coreLocalFile)) {
updateLastETag(etag)
if (!obj.etagValue || obj.eTagValue !== etag || !fs.existsSync(obj.jsonFile)) {
obj.etagValue = etag
fs.writeFileSync(obj.etagFile, etag)
debug('Creating local core.json')
resolve(fetchCoreIndex())
resolve(fetchJson(obj))
} else {
debug(`No updates from upstream. Getting a cached version: ${coreLocalFile}`)
resolve(readLocal(coreLocalFile))
debug(`No updates from upstream. Getting a cached version: ${obj.jsonFile}`)
resolve(readLocal(obj.jsonFile))
}
}).on('error', (err) => {
console.error(`Request to Github returned error ${err.message}. Aborting...`)
Expand All @@ -94,6 +99,7 @@ const checkPlatform = platform => {
throw new Error(`platform ${platform} is not valid. Please use ${availablePlatforms.join(',')}.`)
}
}

const isSystemAffected = (platform, affectedEnvironments) => {
// No platform specified (legacy mode)
if (!platform || !Array.isArray(affectedEnvironments)) {
Expand Down Expand Up @@ -127,15 +133,17 @@ function getVulnerabilityList (currentVersion, data, platform) {

async function cli (currentVersion, platform) {
checkPlatform(platform)

const isEOL = await isNodeEOL(currentVersion)
if (isEOL) {
console.error(danger)
console.error(`${currentVersion} is end-of-life. There are high chances of being vulnerable. Please upgrade it.`)
process.exit(1)
}

const coreIndex = await getCoreIndex()
const list = getVulnerabilityList(currentVersion, coreIndex, platform)
const securityJson = await getJson(STORE.security)
const list = getVulnerabilityList(currentVersion, securityJson, platform)

if (list.length) {
console.error(danger)
console.error(vulnerableWarning + '\n')
Expand All @@ -146,25 +154,40 @@ async function cli (currentVersion, platform) {
}
}

async function getVersionInfo (version) {
const scheduleJson = await getJson(STORE.schedule)

if (scheduleJson[version.toLowerCase()]) {
return scheduleJson[version.toLowerCase()]
}

for (const [key, value] of Object.entries(scheduleJson)) {
if (satisfies(version, key)) {
return value
}
}

return null
}

/**
* @param {string} version
* @returns {Promise<boolean>} true if the version is end-of-life
*/
async function isNodeEOL (version) {
const myVersionInfo = await nv(version)
const myVersionInfo = await getVersionInfo(version)

if (!myVersionInfo) {
// i.e. isNodeEOL('abcd')
// i.e. isNodeEOL('abcd') or isNodeEOL('lts') or isNodeEOL('99')
throw Error(`Could not fetch version information for ${version}`)
} else if (myVersionInfo.length !== 1) {
// i.e. isNodeEOL('lts') or isNodeEOL('99')
throw Error(`Did not get exactly one version record for ${version}`)
} else if (!myVersionInfo[0].end) {
} else if (!myVersionInfo.end) {
// We got a record, but..
// v0.12.18 etc does not have an EOL date, which probably means too old.
return true
}

const now = new Date()
const end = new Date(myVersionInfo[0].end)
const end = new Date(myVersionInfo.end)
return now > end
}

Expand All @@ -175,7 +198,7 @@ async function isNodeVulnerable (version, platform) {
return true
}

const coreIndex = await getCoreIndex()
const coreIndex = await getJson(STORE.security)
const list = getVulnerabilityList(version, coreIndex, platform)
return list.length > 0
}
Expand Down
Loading
Loading