]>
Commit | Line | Data |
---|---|---|
41fb13c3 | 1 | import { IPv4, IPv6, parse, subnetMatch } from 'ipaddr.js' |
6dd9de95 | 2 | import { CONFIG } from '../initializers/config' |
41fb13c3 | 3 | import { UserModel } from '../models/user/user' |
6dd9de95 | 4 | |
06215f15 C |
5 | const isCidr = require('is-cidr') |
6 | ||
4ce7eb71 | 7 | async function isSignupAllowed (): Promise<{ allowed: boolean, errorMessage?: string }> { |
06215f15 | 8 | if (CONFIG.SIGNUP.ENABLED === false) { |
4ce7eb71 | 9 | return { allowed: false } |
06215f15 C |
10 | } |
11 | ||
12 | // No limit and signup is enabled | |
13 | if (CONFIG.SIGNUP.LIMIT === -1) { | |
4ce7eb71 | 14 | return { allowed: true } |
06215f15 C |
15 | } |
16 | ||
17 | const totalUsers = await UserModel.countTotal() | |
18 | ||
4ce7eb71 | 19 | return { allowed: totalUsers < CONFIG.SIGNUP.LIMIT } |
06215f15 C |
20 | } |
21 | ||
22 | function isSignupAllowedForCurrentIP (ip: string) { | |
22820226 C |
23 | if (!ip) return false |
24 | ||
41fb13c3 | 25 | const addr = parse(ip) |
a1587156 | 26 | const excludeList = [ 'blacklist' ] |
06215f15 C |
27 | let matched = '' |
28 | ||
29 | // if there is a valid, non-empty whitelist, we exclude all unknown adresses too | |
30 | if (CONFIG.SIGNUP.FILTERS.CIDR.WHITELIST.filter(cidr => isCidr(cidr)).length > 0) { | |
31 | excludeList.push('unknown') | |
32 | } | |
33 | ||
34 | if (addr.kind() === 'ipv4') { | |
41fb13c3 | 35 | const addrV4 = IPv4.parse(ip) |
06215f15 C |
36 | const rangeList = { |
37 | whitelist: CONFIG.SIGNUP.FILTERS.CIDR.WHITELIST.filter(cidr => isCidr.v4(cidr)) | |
41fb13c3 | 38 | .map(cidr => IPv4.parseCIDR(cidr)), |
06215f15 | 39 | blacklist: CONFIG.SIGNUP.FILTERS.CIDR.BLACKLIST.filter(cidr => isCidr.v4(cidr)) |
41fb13c3 | 40 | .map(cidr => IPv4.parseCIDR(cidr)) |
06215f15 | 41 | } |
41fb13c3 | 42 | matched = subnetMatch(addrV4, rangeList, 'unknown') |
06215f15 | 43 | } else if (addr.kind() === 'ipv6') { |
41fb13c3 | 44 | const addrV6 = IPv6.parse(ip) |
06215f15 C |
45 | const rangeList = { |
46 | whitelist: CONFIG.SIGNUP.FILTERS.CIDR.WHITELIST.filter(cidr => isCidr.v6(cidr)) | |
41fb13c3 | 47 | .map(cidr => IPv6.parseCIDR(cidr)), |
06215f15 | 48 | blacklist: CONFIG.SIGNUP.FILTERS.CIDR.BLACKLIST.filter(cidr => isCidr.v6(cidr)) |
41fb13c3 | 49 | .map(cidr => IPv6.parseCIDR(cidr)) |
06215f15 | 50 | } |
41fb13c3 | 51 | matched = subnetMatch(addrV6, rangeList, 'unknown') |
06215f15 C |
52 | } |
53 | ||
54 | return !excludeList.includes(matched) | |
55 | } | |
56 | ||
57 | // --------------------------------------------------------------------------- | |
58 | ||
59 | export { | |
60 | isSignupAllowed, | |
61 | isSignupAllowedForCurrentIP | |
62 | } |