Merge pull request #2 from barakwei/master

Added file size and UNC support

Author: JohnWintellect

FF/Constants.Designer.cs

@@ -182,4 +182,10 @@ Examples:
   <data name="InvalidRegExFmt" xml:space="preserve">
     <value>"Invalid regular expression: '{0}' Error : {1}"</value>
   </data>
+  <data name="TotalMatchesSizeFmt" xml:space="preserve">
+    <value>Total matches size          : {0} bytes</value>
+  </data>
+  <data name="TotalSizeFmt" xml:space="preserve">
+    <value>Total size                  : {0} bytes</value>
+  </data>
 </root>

FF/Constants.resx

@@ -34,6 +34,11 @@ internal sealed class Program
         /// </summary>
         private static Int64 totalMatches;
 
+        /// <summary>
+        /// The total number of bytes the matching file consume.
+        /// </summary>
+        private static Int64 totalMatchesSize;
+
         /// <summary>
         /// The total number of files looked at.
         /// </summary>
@@ -44,6 +49,11 @@ internal sealed class Program
         /// </summary>
         private static Int64 totalDirectories;
 
+        /// <summary>
+        /// The total number of bytes the files looked at consume.
+        /// </summary>
+        private static Int64 totalSize;
+
         /// <summary>
         /// The collection to hold found strings so they can be printed in batch mode.
         /// </summary>
@@ -72,8 +82,10 @@ internal static Int32 Main(String[] args)
             Boolean parsed = Options.Parse(args);
 
             totalMatches = 0;
+            totalMatchesSize = 0;
             totalFiles = 0;
             totalDirectories = 0;
+            totalSize = 0;
 
             if (parsed)
             {
@@ -96,9 +108,11 @@ internal static Int32 Main(String[] args)
                     Console.WriteLine(Constants.TotalTimeFmt, timer.ElapsedMilliseconds.ToString("N0", CultureInfo.CurrentCulture));
                     Console.WriteLine(Constants.TotalFilesFmt, totalFiles.ToString("N0", CultureInfo.CurrentCulture));
                     Console.WriteLine(Constants.TotalDirectoriesFmt, totalDirectories.ToString("N0", CultureInfo.CurrentCulture));
+                    Console.WriteLine(Constants.TotalSizeFmt, totalSize.ToString("N0", CultureInfo.CurrentCulture));
                     Console.WriteLine(Constants.TotalMatchesFmt, totalMatches.ToString("N0", CultureInfo.CurrentCulture));
+                    Console.WriteLine(Constants.TotalMatchesSizeFmt, totalMatchesSize.ToString("N0", CultureInfo.CurrentCulture));
                 }
-            }
+              }
             else
             {
                 returnValue = 1;
@@ -256,7 +270,15 @@ private static void QueueConsoleWriteLine(String line)
         /// </remarks>
         static private void RecurseFiles(String directory)
         {
-            String lookUpdirectory = "\\\\?\\" + directory + "\\*";
+            String lookUpdirectory = String.Empty;
+            if (directory.StartsWith(@"\\", StringComparison.OrdinalIgnoreCase))
+            {
+              lookUpdirectory += directory.Replace(@"\\", @"\\?\UNC\") + "\\*";
+            }
+            else
+            {
+              lookUpdirectory = "\\\\?\\" + directory + "\\*";
+            }
             NativeMethods.WIN32_FIND_DATA w32FindData;
 
             using (SafeFindFileHandle fileHandle = NativeMethods.FindFirstFileEx(lookUpdirectory,
@@ -297,9 +319,12 @@ static private void RecurseFiles(String directory)
                         }
                         else
                         {
-                            // It's a file so look at it.
+                          // It's a file so look at it.
                             Interlocked.Increment(ref totalFiles);
 
+                            Int64 fileSize = w32FindData.nFileSizeLow + ((Int64)w32FindData.nFileSizeHigh << 32);
+                            Interlocked.Add(ref totalSize, fileSize);
+
                             String fullFile = directory;
                             if (!directory.EndsWith("\\", StringComparison.OrdinalIgnoreCase))
                             {
@@ -317,6 +342,7 @@ static private void RecurseFiles(String directory)
                             if (IsNameMatch(matchName))
                             {
                                 Interlocked.Increment(ref totalMatches);
+                                Interlocked.Add(ref totalMatchesSize, fileSize);
                                 QueueConsoleWriteLine(fullFile);
                             }
                         }