[PATCH] modifyrepo: sanitize opts.sumtype. BZ 1013614

Zdenek Pavlas <[email protected]> Mon, 30 Sep 2013 15:20:24 +0200
Newsgroups gmane.linux.rpm.metadata
Message-ID <[email protected]>
---
 modifyrepo.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/modifyrepo.py b/modifyrepo.py
index 9cf4bae..e52507c 100755
--- a/modifyrepo.py
+++ b/modifyrepo.py
@@ -32,7 +32,7 @@ import sys
 from createrepo import __version__
 from createrepo.utils import checksum_and_rename, compressOpen, MDError
 from createrepo.utils import _available_compression
-from yum.misc import checksum
+from yum.misc import checksum, _available_checksums
 
 from yum.repoMDObject import RepoMD, RepoMDError, RepoData
 from xml.dom import minidom
@@ -231,6 +231,9 @@ def main(args):
     if opts.compress_type not in _available_compression:
         print "Compression %s not available: Please choose from: %s" % (opts.compress_type, ', '.join(_available_compression))
         return 1
+    if opts.sumtype not in _available_checksums:
+        print "Checksum %s not available: Please choose from: %s" % (opts.sumtype, ', '.join(_available_checksums))
+        return 1
     repomd.compress_type = opts.compress_type
 
     # remove
-- 
1.7.11.7