hongaik commited on
Commit
e72ff75
·
1 Parent(s): 6a18847
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -295,14 +295,16 @@ with st.form("inputs"):
295
  st.write(st.secrets['private_key'])
296
  import re
297
  r = re.compile("\s*-----BEGIN (.*)-----\n")
298
- m = r.match(st.secrets["private_key"].replace("-----BEGIN PRIVATE KEY-----", "-----BEGIN PRIVATE KEY-----\n"))
299
  if not m:
300
  raise ValueError("Not a valid PEM pre boundaryAAA")
301
- st.write(m.group(1))
 
302
 
 
303
  r = re.compile("-----END (.*)-----\s*$")
304
- m = r.search(st.secrets["private_key"].replace("-----BEGIN PRIVATE KEY-----", "-----BEGIN PRIVATE KEY-----\n")[:-2])
305
- if not m:
306
  raise ValueError("Not a valid PEM post boundaryAAA")
307
  st.write(m.group(1))
308
 
 
295
  st.write(st.secrets['private_key'])
296
  import re
297
  r = re.compile("\s*-----BEGIN (.*)-----\n")
298
+ m = r.match(st.secrets['private_key'])
299
  if not m:
300
  raise ValueError("Not a valid PEM pre boundaryAAA")
301
+ marker = m.group(1)
302
+ st.write(marker)
303
 
304
+ # Verify Post-Encapsulation Boundary
305
  r = re.compile("-----END (.*)-----\s*$")
306
+ m = r.search(st.secrets['private_key'])
307
+ if not m or m.group(1) != marker:
308
  raise ValueError("Not a valid PEM post boundaryAAA")
309
  st.write(m.group(1))
310