Changeset 13800

Timestamp:

Jan 17, 2013, 6:55:51 PM (8 years ago)

Author:

jordan

Message:

(utils) #4137 'support user-defined piece sizes in transmission-create' -- done. Initial patch by lav.

Location:

trunk

Files:

• libtransmission/makemeta.c (modified) (1 diff)
• libtransmission/makemeta.h (modified) (1 diff)
• utils/create.c (modified) (5 diffs)
• utils/transmission-create.1 (modified) (2 diffs)

trunk/libtransmission/makemeta.c

@@ -163 +163 @@
            builderFileCompare);
 
-    ret->pieceSize = bestPieceSize (ret->totalSize);
-    ret->pieceCount = (int)(ret->totalSize / ret->pieceSize);
-    if (ret->totalSize % ret->pieceSize)
-        ++ret->pieceCount;
+    tr_metaInfoBuilderSetPieceSize (ret, bestPieceSize (ret->totalSize));
 
     return ret;
 }
+
+void
+tr_metaInfoBuilderSetPieceSize (tr_metainfo_builder * b,
+                                uint32_t              bytes)
+{
+  b->pieceSize = bytes;
+
+  b->pieceCount = (int)(b->totalSize / b->pieceSize);
+  if (b->totalSize % b->pieceSize)
+    ++b->pieceCount;
+}
+
 
 void

trunk/libtransmission/makemeta.h

@@ -92 +92 @@
 
 
-tr_metainfo_builder*tr_metaInfoBuilderCreate (const char * topFile);
+tr_metainfo_builder * tr_metaInfoBuilderCreate (const char * topFile);
 
-void                tr_metaInfoBuilderFree (tr_metainfo_builder*);
+/**
+ * Call this before tr_makeMetaInfo() to override the builder.pieceSize
+ * and builder.pieceCount values that were set by tr_metainfoBuilderCreate()
+ */
+void tr_metaInfoBuilderSetPieceSize (tr_metainfo_builder * builder,
+                                     uint32_t              bytes);
+
+void tr_metaInfoBuilderFree (tr_metainfo_builder*);
 
 /**

trunk/utils/create.c

@@ -12 +12 @@
 
 #include <errno.h>
-#include <stdio.h> /* fprintf () */
-#include <stdlib.h> /* EXIT_FAILURE */
-#include <unistd.h> /* getcwd () */
+#include <stdio.h> /* fprintf() */
+#include <stdlib.h> /* strtoul(), EXIT_FAILURE */
+#include <unistd.h> /* getcwd() */
 
 #include <libtransmission/transmission.h>
@@ -25 +25 @@
 
 #define MAX_TRACKERS 128
+static const uint32_t KiB = 1024;
 static tr_tracker_info trackers[MAX_TRACKERS];
 static int trackerCount = 0;
 static bool isPrivate = false;
 static bool showVersion = false;
-const char * comment = NULL;
-const char * outfile = NULL;
-const char * infile = NULL;
+static const char * comment = NULL;
+static const char * outfile = NULL;
+static const char * infile = NULL;
+static uint32_t piecesize_kib = 0;
 
 static tr_option options[] =
@@ -37 +39 @@
   { 'p', "private", "Allow this torrent to only be used with the specified tracker(s)", "p", 0, NULL },
   { 'o', "outfile", "Save the generated .torrent to this filename", "o", 1, "<file>" },
+  { 's', "piecesize", "Set how many KiB each piece should be, overriding the preferred default", "s", 1, "<size in KiB>" },
   { 'c', "comment", "Add a comment", "c", 1, "<comment>" },
   { 't', "tracker", "Add a tracker's announce URL", "t", 1, "<url>" },
@@ -81 +84 @@
                 trackers[trackerCount].announce = (char*) optarg;
                   ++trackerCount;
+              }
+            break;
+
+          case 's':
+            if (optarg) 
+              { 
+                char * endptr = NULL;
+                piecesize_kib = strtoul (optarg, &endptr, 10);
+                if (endptr && *endptr=='M')
+                  piecesize_kib *= KiB; 
               }
             break;
@@ -170 +183 @@
 
   b = tr_metaInfoBuilderCreate (infile);
+
+  if (piecesize_kib != 0)
+    tr_metaInfoBuilderSetPieceSize (b, piecesize_kib * KiB);
+
   tr_makeMetaInfo (b, outfile, trackers, trackerCount, comment, isPrivate);
   while (!b->isDone)

trunk/utils/transmission-create.1

@@ -13 +13 @@
 .Op Fl c Ar comment
 .Op Fl t Ar tracker
+.Op Fl s Ar piece-size-KiB
 .Op Ar source file or directory
 .Ek
@@ -28 +29 @@
 .It Fl c Fl -comment
 Add a comment to the torrent file.
+.It Fl s Fl -piecesize
+Set how many KiB each piece should be, overriding the preferred default
 .It Fl t Fl -tracker
 Add a tracker's