1/* 2 * Check-out files from the "current cache directory" 3 * 4 * Copyright (C) 2005 Linus Torvalds 5 * 6 * Careful: order of argument flags does matter. For example, 7 * 8 * checkout-cache -a -f file.c 9 * 10 * Will first check out all files listed in the cache (but not 11 * overwrite any old ones), and then force-checkout "file.c" a 12 * second time (ie that one _will_ overwrite any old contents 13 * with the same filename). 14 * 15 * Also, just doing "checkout-cache" does nothing. You probably 16 * meant "checkout-cache -a". And if you want to force it, you 17 * want "checkout-cache -f -a". 18 * 19 * Intuitiveness is not the goal here. Repeatability is. The 20 * reason for the "no arguments means no work" thing is that 21 * from scripts you are supposed to be able to do things like 22 * 23 * find . -name '*.h' -print0 | xargs -0 checkout-cache -f -- 24 * 25 * which will force all existing *.h files to be replaced with 26 * their cached copies. If an empty command line implied "all", 27 * then this would force-refresh everything in the cache, which 28 * was not the point. 29 * 30 * Oh, and the "--" is just a good idea when you know the rest 31 * will be filenames. Just so that you wouldn't have a filename 32 * of "-a" causing problems (not possible in the above example, 33 * but get used to it in scripting!). 34 */ 35#include"cache.h" 36 37static int force =0, quiet =0, not_new =0; 38 39static voidcreate_directories(const char*path) 40{ 41int len =strlen(path); 42char*buf =xmalloc(len +1); 43const char*slash = path; 44 45while((slash =strchr(slash+1,'/')) != NULL) { 46 len = slash - path; 47memcpy(buf, path, len); 48 buf[len] =0; 49mkdir(buf,0755); 50} 51} 52 53static intcreate_file(const char*path,unsigned int mode) 54{ 55int fd; 56 57 mode = (mode &0100) ?0777:0666; 58 fd =open(path, O_WRONLY | O_TRUNC | O_CREAT, mode); 59if(fd <0) { 60if(errno == ENOENT) { 61create_directories(path); 62 fd =open(path, O_WRONLY | O_TRUNC | O_CREAT, mode); 63} 64} 65return fd; 66} 67 68static intwrite_entry(struct cache_entry *ce,const char*path) 69{ 70int fd; 71void*new; 72unsigned long size; 73long wrote; 74char type[20]; 75 76new=read_sha1_file(ce->sha1, type, &size); 77if(!new||strcmp(type,"blob")) { 78returnerror("checkout-cache: unable to read sha1 file of%s(%s)", 79 path,sha1_to_hex(ce->sha1)); 80} 81 fd =create_file(path,ntohl(ce->ce_mode)); 82if(fd <0) { 83free(new); 84returnerror("checkout-cache: unable to create%s(%s)", 85 path,strerror(errno)); 86} 87 wrote =write(fd,new, size); 88close(fd); 89free(new); 90if(wrote != size) 91returnerror("checkout-cache: unable to write%s", path); 92return0; 93} 94 95static intcheckout_entry(struct cache_entry *ce,const char*base_dir) 96{ 97struct stat st; 98static char path[MAXPATHLEN+1]; 99int len =strlen(base_dir); 100 101memcpy(path, base_dir, len); 102strcpy(path + len, ce->name); 103 104if(!stat(path, &st)) { 105unsigned changed =cache_match_stat(ce, &st); 106if(!changed) 107return0; 108if(!force) { 109if(!quiet) 110fprintf(stderr,"checkout-cache:%salready exists\n", path); 111return0; 112} 113 114/* 115 * We unlink the old file, to get the new one with the 116 * right permissions (including umask, which is nasty 117 * to emulate by hand - much easier to let the system 118 * just do the right thing) 119 */ 120unlink(path); 121}else if(not_new) 122return0; 123returnwrite_entry(ce, path); 124} 125 126static intcheckout_file(const char*name,const char*base_dir) 127{ 128int pos =cache_name_pos(name,strlen(name)); 129if(pos <0) { 130if(!quiet) { 131 pos = -pos -1; 132fprintf(stderr, 133"checkout-cache:%sis%s.\n", 134 name, 135(pos < active_nr && 136!strcmp(active_cache[pos]->name, name)) ? 137"unmerged":"not in the cache"); 138} 139return-1; 140} 141returncheckout_entry(active_cache[pos], base_dir); 142} 143 144static intcheckout_all(const char*base_dir) 145{ 146int i; 147 148for(i =0; i < active_nr ; i++) { 149struct cache_entry *ce = active_cache[i]; 150if(ce_stage(ce)) 151continue; 152if(checkout_entry(ce, base_dir) <0) 153return-1; 154} 155return0; 156} 157 158intmain(int argc,char**argv) 159{ 160int i, force_filename =0; 161const char*base_dir =""; 162 163if(read_cache() <0) { 164die("invalid cache"); 165} 166 167for(i =1; i < argc; i++) { 168const char*arg = argv[i]; 169if(!force_filename) { 170if(!strcmp(arg,"-a")) { 171checkout_all(base_dir); 172continue; 173} 174if(!strcmp(arg,"--")) { 175 force_filename =1; 176continue; 177} 178if(!strcmp(arg,"-f")) { 179 force =1; 180continue; 181} 182if(!strcmp(arg,"-q")) { 183 quiet =1; 184continue; 185} 186if(!strcmp(arg,"-n")) { 187 not_new =1; 188continue; 189} 190if(!memcmp(arg,"--prefix=",9)) { 191 base_dir = arg+9; 192continue; 193} 194} 195checkout_file(arg, base_dir); 196} 197return0; 198}