You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

129 lines
3.8 KiB

5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
  1. #!/usr/bin/env node
  2. /*
  3. Copyright 2018 0KIMS association.
  4. This file is part of circom (Zero Knowledge Circuit Compiler).
  5. circom is a free software: you can redistribute it and/or modify it
  6. under the terms of the GNU General Public License as published by
  7. the Free Software Foundation, either version 3 of the License, or
  8. (at your option) any later version.
  9. circom is distributed in the hope that it will be useful, but WITHOUT
  10. ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  11. or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
  12. License for more details.
  13. You should have received a copy of the GNU General Public License
  14. along with circom. If not, see <https://www.gnu.org/licenses/>.
  15. */
  16. /* eslint-disable no-console */
  17. const fs = require("fs");
  18. const path = require("path");
  19. const compiler = require("./src/compiler");
  20. const version = require("./package").version;
  21. const argv = require("yargs")
  22. .version(version)
  23. .usage("circom [input source circuit file] -o [output definition circuit file] -c [output c file]")
  24. .alias("o", "output")
  25. .alias("c", "csource")
  26. .alias("s", "sym")
  27. .alias("r", "r1cs")
  28. .help("h")
  29. .alias("h", "help")
  30. .option("verbose", {
  31. alias: "v",
  32. type: "boolean",
  33. description: "Run with verbose logging"
  34. })
  35. .option("fast", {
  36. alias: "f",
  37. type: "boolean",
  38. description: "Do not optimize constraints"
  39. })
  40. .epilogue(`Copyright (C) 2018 0kims association
  41. This program comes with ABSOLUTELY NO WARRANTY;
  42. This is free software, and you are welcome to redistribute it
  43. under certain conditions; see the COPYING file in the official
  44. repo directory at https://github.com/iden3/circom `)
  45. .argv;
  46. let inputFile;
  47. if (argv._.length == 0) {
  48. inputFile = "circuit.circom";
  49. } else if (argv._.length == 1) {
  50. inputFile = argv._[0];
  51. } else {
  52. console.log("Only one circuit at a time is permited");
  53. process.exit(1);
  54. }
  55. const fullFileName = path.resolve(process.cwd(), inputFile);
  56. const fileName = path.basename(fullFileName, ".circom");
  57. const outName = argv.output ? argv.output : fileName + ".json";
  58. const cSourceName = typeof(argv.csource) === "string" ? argv.csource : fileName + ".cpp";
  59. const r1csName = typeof(argv.r1cs) === "string" ? argv.r1cs : fileName + ".r1cs";
  60. const symName = typeof(argv.sym) === "string" ? argv.sym : fileName + ".sym";
  61. const options = {};
  62. options.reduceConstraints = !argv.fast;
  63. options.verbose = argv.verbose || false;
  64. if (argv.csource) {
  65. options.cSourceWriteStream = fs.createWriteStream(cSourceName);
  66. }
  67. if (argv.r1cs) {
  68. options.r1csWriteStream = fs.createWriteStream(r1csName);
  69. }
  70. if (argv.sym) {
  71. options.symWriteStream = fs.createWriteStream(symName);
  72. }
  73. compiler(fullFileName, options).then( () => {
  74. let r1csDone = false;
  75. let cSourceDone = false;
  76. if (options.r1csWriteStream) {
  77. options.r1csWriteStream.end(() => {
  78. r1csDone = true;
  79. finishIfDone();
  80. });
  81. } else {
  82. r1csDone = true;
  83. }
  84. if (options.cSourceWriteStream) {
  85. options.cSourceWriteStream.end(() => {
  86. cSourceDone = true;
  87. finishIfDone();
  88. });
  89. } else {
  90. cSourceDone = true;
  91. }
  92. function finishIfDone() {
  93. if ((r1csDone)&&(cSourceDone)) {
  94. process.exit(0);
  95. }
  96. }
  97. }, (err) => {
  98. // console.log(err);
  99. console.log(err.stack);
  100. if (err.pos) {
  101. console.error(`ERROR at ${err.errFile}:${err.pos.first_line},${err.pos.first_column}-${err.pos.last_line},${err.pos.last_column} ${err.errStr}`);
  102. } else {
  103. console.log(err.message);
  104. if (argv.verbose) console.log(err.stack);
  105. }
  106. if (err.ast) {
  107. console.error(JSON.stringify(err.ast, null, 1));
  108. }
  109. process.exit(1);
  110. });