1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
| Patch source code of shell.c :
==============================
/* patch for shell.c version 3_6_23_1 */
// I modified the local_getline() function
// from line 41:
#if defined(HAVE_READLINE) && HAVE_READLINE==1
# include <readline/readline.h>
# include <readline/history.h>
#else
# define readline(p) local_getline(p,stdin,0)
# define add_history(X)
# define read_history(X)
# define write_history(X)
# define stifle_history(X)
#endif
// from line 368:
static char *one_input_line(const char *zPrior, FILE *in){
char *zPrompt;
char *zResult;
if( in!=0 ){
return local_getline(0, in, 0); /* not in csv mode */
}
// function local_getline(), from line 320 :
/* NB: bCsvMode = 1, if the function is used by the .import command */
static char *local_getline(char *zPrompt, FILE *in, const int bCsvMode){
char *zLine;
int nLine;
int n;
int eol;
int bEscaped; /* Escape CRLF when quoted */
if( zPrompt && *zPrompt ){
printf("%s",zPrompt);
fflush(stdout);
}
nLine = 100;
zLine = malloc( nLine );
if( zLine==0 ) return 0;
n = 0;
eol = 0;
bEscaped = 0;
while( !eol ){
if( n+100>nLine ){
nLine = nLine*2 + 100;
zLine = realloc(zLine, nLine);
if( zLine==0 ) return 0;
}
if( fgets(&zLine[n], nLine - n, in)==0 ){
if( n==0 ){
free(zLine);
return 0;
}
zLine[n] = 0;
eol = 1;
break;
}
while( zLine[n] ){
if( zLine[n]=='"' && bCsvMode ) bEscaped = 1 - bEscaped;
n++;
}
if( !bEscaped && n>0 && zLine[n-1]=='\n' ){
n--;
if( n>0 && zLine[n-1]=='\r' ) n--;
zLine[n] = 0;
eol = 1;
}
}
zLine = realloc( zLine, n+1 );
return zLine;
}
// and from line 1628 :
sqlite3_exec(p->db, "BEGIN", 0, 0, 0);
zCommit = "COMMIT";
while( (zLine = local_getline(0, in, 1))!=0 ){ /* csv mode */
char *z;
char *q;
int bEscaped; /* to escape quote + comma + quote */
int bFieldQuoted;
bEscaped = 0;
i = 0;
lineno++;
if( *zLine=='"' ){
azCol[0] = zLine + 1; /* ignore the first quote */
bFieldQuoted = 1;
}else{
azCol[0] = zLine;
bFieldQuoted = 0;
}
for(i=0, z=zLine, q=zLine; *z ; z++){
if( *z=='"' ) bEscaped = 1 - bEscaped;
if( bEscaped==0 && *z==p->separator[0] && strncmp(z, p->separator, nSep)==0 ){
if( bFieldQuoted==1 && *q=='"') *q = 0; /* ignore the last quote */
*z = 0;
i++;
if( i<nCol ){
azCol[i] = &z[nSep];
if( *azCol[i]=='"' ){
azCol[i]++; /* ignore the first quote */
bFieldQuoted = 1;
}else{
bFieldQuoted = 0;
}
z += nSep-1;
}
}
q = z;
} /* end for */
if( bEscaped==0 && bFieldQuoted==1 && *q=='"' ) *q = 0; /* ignore the last quote */
*z = 0;
if( i+1!=nCol ){
fprintf(stderr,
"Error: %s line %d: expected %d columns of data but found %d\n",
zFile, lineno, nCol, i+1);
zCommit = "ROLLBACK";
free(zLine);
rc = 1;
break; /* from while */
}
for(i=0; i<nCol; i++){
/* find and reduce double quotes */
for( bEscaped=0, z=azCol[i], q=azCol[i]; *z ; z++, q++){
if( *z=='"' ) z++;
*q = *z;
}
*q = 0;
sqlite3_bind_text(pStmt, i+1, azCol[i], -1, SQLITE_STATIC);
}
sqlite3_step(pStmt);
rc = sqlite3_reset(pStmt);
free(zLine);
// end |
