-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
190 lines (165 loc) · 7.15 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
#!/usr/bin/env python3
from subprocess import check_output, CalledProcessError
from sys import argv, platform
from shlex import split as s_split
import sysconfig
from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext as BuildCommand
import setuptools.command.build as build
from setuptools import dist
# Determine if a base directory has been provided with the --basedir option
basedir = None
in_tree = False
# Add compiler flags if debug is set
compile_args = ["-std=c++17", "-Wunused-function", "-fpermissive"]
link_args = []
for arg in argv:
if arg.startswith("--debug"):
# Note from GCC manual:
# If you use multiple -O options, with or without level numbers,
# the last such option is the one that is effective.
compile_args.extend(["-Wall", "-O0", "-g"])
elif arg.startswith("--basedir="):
basedir = arg.split("=")[1]
in_tree = True
# If a base directory has been provided, we use it
if in_tree:
base_cmd = "{0}/net-snmp-config {{{0}}}".format(basedir)
libs_cmd = base_cmd.format("--build-lib-dirs {0}".format(basedir))
incl_cmd = base_cmd.format("--build-includes {0}".format(basedir))
netsnmp_libs = check_output(base_cmd.format("--libs"), shell=True).decode()
libdirs = check_output(libs_cmd, shell=True).decode()
incdirs = check_output(incl_cmd, shell=True).decode()
libs = [flag[2:] for flag in s_split(netsnmp_libs) if flag[:2] == "-l"]
libdirs = [flag[2:] for flag in s_split(libdirs) if flag[:2] == "-L"]
incdirs = [flag[2:] for flag in s_split(incdirs) if flag[:2] == "-I"]
# Otherwise, we use the system-installed SNMP libraries
else:
netsnmp_libs = check_output("net-snmp-config --libs", shell=True).decode()
pass_next = False
# macOS-specific
has_arg = ("-framework",)
for flag in s_split(netsnmp_libs):
if pass_next:
link_args.append(flag)
pass_next = False
elif flag in has_arg: # -framework CoreFoundation
link_args.append(flag)
pass_next = True
elif flag == "-flat_namespace":
link_args.append(flag)
pass_next = False
# link_args += [flag for flag in s_split(netsnmp_libs) if flag[:2] == '-f']
libs = [flag[2:] for flag in s_split(netsnmp_libs) if flag[:2] == "-l"]
libdirs = [flag[2:] for flag in s_split(netsnmp_libs) if flag[:2] == "-L"]
incdirs = []
if platform == "darwin": # OS X
# Check if net-snmp is installed via Brew
try:
brew = check_output("brew list net-snmp 2>/dev/null", shell=True).decode()
lines = brew.splitlines()
include_dir = list(filter(lambda l: "include/net-snmp" in l, lines))[0]
incdirs.append(include_dir[: include_dir.index("include/net-snmp") + 7])
lib_dir = list(filter(lambda l: "lib/libnetsnmp.dylib" in l, lines))[0]
libdirs.append(lib_dir[: lib_dir.index("lib/libnetsnmp.dylib") + 3])
# The homebrew version also depends on the Openssl keg
brew = check_output("brew info net-snmp", shell=True).decode()
openssl_ver = list(
filter(
lambda o: "openssl" in o,
*map(
str.split,
filter(
lambda l: "openssl" in l,
str(brew.replace("'", "")).split("\n"),
),
),
)
)[0]
brew = check_output(
"brew info {0}".format(openssl_ver), shell=True
).decode()
temp = brew.split("\n")
# As of 06/04/2024 brew info openssl spits out lines. the fifth one is what we care about
# This works for now, but we need a better solution
# ==> openssl@3: stable 3.3.0 (bottled)
# Cryptography and SSL/TLS Toolkit
# https://openssl.org/
# Installed
# /opt/homebrew/Cellar/openssl@3/3.3.0 (6,977 files, 32.4MB) *
# Poured from bottle using the formulae.brew.sh API on 2024-06-04 at 21:17:37
# From: https://github.com/Homebrew/homebrew-core/blob/HEAD/Formula/o/openssl@3.rb
# License: Apache-2.0
# ...
# print(temp)
temp_path = str(temp[4].split("(")[0]).strip()
libdirs.append(temp_path + "/lib")
incdirs.append(temp_path + "/include")
print(f"libdirs: {libdirs}")
print(f"incdirs: {incdirs}")
print(f"openssl_ver: {openssl_ver}")
except CalledProcessError:
print("A brew command failed...")
pass
print(f"in_tree: {in_tree}")
print(f"compile_args: {compile_args}")
print(f"link_args: {link_args}")
print(f"platform: {platform}")
class RelinkLibraries(BuildCommand):
"""Fix dylib path for macOS
Depending on system configuration and Brew setup, interface.so may get linked to
the wrong dylib file (prioritizing the system's 5.6.2.1 over brew's 5.8+). This
will change the path if net-snmp is installed via `brew`.
Non-brew installations and non-macOS systems will not be affected.
"""
def run(self):
BuildCommand.run(self)
if platform == "darwin": # Newer Net-SNMP dylib may not be linked to properly
try:
brew = check_output(
"brew list net-snmp 2>/dev/null", shell=True
).decode()
except CalledProcessError:
return
lib_dir = list(filter(lambda l: "lib/libnetsnmp.dylib" in l, lines))[0]
b = build.build(dist.Distribution()) # Dynamically determine build path
b.finalize_options()
ext = sysconfig.get_config_var("EXT_SUFFIX") or ".so" # None for Python 2
linked = (
check_output(
(
"otool -L {0}/ezsnmp/interface{1} | "
r"egrep 'libnetsnmp\.' | "
"tr -s '\t' ' ' | "
"cut -d' ' -f2"
).format(b.build_platlib, ext),
shell=True,
)
.decode()
.strip()
)
if linked:
# install_name_tool -change /opt/homebrew/opt/net-snmp/lib/libnetsnmp.40.dylib /opt/homebrew/Cellar/net-snmp/5.9.4/lib/libnetsnmp.dylib build/lib.macosx-10.9-universal2-cpython-39/ezsnmp/interface.cpython-39-darwin.so
cmd_to_run = (
"install_name_tool -change {0} {1} {2}/ezsnmp/interface{3}".format(
linked, lib_dir, b.build_platlib, ext
)
)
print("cmd_to_run: ", cmd_to_run)
_ = check_output(
cmd_to_run,
shell=True,
)
setup(
ext_modules=[
Extension(
"ezsnmp.interface",
["ezsnmp/interface.cpp"],
library_dirs=libdirs,
include_dirs=incdirs,
libraries=libs,
extra_compile_args=compile_args,
extra_link_args=link_args,
)
],
)