IPKM/python-pygmx
3
0
Fork 0
Code Issues Pull Requests Activity

Catch some possible seg faults when opening invalid xtc files.

Browse Source
This commit is contained in:
Niels Müller
2016-06-06 16:59:16 +02:00
parent b16e4aac11
commit 208a51acfe
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pygmx/xtcio.pyx
View File

@ -67,6 +67,9 @@ cdef array get_xtc_index(t_fileio *fio):
return cache[:-1] return cache[:-1]
class XTCError(Exception): pass
cdef class XTCReader: cdef class XTCReader:
cdef: cdef:
t_fileio *fio t_fileio *fio
@ -158,6 +161,11 @@ cdef class XTCReader:
real time, prec real time, prec
rvec *x rvec *x
if not os.path.exists(filename):
raise OSError('File not found: {}'.format(filename))
if filename.decode().split('.')[-1] != 'xtc':
raise XTCError('File is not of xtc type: {}'.format(filename))
self.fio = open_xtc(filename, b'r') self.fio = open_xtc(filename, b'r')
read_first_xtc(self.fio, &self.natoms, &step, &time, box, &x, &prec, &_bOK) read_first_xtc(self.fio, &self.natoms, &step, &time, box, &x, &prec, &_bOK)