/ Hex Artifact Content
Login

Artifact beea82bb5055b6590cffe2d2e6d922905894f691:


0000: 2f 2a 0a 2a 2a 20 32 30 31 34 2d 30 36 2d 31 33  /*.** 2014-06-13
0010: 0a 2a 2a 0a 2a 2a 20 54 68 65 20 61 75 74 68 6f  .**.** The autho
0020: 72 20 64 69 73 63 6c 61 69 6d 73 20 63 6f 70 79  r disclaims copy
0030: 72 69 67 68 74 20 74 6f 20 74 68 69 73 20 73 6f  right to this so
0040: 75 72 63 65 20 63 6f 64 65 2e 20 20 49 6e 20 70  urce code.  In p
0050: 6c 61 63 65 20 6f 66 0a 2a 2a 20 61 20 6c 65 67  lace of.** a leg
0060: 61 6c 20 6e 6f 74 69 63 65 2c 20 68 65 72 65 20  al notice, here 
0070: 69 73 20 61 20 62 6c 65 73 73 69 6e 67 3a 0a 2a  is a blessing:.*
0080: 2a 0a 2a 2a 20 20 20 20 4d 61 79 20 79 6f 75 20  *.**    May you 
0090: 64 6f 20 67 6f 6f 64 20 61 6e 64 20 6e 6f 74 20  do good and not 
00a0: 65 76 69 6c 2e 0a 2a 2a 20 20 20 20 4d 61 79 20  evil..**    May 
00b0: 79 6f 75 20 66 69 6e 64 20 66 6f 72 67 69 76 65  you find forgive
00c0: 6e 65 73 73 20 66 6f 72 20 79 6f 75 72 73 65 6c  ness for yoursel
00d0: 66 20 61 6e 64 20 66 6f 72 67 69 76 65 20 6f 74  f and forgive ot
00e0: 68 65 72 73 2e 0a 2a 2a 20 20 20 20 4d 61 79 20  hers..**    May 
00f0: 79 6f 75 20 73 68 61 72 65 20 66 72 65 65 6c 79  you share freely
0100: 2c 20 6e 65 76 65 72 20 74 61 6b 69 6e 67 20 6d  , never taking m
0110: 6f 72 65 20 74 68 61 6e 20 79 6f 75 20 67 69 76  ore than you giv
0120: 65 2e 0a 2a 2a 0a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  e..**.**********
0130: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0140: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0150: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0160: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0170: 2a 2a 2a 2a 0a 2a 2a 0a 2a 2a 20 54 68 69 73 20  ****.**.** This 
0180: 53 51 4c 69 74 65 20 65 78 74 65 6e 73 69 6f 6e  SQLite extension
0190: 20 69 6d 70 6c 65 6d 65 6e 74 73 20 53 51 4c 20   implements SQL 
01a0: 66 75 6e 63 74 69 6f 6e 73 20 72 65 61 64 66 69  functions readfi
01b0: 6c 65 28 29 20 61 6e 64 0a 2a 2a 20 77 72 69 74  le() and.** writ
01c0: 65 66 69 6c 65 28 29 2e 0a 2a 2f 0a 23 69 6e 63  efile()..*/.#inc
01d0: 6c 75 64 65 20 22 73 71 6c 69 74 65 33 65 78 74  lude "sqlite3ext
01e0: 2e 68 22 0a 53 51 4c 49 54 45 5f 45 58 54 45 4e  .h".SQLITE_EXTEN
01f0: 53 49 4f 4e 5f 49 4e 49 54 31 0a 23 69 6e 63 6c  SION_INIT1.#incl
0200: 75 64 65 20 3c 73 74 64 69 6f 2e 68 3e 0a 0a 2f  ude <stdio.h>../
0210: 2a 0a 2a 2a 20 49 6d 70 6c 65 6d 65 6e 74 61 74  *.** Implementat
0220: 69 6f 6e 20 6f 66 20 74 68 65 20 22 72 65 61 64  ion of the "read
0230: 66 69 6c 65 28 58 29 22 20 53 51 4c 20 66 75 6e  file(X)" SQL fun
0240: 63 74 69 6f 6e 2e 20 20 54 68 65 20 65 6e 74 69  ction.  The enti
0250: 72 65 20 63 6f 6e 74 65 6e 74 0a 2a 2a 20 6f 66  re content.** of
0260: 20 74 68 65 20 66 69 6c 65 20 6e 61 6d 65 64 20   the file named 
0270: 58 20 69 73 20 72 65 61 64 20 61 6e 64 20 72 65  X is read and re
0280: 74 75 72 6e 65 64 20 61 73 20 61 20 42 4c 4f 42  turned as a BLOB
0290: 2e 20 20 4e 55 4c 4c 20 69 73 20 72 65 74 75 72  .  NULL is retur
02a0: 6e 65 64 0a 2a 2a 20 69 66 20 74 68 65 20 66 69  ned.** if the fi
02b0: 6c 65 20 64 6f 65 73 20 6e 6f 74 20 65 78 69 73  le does not exis
02c0: 74 20 6f 72 20 69 73 20 75 6e 72 65 61 64 61 62  t or is unreadab
02d0: 6c 65 2e 0a 2a 2f 0a 73 74 61 74 69 63 20 76 6f  le..*/.static vo
02e0: 69 64 20 72 65 61 64 66 69 6c 65 46 75 6e 63 28  id readfileFunc(
02f0: 0a 20 20 73 71 6c 69 74 65 33 5f 63 6f 6e 74 65  .  sqlite3_conte
0300: 78 74 20 2a 63 6f 6e 74 65 78 74 2c 0a 20 20 69  xt *context,.  i
0310: 6e 74 20 61 72 67 63 2c 0a 20 20 73 71 6c 69 74  nt argc,.  sqlit
0320: 65 33 5f 76 61 6c 75 65 20 2a 2a 61 72 67 76 0a  e3_value **argv.
0330: 29 7b 0a 20 20 63 6f 6e 73 74 20 63 68 61 72 20  ){.  const char 
0340: 2a 7a 4e 61 6d 65 3b 0a 20 20 46 49 4c 45 20 2a  *zName;.  FILE *
0350: 69 6e 3b 0a 20 20 6c 6f 6e 67 20 6e 49 6e 3b 0a  in;.  long nIn;.
0360: 20 20 76 6f 69 64 20 2a 70 42 75 66 3b 0a 0a 20    void *pBuf;.. 
0370: 20 7a 4e 61 6d 65 20 3d 20 28 63 6f 6e 73 74 20   zName = (const 
0380: 63 68 61 72 2a 29 73 71 6c 69 74 65 33 5f 76 61  char*)sqlite3_va
0390: 6c 75 65 5f 74 65 78 74 28 61 72 67 76 5b 30 5d  lue_text(argv[0]
03a0: 29 3b 0a 20 20 69 66 28 20 7a 4e 61 6d 65 3d 3d  );.  if( zName==
03b0: 30 20 29 20 72 65 74 75 72 6e 3b 0a 20 20 69 6e  0 ) return;.  in
03c0: 20 3d 20 66 6f 70 65 6e 28 7a 4e 61 6d 65 2c 20   = fopen(zName, 
03d0: 22 72 62 22 29 3b 0a 20 20 69 66 28 20 69 6e 3d  "rb");.  if( in=
03e0: 3d 30 20 29 20 72 65 74 75 72 6e 3b 0a 20 20 66  =0 ) return;.  f
03f0: 73 65 65 6b 28 69 6e 2c 20 30 2c 20 53 45 45 4b  seek(in, 0, SEEK
0400: 5f 45 4e 44 29 3b 0a 20 20 6e 49 6e 20 3d 20 66  _END);.  nIn = f
0410: 74 65 6c 6c 28 69 6e 29 3b 0a 20 20 72 65 77 69  tell(in);.  rewi
0420: 6e 64 28 69 6e 29 3b 0a 20 20 70 42 75 66 20 3d  nd(in);.  pBuf =
0430: 20 73 71 6c 69 74 65 33 5f 6d 61 6c 6c 6f 63 28   sqlite3_malloc(
0440: 20 6e 49 6e 20 29 3b 0a 20 20 69 66 28 20 70 42   nIn );.  if( pB
0450: 75 66 20 26 26 20 31 3d 3d 66 72 65 61 64 28 70  uf && 1==fread(p
0460: 42 75 66 2c 20 6e 49 6e 2c 20 31 2c 20 69 6e 29  Buf, nIn, 1, in)
0470: 20 29 7b 0a 20 20 20 20 73 71 6c 69 74 65 33 5f   ){.    sqlite3_
0480: 72 65 73 75 6c 74 5f 62 6c 6f 62 28 63 6f 6e 74  result_blob(cont
0490: 65 78 74 2c 20 70 42 75 66 2c 20 6e 49 6e 2c 20  ext, pBuf, nIn, 
04a0: 73 71 6c 69 74 65 33 5f 66 72 65 65 29 3b 0a 20  sqlite3_free);. 
04b0: 20 7d 65 6c 73 65 7b 0a 20 20 20 20 73 71 6c 69   }else{.    sqli
04c0: 74 65 33 5f 66 72 65 65 28 70 42 75 66 29 3b 0a  te3_free(pBuf);.
04d0: 20 20 7d 0a 20 20 66 63 6c 6f 73 65 28 69 6e 29    }.  fclose(in)
04e0: 3b 0a 7d 0a 0a 2f 2a 0a 2a 2a 20 49 6d 70 6c 65  ;.}../*.** Imple
04f0: 6d 65 6e 74 61 74 69 6f 6e 20 6f 66 20 74 68 65  mentation of the
0500: 20 22 77 72 69 74 65 66 69 6c 65 28 58 2c 59 29   "writefile(X,Y)
0510: 22 20 53 51 4c 20 66 75 6e 63 74 69 6f 6e 2e 20  " SQL function. 
0520: 20 54 68 65 20 61 72 67 75 6d 65 6e 74 20 59 0a   The argument Y.
0530: 2a 2a 20 69 73 20 77 72 69 74 74 65 6e 20 69 6e  ** is written in
0540: 74 6f 20 66 69 6c 65 20 58 2e 20 20 54 68 65 20  to file X.  The 
0550: 6e 75 6d 62 65 72 20 6f 66 20 62 79 74 65 73 20  number of bytes 
0560: 77 72 69 74 74 65 6e 20 69 73 20 72 65 74 75 72  written is retur
0570: 6e 65 64 2e 20 20 4f 72 0a 2a 2a 20 4e 55 4c 4c  ned.  Or.** NULL
0580: 20 69 73 20 72 65 74 75 72 6e 65 64 20 69 66 20   is returned if 
0590: 73 6f 6d 65 74 68 69 6e 67 20 67 6f 65 73 20 77  something goes w
05a0: 72 6f 6e 67 2c 20 73 75 63 68 20 61 73 20 62 65  rong, such as be
05b0: 69 6e 67 20 75 6e 61 62 6c 65 20 74 6f 20 6f 70  ing unable to op
05c0: 65 6e 0a 2a 2a 20 66 69 6c 65 20 58 20 66 6f 72  en.** file X for
05d0: 20 77 72 69 74 69 6e 67 2e 0a 2a 2f 0a 73 74 61   writing..*/.sta
05e0: 74 69 63 20 76 6f 69 64 20 77 72 69 74 65 66 69  tic void writefi
05f0: 6c 65 46 75 6e 63 28 0a 20 20 73 71 6c 69 74 65  leFunc(.  sqlite
0600: 33 5f 63 6f 6e 74 65 78 74 20 2a 63 6f 6e 74 65  3_context *conte
0610: 78 74 2c 0a 20 20 69 6e 74 20 61 72 67 63 2c 0a  xt,.  int argc,.
0620: 20 20 73 71 6c 69 74 65 33 5f 76 61 6c 75 65 20    sqlite3_value 
0630: 2a 2a 61 72 67 76 0a 29 7b 0a 20 20 46 49 4c 45  **argv.){.  FILE
0640: 20 2a 6f 75 74 3b 0a 20 20 63 6f 6e 73 74 20 63   *out;.  const c
0650: 68 61 72 20 2a 7a 3b 0a 20 20 69 6e 74 20 6e 3b  har *z;.  int n;
0660: 0a 20 20 73 71 6c 69 74 65 33 5f 69 6e 74 36 34  .  sqlite3_int64
0670: 20 72 63 3b 0a 20 20 63 6f 6e 73 74 20 63 68 61   rc;.  const cha
0680: 72 20 2a 7a 46 69 6c 65 3b 0a 0a 20 20 7a 46 69  r *zFile;..  zFi
0690: 6c 65 20 3d 20 28 63 6f 6e 73 74 20 63 68 61 72  le = (const char
06a0: 2a 29 73 71 6c 69 74 65 33 5f 76 61 6c 75 65 5f  *)sqlite3_value_
06b0: 74 65 78 74 28 61 72 67 76 5b 30 5d 29 3b 0a 20  text(argv[0]);. 
06c0: 20 69 66 28 20 7a 46 69 6c 65 3d 3d 30 20 29 20   if( zFile==0 ) 
06d0: 72 65 74 75 72 6e 3b 0a 20 20 6f 75 74 20 3d 20  return;.  out = 
06e0: 66 6f 70 65 6e 28 7a 46 69 6c 65 2c 20 22 77 62  fopen(zFile, "wb
06f0: 22 29 3b 0a 20 20 69 66 28 20 6f 75 74 3d 3d 30  ");.  if( out==0
0700: 20 29 20 72 65 74 75 72 6e 3b 0a 20 20 7a 20 3d   ) return;.  z =
0710: 20 28 63 6f 6e 73 74 20 63 68 61 72 2a 29 73 71   (const char*)sq
0720: 6c 69 74 65 33 5f 76 61 6c 75 65 5f 62 6c 6f 62  lite3_value_blob
0730: 28 61 72 67 76 5b 31 5d 29 3b 0a 20 20 69 66 28  (argv[1]);.  if(
0740: 20 7a 3d 3d 30 20 29 7b 0a 20 20 20 20 6e 20 3d   z==0 ){.    n =
0750: 20 30 3b 0a 20 20 20 20 72 63 20 3d 20 30 3b 0a   0;.    rc = 0;.
0760: 20 20 7d 65 6c 73 65 7b 0a 20 20 20 20 6e 20 3d    }else{.    n =
0770: 20 73 71 6c 69 74 65 33 5f 76 61 6c 75 65 5f 62   sqlite3_value_b
0780: 79 74 65 73 28 61 72 67 76 5b 31 5d 29 3b 0a 20  ytes(argv[1]);. 
0790: 20 20 20 72 63 20 3d 20 66 77 72 69 74 65 28 7a     rc = fwrite(z
07a0: 2c 20 31 2c 20 6e 2c 20 6f 75 74 29 3b 0a 20 20  , 1, n, out);.  
07b0: 7d 0a 20 20 66 63 6c 6f 73 65 28 6f 75 74 29 3b  }.  fclose(out);
07c0: 0a 20 20 73 71 6c 69 74 65 33 5f 72 65 73 75 6c  .  sqlite3_resul
07d0: 74 5f 69 6e 74 36 34 28 63 6f 6e 74 65 78 74 2c  t_int64(context,
07e0: 20 72 63 29 3b 0a 7d 0a 0a 0a 23 69 66 64 65 66   rc);.}...#ifdef
07f0: 20 5f 57 49 4e 33 32 0a 5f 5f 64 65 63 6c 73 70   _WIN32.__declsp
0800: 65 63 28 64 6c 6c 65 78 70 6f 72 74 29 0a 23 65  ec(dllexport).#e
0810: 6e 64 69 66 0a 69 6e 74 20 73 71 6c 69 74 65 33  ndif.int sqlite3
0820: 5f 66 69 6c 65 69 6f 5f 69 6e 69 74 28 0a 20 20  _fileio_init(.  
0830: 73 71 6c 69 74 65 33 20 2a 64 62 2c 20 0a 20 20  sqlite3 *db, .  
0840: 63 68 61 72 20 2a 2a 70 7a 45 72 72 4d 73 67 2c  char **pzErrMsg,
0850: 20 0a 20 20 63 6f 6e 73 74 20 73 71 6c 69 74 65   .  const sqlite
0860: 33 5f 61 70 69 5f 72 6f 75 74 69 6e 65 73 20 2a  3_api_routines *
0870: 70 41 70 69 0a 29 7b 0a 20 20 69 6e 74 20 72 63  pApi.){.  int rc
0880: 20 3d 20 53 51 4c 49 54 45 5f 4f 4b 3b 0a 20 20   = SQLITE_OK;.  
0890: 53 51 4c 49 54 45 5f 45 58 54 45 4e 53 49 4f 4e  SQLITE_EXTENSION
08a0: 5f 49 4e 49 54 32 28 70 41 70 69 29 3b 0a 20 20  _INIT2(pApi);.  
08b0: 28 76 6f 69 64 29 70 7a 45 72 72 4d 73 67 3b 20  (void)pzErrMsg; 
08c0: 20 2f 2a 20 55 6e 75 73 65 64 20 70 61 72 61 6d   /* Unused param
08d0: 65 74 65 72 20 2a 2f 0a 20 20 72 63 20 3d 20 73  eter */.  rc = s
08e0: 71 6c 69 74 65 33 5f 63 72 65 61 74 65 5f 66 75  qlite3_create_fu
08f0: 6e 63 74 69 6f 6e 28 64 62 2c 20 22 72 65 61 64  nction(db, "read
0900: 66 69 6c 65 22 2c 20 31 2c 20 53 51 4c 49 54 45  file", 1, SQLITE
0910: 5f 55 54 46 38 2c 20 30 2c 0a 20 20 20 20 20 20  _UTF8, 0,.      
0920: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0930: 20 20 20 20 20 20 20 20 20 72 65 61 64 66 69 6c           readfil
0940: 65 46 75 6e 63 2c 20 30 2c 20 30 29 3b 0a 20 20  eFunc, 0, 0);.  
0950: 69 66 28 20 72 63 3d 3d 53 51 4c 49 54 45 5f 4f  if( rc==SQLITE_O
0960: 4b 20 29 7b 0a 20 20 20 20 72 63 20 3d 20 73 71  K ){.    rc = sq
0970: 6c 69 74 65 33 5f 63 72 65 61 74 65 5f 66 75 6e  lite3_create_fun
0980: 63 74 69 6f 6e 28 64 62 2c 20 22 77 72 69 74 65  ction(db, "write
0990: 66 69 6c 65 22 2c 20 32 2c 20 53 51 4c 49 54 45  file", 2, SQLITE
09a0: 5f 55 54 46 38 2c 20 30 2c 0a 20 20 20 20 20 20  _UTF8, 0,.      
09b0: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
09c0: 20 20 20 20 20 20 20 20 20 20 20 77 72 69 74 65             write
09d0: 66 69 6c 65 46 75 6e 63 2c 20 30 2c 20 30 29 3b  fileFunc, 0, 0);
09e0: 0a 20 20 7d 0a 20 20 72 65 74 75 72 6e 20 72 63  .  }.  return rc
09f0: 3b 0a 7d 0a                                      ;.}.