Add isCString() - returns true if a ConstantArray is a CString.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@31201 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/include/llvm/Constants.h b/include/llvm/Constants.h
index 3bf935bb5b0233c09301b7ff7d0cf12a1a75dc05..f5fff4d7fb494a97e0ff72befbb456b2cfd5765a 100644 (file)
--- a/include/llvm/Constants.h
+++ b/include/llvm/Constants.h
@@ -334,6 +334,11 @@ public:
   /// ubyte, and if the elements of the array are all ConstantInt's.
   bool isString() const;
 
+  /// isCString - This method returns true if the array is a string (see
+  /// isString) and it ends in a null byte \0 and does not contains any other
+  /// null bytes except its terminator.
+  bool isCString() const;
+
   /// getAsString - If this array is isString(), then this method converts the
   /// array to an std::string and returns it.  Otherwise, it asserts out.
   ///

diff --git a/lib/VMCore/Constants.cpp b/lib/VMCore/Constants.cpp
index 2c996f5301f4330f8adfcfc07e378a2e1d2a0624..ce29a21f86c520da2040361ec7b3ab19bbadf065 100644 (file)
--- a/lib/VMCore/Constants.cpp
+++ b/lib/VMCore/Constants.cpp
@@ -1073,6 +1073,19 @@ bool ConstantArray::isString() const {
   return true;
 }
 
+/// isCString - This method returns true if the array is a string (see
+/// isString) and it ends in a null byte \0 and does not contains any other
+/// null bytes except its terminator.
+bool ConstantArray::isCString() const {
+  if (!isString()) return false;
+  // This is safe because a ConstantArray cannot be a zero array.
+  for (unsigned i = 0, e = getNumOperands()-1; i != e; ++i)
+    if (cast<ConstantInt>(getOperand(i))->getZExtValue() == 0)
+      return false;
+  return true;
+}
+
+
 // getAsString - If the sub-element type of this array is either sbyte or ubyte,
 // then this method converts the array to an std::string and returns it.
 // Otherwise, it asserts out.